diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2017-07-13 18:32:42 -0400 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2017-08-07 15:05:00 -0400 |
| commit | 68879d50faa9e2602e55d5d191647b1cf864e5ab (patch) | |
| tree | be9f9e906a3674aa7237ae564eee244931399bae /lib/sqlalchemy/testing | |
| parent | 4b4f8fbf25f1a5a76c1579c1a3fd6ffad07c8c66 (diff) | |
| download | sqlalchemy-68879d50faa9e2602e55d5d191647b1cf864e5ab.tar.gz | |
Enable multi-level selectin polymorphic loading
Change-Id: Icc742bbeecdb7448ce84caccd63e086af16e81c1
Fixes: #4026
Diffstat (limited to 'lib/sqlalchemy/testing')
| -rw-r--r-- | lib/sqlalchemy/testing/assertions.py | 13 |
1 files changed, 5 insertions, 8 deletions
diff --git a/lib/sqlalchemy/testing/assertions.py b/lib/sqlalchemy/testing/assertions.py index c0854ea55..08d0f0aac 100644 --- a/lib/sqlalchemy/testing/assertions.py +++ b/lib/sqlalchemy/testing/assertions.py @@ -520,13 +520,10 @@ class AssertsExecutionResults(object): db, callable_, assertsql.CountStatements(count)) @contextlib.contextmanager - def assert_execution(self, *rules): - assertsql.asserter.add_rules(rules) - try: + def assert_execution(self, db, *rules): + with self.sql_execution_asserter(db) as asserter: yield - assertsql.asserter.statement_complete() - finally: - assertsql.asserter.clear_rules() + asserter.assert_(*rules) - def assert_statement_count(self, count): - return self.assert_execution(assertsql.CountStatements(count)) + def assert_statement_count(self, db, count): + return self.assert_execution(db, assertsql.CountStatements(count)) |