Revert "Delete grpc.WithBlock() and grpc.WaitForReady(true) options (#…

…917)" (#918) This reverts commit 2d3d13c. Signed-off-by: Nikita Skrynnik <[email protected]>
networkservicemesh · Aug 14, 2023 · 1b810ff · 1b810ff
1 parent 2d3d13c
commit 1b810ff
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/internal/tests/suite_healthcheck_test.go b/internal/tests/suite_healthcheck_test.go
@@ -1,4 +1,4 @@
-// Copyright (c) 2020-2023 Cisco and/or its affiliates.
+// Copyright (c) 2020-2022 Cisco and/or its affiliates.
 //
 // SPDX-License-Identifier: Apache-2.0
 //
@@ -23,6 +23,7 @@ import (
 
 	"github.com/networkservicemesh/sdk/pkg/tools/log/logruslogger"
 
+	"google.golang.org/grpc"
 	"google.golang.org/grpc/health/grpc_health_v1"
 
 	"github.com/networkservicemesh/sdk/pkg/tools/log"
@@ -37,6 +38,7 @@ func (f *ForwarderTestSuite) TestHealthCheck() {
 		&grpc_health_v1.HealthCheckRequest{
 			Service: "networkservice.NetworkService",
 		},
+		grpc.WaitForReady(true),
 	)
 	f.NoError(err)
 	f.Require().NotNil(healthResponse)

diff --git a/internal/tests/suite_setup_test.go b/internal/tests/suite_setup_test.go
@@ -173,6 +173,7 @@ func (f *ForwarderTestSuite) SetupSuite() {
 	f.sutCC, err = grpc.DialContext(f.ctx,
 		regEndpoint.NetworkServiceEndpoint.GetUrl(),
 		grpc.WithTransportCredentials(clientCreds),
+		grpc.WithBlock(),
 		grpc.WithDefaultCallOptions(
 			grpc.PerRPCCredentials(token.NewPerRPCCredentials(spiffejwt.TokenGeneratorFunc(source, f.config.MaxTokenLifetime))),
 		),

diff --git a/main.go b/main.go
@@ -228,7 +228,9 @@ func main() {
 
 	dialOptions := append(
 		tracing.WithTracingDial(),
+		grpc.WithBlock(),
 		grpc.WithDefaultCallOptions(
+			grpc.WaitForReady(true),
 			grpc.PerRPCCredentials(token.NewPerRPCCredentials(spiffejwt.TokenGeneratorFunc(source, cfg.MaxTokenLifetime)))),
 		grpc.WithTransportCredentials(
 			grpcfd.TransportCredentials(