diff options
Diffstat (limited to 'lib/sqlalchemy/testing/suite/test_insert.py')
| -rw-r--r-- | lib/sqlalchemy/testing/suite/test_insert.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/lib/sqlalchemy/testing/suite/test_insert.py b/lib/sqlalchemy/testing/suite/test_insert.py index f0e4bfcc6..2307d3b3f 100644 --- a/lib/sqlalchemy/testing/suite/test_insert.py +++ b/lib/sqlalchemy/testing/suite/test_insert.py @@ -125,10 +125,14 @@ class InsertBehaviorTest(fixtures.TablesTest): # case, the row had to have been consumed at least. assert not r.returns_rows or r.fetchone() is None - @requirements.returning + @requirements.insert_returning def test_autoclose_on_insert_implicit_returning(self, connection): r = connection.execute( - self.tables.autoinc_pk.insert(), dict(data="some data") + # return_defaults() ensures RETURNING will be used, + # new in 2.0 as sqlite/mariadb offer both RETURNING and + # cursor.lastrowid + self.tables.autoinc_pk.insert().return_defaults(), + dict(data="some data"), ) assert r._soft_closed assert not r.closed @@ -295,7 +299,7 @@ class InsertBehaviorTest(fixtures.TablesTest): class ReturningTest(fixtures.TablesTest): run_create_tables = "each" - __requires__ = "returning", "autoincrement_insert" + __requires__ = "insert_returning", "autoincrement_insert" __backend__ = True def _assert_round_trip(self, table, conn): |