mirror of
https://github.com/lingble/talos.git
synced 2025-11-07 16:23:19 +00:00
96aa9638f7
There's a cyclic dependency on siderolink library which imports talos machinery back. We will fix that after we get talos pushed under a new name. Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
75 lines
1.5 KiB
Go
75 lines
1.5 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package backend_test
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/siderolabs/grpc-proxy/proxy"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/siderolabs/talos/internal/app/apid/pkg/backend"
|
|
)
|
|
|
|
type APIDFactorySuite struct {
|
|
suite.Suite
|
|
|
|
f *backend.APIDFactory
|
|
}
|
|
|
|
func (suite *APIDFactorySuite) SetupSuite() {
|
|
suite.f = backend.NewAPIDFactory(&tls.Config{})
|
|
}
|
|
|
|
func (suite *APIDFactorySuite) TestGet() {
|
|
b1, err := suite.f.Get("127.0.0.1")
|
|
suite.Require().NoError(err)
|
|
suite.Require().NotNil(b1)
|
|
|
|
b2, err := suite.f.Get("127.0.0.1")
|
|
suite.Require().NoError(err)
|
|
suite.Require().Equal(b1, b2)
|
|
|
|
b3, err := suite.f.Get("127.0.0.2")
|
|
suite.Require().NoError(err)
|
|
suite.Require().NotEqual(b1, b3)
|
|
|
|
_, err = suite.f.Get("127.0.0.2:50000")
|
|
suite.Require().Error(err)
|
|
}
|
|
|
|
func (suite *APIDFactorySuite) TestGetConcurrent() {
|
|
// for race detector
|
|
var wg sync.WaitGroup
|
|
|
|
backendCh := make(chan proxy.Backend, 10)
|
|
|
|
for i := 0; i < 10; i++ {
|
|
wg.Add(1)
|
|
|
|
go func() {
|
|
defer wg.Done()
|
|
|
|
b, _ := suite.f.Get("10.0.0.1") //nolint:errcheck
|
|
backendCh <- b
|
|
}()
|
|
}
|
|
|
|
wg.Wait()
|
|
close(backendCh)
|
|
|
|
b := <-backendCh
|
|
|
|
for anotherB := range backendCh {
|
|
suite.Assert().Equal(b, anotherB)
|
|
}
|
|
}
|
|
|
|
func TestAPIDFactorySuite(t *testing.T) {
|
|
suite.Run(t, new(APIDFactorySuite))
|
|
}
|