RM-96 autoload with mock_db engine

bluelabsio · Jul 11, 2023 · 0a37913 · 0a37913
1 parent fec371c
commit 0a37913
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/db/postgres/test_loader.py b/tests/unit/db/postgres/test_loader.py
@@ -173,7 +173,7 @@ def test_load(self,
         mock_Table.assert_called_with(mock_table,
                                       self.mock_meta,
                                       schema=mock_schema,
-                                      autoload_with=self.mock_db)
+                                      autoload_with=self.mock_db.engine)
         mock_conn = self.mock_db.engine.begin.return_value.__enter__.return_value
         mock_quote_value.assert_called_with(mock_conn, 'ISO, DATE_ORDER_STYLE')
         str_arg = str(mock_conn.execute.call_args.args[0])