From 0a37913528a19c71b021bc54332aee256cb01374 Mon Sep 17 00:00:00 2001 From: Tim Ryan Date: Tue, 11 Jul 2023 15:00:33 -0400 Subject: [PATCH] RM-96 autoload with mock_db engine --- tests/unit/db/postgres/test_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/db/postgres/test_loader.py b/tests/unit/db/postgres/test_loader.py index f90fe6c35..79001662e 100644 --- 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])