From 752c43d7b545cfcecb02895aee463ae91cdc7c69 Mon Sep 17 00:00:00 2001 From: Artem Belov Date: Tue, 9 Mar 2021 20:26:47 +0700 Subject: [PATCH] Fix success request waiting for windows Signed-off-by: Artem Belov --- pkg/networkservice/chains/nsmgr/server_heal_test.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pkg/networkservice/chains/nsmgr/server_heal_test.go b/pkg/networkservice/chains/nsmgr/server_heal_test.go index cab0aeb5f..9dd393294 100644 --- 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)