diff options
-rw-r--r-- | oslo_db/tests/sqlalchemy/test_sqlalchemy.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/oslo_db/tests/sqlalchemy/test_sqlalchemy.py b/oslo_db/tests/sqlalchemy/test_sqlalchemy.py index 7acbc1e..bcc4a1c 100644 --- a/oslo_db/tests/sqlalchemy/test_sqlalchemy.py +++ b/oslo_db/tests/sqlalchemy/test_sqlalchemy.py @@ -665,9 +665,11 @@ class PatchStacktraceTest(test_base.DbTestCase): session._add_trace_comments(engine) conn = engine.connect() + orig_do_exec = engine.dialect.do_execute with mock.patch.object(engine.dialect, "do_execute") as mock_exec: - conn.execute("select * from table") + mock_exec.side_effect = orig_do_exec + conn.execute("select 1;") call = mock_exec.mock_calls[0] |