fix races in network test middleware

onflow · Apr 12, 2022 · 71b3fdc · 71b3fdc
1 parent 4851614
commit 71b3fdc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/network/test/middleware_test.go b/network/test/middleware_test.go
@@ -120,11 +120,12 @@ func (m *MiddlewareTestSuite) SetupTest() {
 	var errChan <-chan error
 	m.mwCtx, errChan = irrecoverable.WithSignaler(ctx)
 
+	mwCtx := m.mwCtx
 	go func() {
 		select {
 		case err := <-errChan:
 			m.T().Error("middlewares encountered fatal error", err)
-		case <-m.mwCtx.Done():
+		case <-mwCtx.Done():
 			return
 		}
 	}()
@@ -425,7 +426,7 @@ func (m *MiddlewareTestSuite) TestLargeMessageSize_SendDirect() {
 	require.NoError(m.Suite.T(), err)
 
 	// check message reception on target
-	unittest.RequireCloseBefore(m.T(), ch, 15*time.Second, "source node failed to send large message to target")
+	unittest.RequireCloseBefore(m.T(), ch, 60*time.Second, "source node failed to send large message to target")
 
 	m.ov[targetIndex].AssertExpectations(m.T())
 }