From e0eb1dfe1ad144c22ffe70ca01d6ffc9d41cf5de Mon Sep 17 00:00:00 2001 From: Tristan Vuong Date: Wed, 7 Aug 2024 22:20:24 +0000 Subject: [PATCH] Use real method for bind service --- .../org/wfanet/measurement/common/grpc/testing/Mocking.kt | 6 +++++- 1 file changed, 5 insertions(+), 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 index 13981d507..04029992c 100644 --- 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 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 mockService(): T = - mock(useConstructor = UseConstructor.parameterless()) { on { context }.thenCallRealMethod() } + mock(useConstructor = UseConstructor.parameterless()) { + on { context }.thenCallRealMethod() + on { bindService() }.thenCallRealMethod() + }