def test_downgrade_not_applied(tmpdir, caplog):
manager = DatabaseManager('sqlite:///:memory:', directory=tmpdir)
manager.revision()
migrations = manager.migration_files
manager.downgrade(migrations[0])
assert 'not yet applied: {}'.format(migrations[0]) in caplog.text
def test_downgrade_not_applied(tmpdir, caplog):
manager = DatabaseManager('sqlite:///:memory:', directory=tmpdir)
manager.revision()
migrations = manager.migration_files
manager.downgrade(migrations[0])
assert 'not yet applied: {}'.format(migrations[0]) in caplog.text
def test_migration_0006_to_0005(db_0006_with_data):
# Verify we have data
with orm.db:
metric_0006 = orm.db.execute_sql('SELECT * FROM "Metric"').fetchall()
data_0006 = orm.db.execute_sql('SELECT * FROM "Datapoint"').fetchall()
migrations = orm.db.execute_sql(
'SELECT * FROM "migration_history"').fetchall()
assert len(metric_0006) != 0
assert len(data_0006) != 0
# Make sure we have migration 0006 applied.
msg = "Migration 0006 is not applied, it should be."
assert any("0006" in m[1] for m in migrations), msg
# Then downgrade to 0005. `orm.create_db` doesn't have any downgrade
# capability, so we need to use `manager.downgrade()`
manager = DatabaseManager(SqliteDatabase(str(db_0006_with_data)))
manager.downgrade("0005")
with orm.db:
metric_0005 = orm.db.execute_sql('SELECT * FROM "Metric"').fetchall()
data_0005 = orm.db.execute_sql('SELECT * FROM "Datapoint"').fetchall()
migrations = orm.db.execute_sql(
'SELECT * FROM "migration_history"').fetchall()
# Ensure that migration 0006 *is not* applied.
msg = "Migration 0006 applied when it shouldn't be."
assert not any("0006" in m[1] for m in migrations), msg
# And that data still matches.
assert len(metric_0005) != 0
assert metric_0005 == metric_0006
assert len(data_0005) != 0
assert data_0005 == data_0006
def outdated_db(up_to_date_db):
"""
Return a database file that is missing some migrations.
"""
# Rename things to avoid confusion
path = up_to_date_db
# Start with an up-to-date database, then downgrade
manager = DatabaseManager(SqliteDatabase(str(path)))
manager.downgrade()
yield path
def test_downgrade_single(tmpdir, caplog):
manager = DatabaseManager('sqlite:///:memory:', directory=tmpdir)
manager.revision()
manager.revision()
manager.upgrade()
migrations = manager.migration_files
assert manager.db_migrations == (migrations[0], migrations[1],)
assert not manager.diff
manager.downgrade()
assert 'downgrade: {}'.format(migrations[1]) in caplog.text
assert manager.db_migrations == (migrations[0],)
assert manager.diff == (migrations[1],)
def test_downgrade_single(tmpdir, caplog):
manager = DatabaseManager('sqlite:///:memory:', directory=tmpdir)
manager.revision()
manager.revision()
manager.upgrade()
migrations = manager.migration_files
assert manager.db_migrations == (
migrations[0],
migrations[1],
)
assert not manager.diff
manager.downgrade()
assert 'downgrade: {}'.format(migrations[1]) in caplog.text
assert manager.db_migrations == (migrations[0], )
assert manager.diff == (migrations[1], )
def test_downgrade_target_error(tmpdir, caplog):
manager = DatabaseManager('sqlite:///:memory:', directory=tmpdir)
manager.revision()
manager.downgrade('does-not-exist')
assert 'could not find migration: does-not-exist' in caplog.text
def test_downgrade_nodiff(tmpdir, caplog):
manager = DatabaseManager('sqlite:///:memory:', directory=tmpdir)
manager.downgrade()
assert 'migrations not yet applied!' in caplog.text
def test_downgrade_target_error(tmpdir, caplog):
manager = DatabaseManager('sqlite:///:memory:', directory=tmpdir)
manager.revision()
manager.downgrade('does-not-exist')
assert 'could not find migration: does-not-exist' in caplog.text
def test_downgrade_nodiff(tmpdir, caplog):
manager = DatabaseManager('sqlite:///:memory:', directory=tmpdir)
manager.downgrade()
assert 'migrations not yet applied!' in caplog.text