Use real method for bind service

world-federation-of-advertisers · Aug 7, 2024 · e0eb1df · e0eb1df
1 parent b6ef502
commit e0eb1df
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/main/kotlin/org/wfanet/measurement/common/grpc/testing/Mocking.kt b/src/main/kotlin/org/wfanet/measurement/common/grpc/testing/Mocking.kt
@@ -27,9 +27,13 @@ inline fun <reified T : AbstractCoroutineServerImpl> mockService(
 ): T =
   mock(useConstructor = UseConstructor.parameterless()) { mock ->
     on { context }.thenCallRealMethod()
+    on { bindService() }.thenCallRealMethod()
     stubbing(mock)
   }
 
 /** Creates a mock for a gRPC coroutine service [T]. */
 inline fun <reified T : AbstractCoroutineServerImpl> mockService(): T =
-  mock(useConstructor = UseConstructor.parameterless()) { on { context }.thenCallRealMethod() }
+  mock(useConstructor = UseConstructor.parameterless()) {
+    on { context }.thenCallRealMethod()
+    on { bindService() }.thenCallRealMethod()
+  }