File tree Expand file tree Collapse file tree 2 files changed +10
-0
lines changed
Expand file tree Collapse file tree 2 files changed +10
-0
lines changed Original file line number Diff line number Diff line change @@ -376,8 +376,14 @@ async def test__manage_channel_ping_and_warm(self):
376376 """
377377 import time
378378 import threading
379+ if CrossSync .is_async :
380+ from google .cloud .bigtable_v2 .services .bigtable .transports .grpc_asyncio import _LoggingClientAIOInterceptor as Interceptor
381+ else :
382+ from google .cloud .bigtable_v2 .services .bigtable .transports .grpc import _LoggingClientInterceptor as Interceptor
383+
379384
380385 client_mock = mock .Mock ()
386+ client_mock .transport ._interceptor = Interceptor ()
381387 client_mock ._is_closed .is_set .return_value = False
382388 client_mock ._channel_init_time = time .monotonic ()
383389 orig_channel = client_mock .transport .grpc_channel
Original file line number Diff line number Diff line change @@ -289,8 +289,12 @@ def test__manage_channel_ping_and_warm(self):
289289 """_manage channel should call ping and warm internally"""
290290 import time
291291 import threading
292+ from google .cloud .bigtable_v2 .services .bigtable .transports .grpc import (
293+ _LoggingClientInterceptor as Interceptor ,
294+ )
292295
293296 client_mock = mock .Mock ()
297+ client_mock .transport ._interceptor = Interceptor ()
294298 client_mock ._is_closed .is_set .return_value = False
295299 client_mock ._channel_init_time = time .monotonic ()
296300 orig_channel = client_mock .transport .grpc_channel
You can’t perform that action at this time.
0 commit comments