diff options
Diffstat (limited to 'test/sql')
| -rw-r--r-- | test/sql/test_query.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/test/sql/test_query.py b/test/sql/test_query.py index 16ba285b2..0313a9cd0 100644 --- a/test/sql/test_query.py +++ b/test/sql/test_query.py @@ -976,14 +976,22 @@ class QueryTest(fixtures.TestBase): # result.BufferedColumnResultProxy conn = testing.db.connect() - for meth in ('fetchone', 'fetchall', 'first', 'scalar', 'fetchmany'): + for meth in [ + lambda r: r.fetchone(), + lambda r: r.fetchall(), + lambda r: r.first(), + lambda r: r.scalar(), + lambda r: r.fetchmany(), + lambda r: r._getter('user'), + lambda r: r._has_key('user'), + ]: trans = conn.begin() result = conn.execute(users.insert(), user_id=1) assert_raises_message( exc.ResourceClosedError, "This result object does not return rows. " "It has been closed automatically.", - getattr(result, meth), + meth, result, ) trans.rollback() |