modified has_async_methods and has_sync_methods of TestServerBuilderPlugin

1 files changed, 12 insertions, 2 deletions

diff --git a/test/cpp/end2end/server_builder_plugin_test.cc b/test/cpp/end2end/server_builder_plugin_test.cc
index e42bd10832..bebf1d1979 100644
--- a/test/cpp/end2end/server_builder_plugin_test.cc
+++ b/test/cpp/end2end/server_builder_plugin_test.cc
@@ -78,9 +78,19 @@ class TestServerBuilderPlugin : public ServerBuilderPlugin {
     change_arguments_is_called_ = true;
   }
 
-  bool has_async_methods() const GRPC_OVERRIDE { return register_service_; }
+  bool has_async_methods() const GRPC_OVERRIDE {
+    if (register_service_) {
+      return service_->has_async_methods();
+    }
+    return false;
+  }
 
-  bool has_sync_methods() const GRPC_OVERRIDE { return register_service_; }
+  bool has_sync_methods() const GRPC_OVERRIDE {
+    if (register_service_) {
+      return service_->has_synchronous_methods();
+    }
+    return false;
+  }
 
   void SetRegisterService() { register_service_ = true; }