def testDirectAppy(self): command = "create table xxx (yyy integer primary key);" creator = MigrationCreator("migration_test", locationInTestspace()) target = creator.createMigration() self.assertFolderExists("migration_test", target) writeToFile(locationInTestspace("migration_test", target, "up"), command) conf = Config() conf.fromMap(testConfig) migrator = MigrationApplier(locationInTestspace(), conf) migrator.plugin = FakeDatabasePlugin(self) migrator.applySingleMigration(target) migrator.plugin.assertCommandWasExecuted(command)
def testRollback(self): command = "create table xxx (yyy integer primary key); alter blah blah blah;" target = MigrationCreator("migration_test", locationInTestspace()).createMigration() self.assertFolderExists("migration_test", target) writeToFile(locationInTestspace("migration_test", target, "up"), command) conf = Config() conf.fromMap(testConfig) migrator = MigrationApplier(locationInTestspace(), conf) migrator.plugin = FakeDatabasePlugin(self) try: migrator.applyMigration(target) except: pass migrator.plugin.assertNoCommandWasExecuted()