diff options
Diffstat (limited to 'lib/sqlalchemy/testing/suite/test_update_delete.py')
| -rw-r--r-- | lib/sqlalchemy/testing/suite/test_update_delete.py | 31 |
1 files changed, 16 insertions, 15 deletions
diff --git a/lib/sqlalchemy/testing/suite/test_update_delete.py b/lib/sqlalchemy/testing/suite/test_update_delete.py index 97bdf0ad7..6003a0994 100644 --- a/lib/sqlalchemy/testing/suite/test_update_delete.py +++ b/lib/sqlalchemy/testing/suite/test_update_delete.py @@ -22,33 +22,34 @@ class SimpleUpdateDeleteTest(fixtures.TablesTest): @classmethod def insert_data(cls): - config.db.execute( - cls.tables.plain_pk.insert(), - [ - {"id": 1, "data": "d1"}, - {"id": 2, "data": "d2"}, - {"id": 3, "data": "d3"}, - ], - ) - - def test_update(self): + with config.db.connect() as conn: + conn.execute( + cls.tables.plain_pk.insert(), + [ + {"id": 1, "data": "d1"}, + {"id": 2, "data": "d2"}, + {"id": 3, "data": "d3"}, + ], + ) + + def test_update(self, connection): t = self.tables.plain_pk - r = config.db.execute(t.update().where(t.c.id == 2), data="d2_new") + r = connection.execute(t.update().where(t.c.id == 2), data="d2_new") assert not r.is_insert assert not r.returns_rows eq_( - config.db.execute(t.select().order_by(t.c.id)).fetchall(), + connection.execute(t.select().order_by(t.c.id)).fetchall(), [(1, "d1"), (2, "d2_new"), (3, "d3")], ) - def test_delete(self): + def test_delete(self, connection): t = self.tables.plain_pk - r = config.db.execute(t.delete().where(t.c.id == 2)) + r = connection.execute(t.delete().where(t.c.id == 2)) assert not r.is_insert assert not r.returns_rows eq_( - config.db.execute(t.select().order_by(t.c.id)).fetchall(), + connection.execute(t.select().order_by(t.c.id)).fetchall(), [(1, "d1"), (3, "d3")], ) |