erik/converge
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

prepare test for more agents and clietns.

Browse Source
This commit is contained in:
Erik Brakkee 2024-08-22 10:38:45 +02:00
parent eb6da5da99
commit 2c2aeff8f5
1 changed files with 11 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
pkg/server/admin/admin_test.go
View File

@ -37,13 +37,15 @@ type AdminTestSuite struct {
cancelFunc context.CancelFunc
pprofServer *http.Server
agentReadWriter io.ReadWriteCloser
serverReadWriter io.ReadWriteCloser
admin *Admin
hostKey []byte
}
func (s *AdminTestSuite) createPipe() (io.ReadWriteCloser, io.ReadWriteCloser) {
bitpipe := testsupport.NewInmemoryConnection(s.ctx, "inmemory", 10)
return bitpipe.Front(), bitpipe.Back()
}
func (s *AdminTestSuite) SetupSuite() {
s.pprofServer = testsupport.StartPprof("")
}
@ -57,16 +59,6 @@ func (s *AdminTestSuite) SetupTest() {
s.ctx = ctx
s.cancelFunc = cancelFunc
// Could have also used net.Pipe but net.Pipe uses synchronous communication
// by default and the bitpipe implementation can become asynchronous when
// a channels ize > 0 is passed in. Also the test utility respects the context
// so also deals with cancellation much better than net.Pipe.
bitpipe := testsupport.NewInmemoryConnection(s.ctx, "inmemory", 10)
agentReadWriter := bitpipe.Front()
serverReadWriter := bitpipe.Back()
s.agentReadWriter = agentReadWriter
s.serverReadWriter = serverReadWriter
s.admin = NewAdmin()
s.hostKey = make([]byte, 100)
rand.Read(s.hostKey)
@ -81,10 +73,13 @@ func TestAdminTestSuite(t *testing.T) {
func (s *AdminTestSuite) Test_AgentRegisters() {
publicId := "abc"
agentRW, serverRW := s.createPipe()
testsupport.RunAndWait(
&s.Suite,
func() any {
agentConn, err := s.admin.AddAgent(s.hostKey, models.RendezVousId(publicId), comms.EnvironmentInfo{}, s.serverReadWriter)
agentConn, err := s.admin.AddAgent(
s.hostKey, models.RendezVousId(publicId), comms.EnvironmentInfo{},
serverRW)
s.Nil(err)
s.Equal(publicId, string(agentConn.Info.PublicId))
state := s.admin.CreateNotifification()
@ -95,12 +90,12 @@ func (s *AdminTestSuite) Test_AgentRegisters() {
},
func() any {
// verify registration message received
agentRegistration, err := comms.ReceiveRegistrationMessage(s.agentReadWriter)
agentRegistration, err := comms.ReceiveRegistrationMessage(agentRW)
s.Nil(err)
s.True(agentRegistration.Ok)
s.Equal(s.hostKey, agentRegistration.HostPrivateKey)
commChannel, err := comms.NewCommChannel(comms.Agent, s.agentReadWriter)
commChannel, err := comms.NewCommChannel(comms.Agent, agentRW)
s.Nil(err)
s.NotNil(commChannel)
return nil