Added emptycall to stubserver

grpc · Dec 19, 2024 · a2ae0e4 · a2ae0e4
1 parent 2899311
commit a2ae0e4
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 5 deletions.
diff --git a/test/creds_test.go b/test/creds_test.go
@@ -434,7 +434,10 @@ func (s) TestCredsHandshakeAuthority(t *testing.T) {
 	cred := &authorityCheckCreds{}
 	stub := &stubserver.StubServer{
 		Listener: lis,
-		S:        grpc.NewServer(),
+		EmptyCallF: func(ctx context.Context, in *testpb.Empty) (*testpb.Empty, error) {
+			return &testpb.Empty{}, nil
+		},
+		S: grpc.NewServer(),
 	}
 	stubserver.StartTestService(t, stub)
 	defer stub.S.Stop()
@@ -469,7 +472,10 @@ func (s) TestCredsHandshakeServerNameAuthority(t *testing.T) {
 
 	stub := &stubserver.StubServer{
 		Listener: lis,
-		S:        grpc.NewServer(),
+		EmptyCallF: func(ctx context.Context, in *testpb.Empty) (*testpb.Empty, error) {
+			return &testpb.Empty{}, nil
+		},
+		S: grpc.NewServer(),
 	}
 	stubserver.StartTestService(t, stub)
 	defer stub.S.Stop()

diff --git a/test/goaway_test.go b/test/goaway_test.go
@@ -734,7 +734,10 @@ func (s) TestTwoGoAwayPingFrames(t *testing.T) {
 	defer lis.Close()
 	ss := &stubserver.StubServer{
 		Listener: lis,
-		S:        grpc.NewServer(),
+		EmptyCallF: func(ctx context.Context, in *testpb.Empty) (*testpb.Empty, error) {
+			return &testpb.Empty{}, nil
+		},
+		S: grpc.NewServer(),
 	}
 	stubserver.StartTestService(t, ss)
 	defer ss.S.Stop()

diff --git a/test/healthcheck_test.go b/test/healthcheck_test.go
@@ -154,7 +154,10 @@ func setupServer(t *testing.T, watchFunc healthWatchFunc) (*grpc.Server, net.Lis
 
 	stub := &stubserver.StubServer{
 		Listener: lis,
-		S:        grpc.NewServer(),
+		EmptyCallF: func(ctx context.Context, in *testpb.Empty) (*testpb.Empty, error) {
+			return &testpb.Empty{}, nil
+		},
+		S: grpc.NewServer(),
 	}
 	healthgrpc.RegisterHealthServer(stub.S, ts)
 	stubserver.StartTestService(t, stub)
@@ -260,7 +263,10 @@ func (s) TestHealthCheckHealthServerNotRegistered(t *testing.T) {
 	}
 	stub := &stubserver.StubServer{
 		Listener: lis,
-		S:        grpc.NewServer(),
+		EmptyCallF: func(ctx context.Context, in *testpb.Empty) (*testpb.Empty, error) {
+			return &testpb.Empty{}, nil
+		},
+		S: grpc.NewServer(),
 	}
 	defer stub.S.Stop()