Fix heal tests

Signed-off-by: Vladimir Popov <vladimir.popov@xored.com>

pkg/networkservice/chains/nsmgr/heal_test.go

@@ -252,7 +252,7 @@ func testNSMGRHealNSMgr(t *testing.T, nodeNum int, restored bool) {
 	} else {
 		domain.Nodes[nodeNum].NSMgr.Cancel()
 		// We need to be sure that Client has noticed NSMgr death
-		time.Sleep(time.Second)
+		time.Sleep(3 * time.Second)
 		domain.Nodes[nodeNum].NSMgr.Restart()
 	}
 

pkg/networkservice/chains/nsmgr/scale_test.go

@@ -38,7 +38,7 @@ import (
 func TestCreateEndpointDuringRequest(t *testing.T) {
 	t.Cleanup(func() { goleak.VerifyNone(t) })
 
-	ctx, cancel := context.WithTimeout(context.Background(), time.Second)
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
 
 	defer cancel()
 	domain := sandbox.NewBuilder(ctx, t).

pkg/networkservice/chains/nsmgr/suite_test.go

@@ -331,7 +331,7 @@ func (s *nsmgrSuite) Test_ConnectToDeadNSEUsecase() {
 
 	request := defaultRequest(nsReg.Name)
 
-	reqCtx, reqCancel := context.WithTimeout(ctx, time.Second)
+	reqCtx, reqCancel := context.WithTimeout(ctx, 2*time.Second)
 	defer reqCancel()
 
 	conn, err := nsc.Request(reqCtx, request.Clone())
@@ -346,10 +346,7 @@ func (s *nsmgrSuite) Test_ConnectToDeadNSEUsecase() {
 	refreshRequest := request.Clone()
 	refreshRequest.Connection = conn.Clone()
 
-	refreshReqCtx, refreshReqCancel := context.WithTimeout(ctx, time.Second)
-	defer refreshReqCancel()
-
-	_, err = nsc.Request(refreshReqCtx, refreshRequest)
+	_, err = nsc.Request(ctx, refreshRequest)
 	require.Error(t, err)
 	require.NoError(t, reqCtx.Err())
 

pkg/networkservice/common/heal/heal_test.go

@@ -101,5 +101,5 @@ func TestHeal_CloseChain(t *testing.T) {
 
 	require.Eventually(t, func() bool {
 		return counter.Closes() == 1
-	}, time.Second, 10*time.Millisecond)
+	}, 3*time.Second, 10*time.Millisecond)
 }