mock create_channel

googleapis · Nov 12, 2024 · 99dd32a · 99dd32a
1 parent 39f12f1
commit 99dd32a
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/tests/unit/data/_async/test_client.py b/tests/unit/data/_async/test_client.py
@@ -431,15 +431,16 @@ async def test__manage_channel_sleeps(
                     ]
                     client = self._make_client(project="project-id")
                     client.transport._grpc_channel = channel
-                    try:
-                        if refresh_interval is not None:
-                            await client._manage_channel(
-                                refresh_interval, refresh_interval, grace_period=0
-                            )
-                        else:
-                            await client._manage_channel(grace_period=0)
-                    except asyncio.CancelledError:
-                        pass
+                    with mock.patch.object(client.transport, "create_channel", CrossSync.Mock):
+                        try:
+                            if refresh_interval is not None:
+                                await client._manage_channel(
+                                    refresh_interval, refresh_interval, grace_period=0
+                                )
+                            else:
+                                await client._manage_channel(grace_period=0)
+                        except asyncio.CancelledError:
+                            pass
                     assert sleep.call_count == num_cycles
                     total_sleep = sum([call[0][1] for call in sleep.call_args_list])
                     assert (