benjamyn/any-sync
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

yamux fix test race

Browse Source
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
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
net/transport/yamux/yamux_test.go
View File

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