Skip to content

Commit 73a3337

Browse files
committed
fixed test
1 parent 59f5ff8 commit 73a3337

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

tests/unit/data/_async/test_client.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff 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

tests/unit/data/_sync_autogen/test_client.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)