Add missed wait

Signed-off-by: Vladimir Popov <vladimir.popov@xored.com>
networkservicemesh · Apr 27, 2021 · 43d60ec · 43d60ec
1 parent bfb33db
commit 43d60ec
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/networkservice/chains/nsmgr/nsmgr_test.go b/pkg/networkservice/chains/nsmgr/nsmgr_test.go
@@ -126,6 +126,7 @@ func (s *nsmgrSuite) Test_Remote_ParallelUsecase() {
 	require.Equal(t, int32(1), atomic.LoadInt32(&counter.Closes))
 
 	// Endpoint unregister
+	unregisterWG.Wait()
 	_, err = s.domain.Nodes[0].EndpointRegistryClient.Unregister(ctx, nseReg)
 	require.NoError(t, err)
 }
@@ -240,6 +241,7 @@ func (s *nsmgrSuite) Test_Remote_BusyEndpointsUsecase() {
 	require.Equal(t, int32(1), atomic.LoadInt32(&counter.Closes))
 
 	// Endpoint unregister
+	unregisterWG.Wait()
 	for i := 0; i < len(nsesReg); i++ {
 		_, err = s.domain.Nodes[1].EndpointRegistryClient.Unregister(ctx, nsesReg[i])
 		require.NoError(t, err)