1
0
Fork 0
mirror of https://github.com/anyproto/any-sync.git synced 2025-06-09 09:35:03 +09:00

yamux fix test race

This commit is contained in:
Sergey Cherepanov 2023-06-06 18:50:20 +02:00
parent 5090006e2d
commit 2ec42e5e1f
No known key found for this signature in database
GPG key ID: 87F8EDE8FBDF637C

View file

@ -31,7 +31,7 @@ func TestYamuxTransport_Dial(t *testing.T) {
mcC, err := fxC.Dial(ctx, fxS.addr)
require.NoError(t, err)
require.Len(t, fxS.accepter.mcs, 1)
mcS := fxS.accepter.mcs[0]
mcS := <-fxS.accepter.mcs
var (
sData string
@ -83,7 +83,7 @@ func TestWriteBench(t *testing.T) {
mcC, err := fxC.Dial(ctx, fxS.addr)
require.NoError(t, err)
mcS := fxS.accepter.mcs[0]
mcS := <-fxS.accepter.mcs
go func() {
for i := 0; i < numSubConn; i++ {
@ -139,7 +139,7 @@ func newFixture(t *testing.T) *fixture {
yamuxTransport: New().(*yamuxTransport),
ctrl: gomock.NewController(t),
acc: &accounttest.AccountTestService{},
accepter: &testAccepter{},
accepter: &testAccepter{mcs: make(chan transport.MultiConn, 100)},
a: new(app.App),
}
@ -179,11 +179,11 @@ func (c *testConf) GetYamux() Config {
type testAccepter struct {
err error
mcs []transport.MultiConn
mcs chan transport.MultiConn
}
func (t *testAccepter) Accept(mc transport.MultiConn) (err error) {
t.mcs = append(t.mcs, mc)
t.mcs <- mc
return t.err
}