diff options
Diffstat (limited to 'src/mailman/database/tests/test_migrations.py')
| -rw-r--r-- | src/mailman/database/tests/test_migrations.py | 18 |
1 files changed, 10 insertions, 8 deletions
diff --git a/src/mailman/database/tests/test_migrations.py b/src/mailman/database/tests/test_migrations.py index b840842cb..d35a28d0f 100644 --- a/src/mailman/database/tests/test_migrations.py +++ b/src/mailman/database/tests/test_migrations.py @@ -185,16 +185,18 @@ class TestMigrations(unittest.TestCase): sa.sql.column('digests_enabled', sa.Boolean) ) # Downgrading. - for table_id, enabled in IDS_TO_DIGESTABLE: - config.db.store.execute(mlist_table.insert().values( - id=table_id, digests_enabled=enabled)) - config.db.store.commit() - alembic.command.downgrade(alembic_cfg, '47294d3a604') - results = config.db.store.execute( - 'SELECT id, digestable FROM mailinglist').fetchall() + with transaction(): + for table_id, enabled in IDS_TO_DIGESTABLE: + config.db.store.execute(mlist_table.insert().values( + id=table_id, digests_enabled=enabled)) + with transaction(): + alembic.command.downgrade(alembic_cfg, '47294d3a604') + results = config.db.store.execute( + 'SELECT id, digestable FROM mailinglist').fetchall() self.assertEqual(results, IDS_TO_DIGESTABLE) # Upgrading. - alembic.command.upgrade(alembic_cfg, '70af5a4e5790') + with transaction(): + alembic.command.upgrade(alembic_cfg, '70af5a4e5790') results = config.db.store.execute( 'SELECT id, digests_enabled FROM mailinglist').fetchall() self.assertEqual(results, IDS_TO_DIGESTABLE) |