Python SchemaMigration Examples

Example #1

def test_database_fresh():
    db = db_connect(':memory:', 'peewee')

    with DatabaseContext.use(db):
        m = SchemaMigration()

        # Run fresh schema migration
        assert m.run() is True

Example #2

File: schema_tests.py Project: Danik1601/Plex-Trakt-Scrobbler

def test_database_fresh():
    db = db_connect(':memory:', 'peewee')

    with DatabaseContext.use(db):
        m = SchemaMigration()

        # Run fresh schema migration
        assert m.run() is True

Example #3

def test_database_upgrade():
    db = db_connect(':memory:', 'peewee')

    with DatabaseContext.use(db):
        m = SchemaMigration()

        # Run initial migration
        router = m._build_router()
        router.run('000_initial')

        # Run schema migration
        assert m.run() is True

Example #4

File: schema_tests.py Project: Danik1601/Plex-Trakt-Scrobbler

def test_database_upgrade():
    db = db_connect(':memory:', 'peewee')

    with DatabaseContext.use(db):
        m = SchemaMigration()

        # Run initial migration
        router = m._build_router()
        router.run('000_initial')

        # Run schema migration
        assert m.run() is True

Example #5

def test_database_corruption():
    db = db_connect(':memory:', 'peewee')

    with DatabaseContext.use(db):
        m = SchemaMigration()

        # Run initial migration
        router = m._build_router()
        router.run('000_initial')

        # Remove table from database (schema corruption)
        db.execute_sql('DROP TABLE "configuration.option"')

        # Ensure migration validation fails
        assert router.validate() is False

        # Run schema migration
        assert m.run() is True

Example #6

File: schema_tests.py Project: Danik1601/Plex-Trakt-Scrobbler

def test_database_corruption():
    db = db_connect(':memory:', 'peewee')

    with DatabaseContext.use(db):
        m = SchemaMigration()

        # Run initial migration
        router = m._build_router()
        router.run('000_initial')

        # Remove table from database (schema corruption)
        db.execute_sql('DROP TABLE "configuration.option"')

        # Ensure migration validation fails
        assert router.validate() is False

        # Run schema migration
        assert m.run() is True