RM-96 pass mock_conn

bluelabsio · Jul 24, 2023 · 71c5b09 · 71c5b09
1 parent e6fb9cd
commit 71c5b09
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/records/test_prep.py b/tests/unit/records/test_prep.py
@@ -172,10 +172,10 @@ def test_prep_table_exists_drop_explicit(self, mock_quote_schema_and_table):
                                self.mock_tbl.table_name,
                                self.mock_tbl.add_group_perms_for,
                                None,
-                               db_conn=mock_driver)
+                               db_conn=mock_conn)
         mock_driver.set_grant_permissions_for_users.\
             assert_called_with(self.mock_tbl.schema_name,
                                self.mock_tbl.table_name,
                                self.mock_tbl.add_user_perms_for,
                                None,
-                               db_conn=mock_driver)
+                               db_conn=mock_conn)