modifying unit tests

dbt-labs · Oct 23, 2023 · 6dcb67e · 6dcb67e
1 parent ce59491
commit 6dcb67e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/unit/test_bigquery_adapter.py b/tests/unit/test_bigquery_adapter.py
@@ -24,6 +24,7 @@
 from .utils import (
     config_from_parts_or_dicts,
     inject_adapter,
+    mock_connection,
     TestAdapterConversions,
 )
 
@@ -348,8 +349,8 @@ def test_cancel_open_connections_empty(self):
 
     def test_cancel_open_connections_master(self):
         adapter = self.get_adapter("oauth")
-        adapter.connections.thread_connections[0] = object()
-        self.assertEqual(adapter.cancel_open_connections(), None)
+        adapter.connections.thread_connections[0] = mock_connection("master")
+        self.assertEqual(len(list(adapter.cancel_open_connections())), 0)
 
     def test_cancel_open_connections_single(self):
         adapter = self.get_adapter("oauth")