def testInvalidDatabase(self):
conf = Config()
conf.fromMap(testConfig)
creator = MigrationCreator('asdf', locationInTestspace())
target = creator.createMigration()
migrator = MigrationApplier(locationInTestspace(), conf)
self.assertRaises(BaseException, migrator.applyMigration, target)
def testCreatesTable(self):
migrator = MigrationCreator('zyxw', locationInTestspace())
target = migrator.createMigration(args=['tableName','column:type','column2:type2'])
self.assertFileExists('zyxw', target, 'up')
wholeFile = ''
with open(locationInTestspace('zyxw', target, 'up'), 'r') as f:
wholeFile = f.read()
fileContent = """create table tableName (\n column type,\n column2 type2\n );\n"""
self.assertEqual(fileContent, wholeFile)
def testPipesAdvancedOutput(self):
command = "create table xxx (yyy integer primary key);"
creator = MigrationCreator("migration_test", locationInTestspace())
target = creator.createMigration(version=42, advanced=True, body="#!/bin/bash\necho -n '" + command + "' ")
conf = Config()
conf.fromMap(testConfig)
migrator = MigrationApplier(locationInTestspace(), conf)
migrator.plugin = FakeDatabasePlugin(self)
migrator.applyMigrations([target])
migrator.plugin.assertCurrentVersion(42)
migrator.plugin.assertCommandWasExecuted(command)
def testAdvancedMigration(self):
creator = MigrationCreator("migration_test", locationInTestspace())
target = creator.createMigration(
version=42, advanced=True, body="""#!/bin/bash\necho Hello World > testspace/test_output\n"""
)
conf = Config()
conf.fromMap(testConfig)
migrator = MigrationApplier(locationInTestspace(), conf)
migrator.plugin = FakeDatabasePlugin(self)
migrator.applyMigrations([target])
migrator.plugin.assertCurrentVersion(42)
self.assertFileExists("test_output")
def testEmptyUpFile(self):
creator = MigrationCreator("migration_test", locationInTestspace())
target1 = creator.createMigration(version=17, body="""select * from __mig_version__""")
target2 = creator.createMigration(version=23, body="""""")
conf = Config()
conf.fromMap(testConfig)
migrator = MigrationApplier(locationInTestspace(), conf)
migrator.plugin = FakeDatabasePlugin(self)
self.assertRaisesRegexp(
RuntimeError, "Invalid migration: Up file is empty", migrator.applyMigrations, ([target1, target2])
)
migrator.plugin.assertCurrentVersion(17)
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 testAppliesOutOfOrderMigrations(self):
creator = MigrationCreator("migration_test", locationInTestspace())
target1 = creator.createMigration(version=23, body="select 1")
conf = Config()
conf.fromMap(testConfig)
migrator = MigrationApplier(locationInTestspace(), conf)
migrator.plugin = FakeMultiPlugin(self)
migrator.applySingleMigration(target1)
target2 = creator.createMigration(version=17, body="select 2")
migrator.applyMigrations([target1, target2])
migrator.plugin.assertCurrentVersion(23)
migrator.plugin.assertCommandWasExecuted("select 1")
migrator.plugin.assertCommandWasExecuted("select 2")
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()
def testAppliesNoopMigrations(self):
creator = MigrationCreator("migration_test", locationInTestspace())
target1 = creator.createMigration(version=23, body="select 1")
conf = Config()
conf.fromMap(testConfig)
migrator = MigrationApplier(locationInTestspace(), conf)
migrator.plugin = FakeMultiPlugin(self)
migrator.applyMigrations([target1])
migrator.plugin.assertCurrentVersion(23)
migrator.plugin.assertCommandWasExecuted("select 1")
migrator.dry_run = True
migrator.applyMigrations([creator.createMigration(version=27, body="select 2")])
migrator.plugin.assertCurrentVersion(23)
migrator.plugin.assertCommandWasNotExecuted("select 2")
def testTwoMigrationsTogether(self):
command0 = "create table xxx (yyy integer primary key);"
command1 = "create table aaa (bbb integer primary key);"
creator = MigrationCreator("migration_test", locationInTestspace())
targets = []
targets.append(creator.createMigration(version=1, body=command0))
targets.append(creator.createMigration(version=2, body=command1))
conf = Config()
conf.fromMap(testConfig)
migrator = MigrationApplier(locationInTestspace(), conf)
migrator.plugin = FakeDatabasePlugin(self)
migrator.applyMigrations(targets)
migrator.plugin.assertCurrentVersion(2)
migrator.plugin.assertCommandWasExecuted(command0)
migrator.plugin.assertCommandWasExecuted(command1)
def testSecondMigrationFails(self):
command0 = "create table xxx (yyy integer primary key);"
command1 = "alter blah blah blah;"
creator = MigrationCreator("migration_test", locationInTestspace())
targets = []
targets.append(creator.createMigration(version=1, body=command0))
targets.append(creator.createMigration(version=2, body=command1))
conf = Config()
conf.fromMap(testConfig)
migrator = MigrationApplier(locationInTestspace(), conf)
migrator.plugin = FakeDatabasePlugin(self, failOn=command1)
try:
migrator.applyMigrations(targets)
except: # Error is expected
pass
migrator.plugin.assertCurrentVersion(1)
migrator.plugin.assertCommandWasExecuted(command0)
migrator.plugin.assertCommandWasNotExecuted(command1)
def testCreateMigration(self):
migrator = MigrationCreator("zyxw", locationInTestspace())
target = migrator.createMigration()
self.assertTrue(os.path.exists(locationInTestspace('zyxw', target)))
self.assertTrue(os.path.exists(locationInTestspace('zyxw', target, 'up')))
def testSpaceExists(self):
self.assertTrue(os.path.exists(locationInTestspace()))
def testCreateSpecificMigration(self):
migrator = MigrationCreator("zyxw", locationInTestspace())
name = migrator.createMigration(version='abcdef')
self.assertEquals('abcdef', name)
target = locationInTestspace('zyxw', 'abcdef', 'up')
self.assertEquals('testspace/zyxw/abcdef/up', target)
def testCreateMigrationWithBody(self):
migrator = MigrationCreator("zyxw", locationInTestspace())
body = "This is the body of the file."
target = migrator.createMigration(body=body)
with open(locationInTestspace('zyxw', target, 'up'), 'r') as f:
self.assertEquals(body + "\n", f.readline())
def testCreateMigrationWithHeader(self):
migrator = MigrationCreator("zyxw", locationInTestspace())
header = "This is the header of the file."
target = migrator.createMigration(header=header)
with open(locationInTestspace('zyxw', target, 'up'), 'r') as f:
self.assertEquals("-- " + header + "\n", f.readline())
def testAdvancedMigration(self):
migrator = MigrationCreator("zyxw", locationInTestspace())
name = migrator.createMigration(advanced=True)
target = locationInTestspace('zyxw', name, 'up')
self.assertExecutable(target)
