Fix success request waiting for windows

Signed-off-by: Artem Belov <artem.belov@xored.com>
networkservicemesh · Mar 9, 2021 · 59e2b7a · 59e2b7a
1 parent 16d1029
commit 59e2b7a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/networkservice/chains/nsmgr/server_heal_test.go b/pkg/networkservice/chains/nsmgr/server_heal_test.go
@@ -274,7 +274,9 @@ func testNSMGRHealNSMgr(t *testing.T, nodeNum int, customConfig []*sandbox.NodeC
 	require.NoError(t, err)
 
 	// Wait Cross NSE expired and reconnecting through the new Cross NSE
-	require.Eventually(t, checkSecondRequestsReceived(counter.UniqueRequests), timeout, tick)
+	require.Eventually(t, checkSecondRequestsReceived(func() int {
+		return int(atomic.LoadInt32(&counter.Requests))
+	}), timeout, tick)
 
 	// Close.
 	closes := atomic.LoadInt32(&counter.Closes)