Fix memory registry slow receiver test

Signed-off-by: Vladimir Popov <vladimir.popov@xored.com>
networkservicemesh · Feb 5, 2021 · b362e5c · b362e5c
1 parent 40e3c98
commit b362e5c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 14 deletions.
diff --git a/pkg/registry/common/memory/ns_server_test.go b/pkg/registry/common/memory/ns_server_test.go
@@ -173,17 +173,13 @@ func TestNetworkServiceRegistryServer_SlowReceiver(t *testing.T) {
 	findCtx, findCancel := context.WithCancel(ctx)
 
 	stream, err := c.Find(findCtx, &registry.NetworkServiceQuery{
-		NetworkService: &registry.NetworkService{Name: "ns-1"},
+		NetworkService: &registry.NetworkService{Name: "ns"},
 		Watch:          true,
 	})
 	require.NoError(t, err)
 
-	for i := 0; i < 100; i++ {
-		var ns *registry.NetworkService
-		ns, err = s.Register(ctx, &registry.NetworkService{Name: "ns-1"})
-		require.NoError(t, err)
-
-		_, err = s.Unregister(ctx, ns)
+	for i := 0; i < 200; i++ {
+		_, err = s.Register(ctx, &registry.NetworkService{Name: fmt.Sprintf("ns-%d", i)})
 		require.NoError(t, err)
 	}
 

diff --git a/pkg/registry/common/memory/nse_server_test.go b/pkg/registry/common/memory/nse_server_test.go
@@ -240,17 +240,13 @@ func TestNetworkServiceEndpointRegistryServer_SlowReceiver(t *testing.T) {
 	findCtx, findCancel := context.WithCancel(ctx)
 
 	stream, err := c.Find(findCtx, &registry.NetworkServiceEndpointQuery{
-		NetworkServiceEndpoint: &registry.NetworkServiceEndpoint{Name: "nse-1"},
+		NetworkServiceEndpoint: &registry.NetworkServiceEndpoint{Name: "nse"},
 		Watch:                  true,
 	})
 	require.NoError(t, err)
 
-	for i := 0; i < 100; i++ {
-		var nse *registry.NetworkServiceEndpoint
-		nse, err = s.Register(ctx, &registry.NetworkServiceEndpoint{Name: "nse-1"})
-		require.NoError(t, err)
-
-		_, err = s.Unregister(ctx, nse)
+	for i := 0; i < 200; i++ {
+		_, err = s.Register(ctx, &registry.NetworkServiceEndpoint{Name: fmt.Sprintf("nse-%d", i)})
 		require.NoError(t, err)
 	}