def test_phase_complete(self):
v = pygration.migration.VersionNumber("v1-5")
state = pygration.db.PygrationState(str(v)
, MockStep.step_id, MockStep.step_name)
mig = StepMigrator(v, MockStep(), state)
self.assertFalse(mig.phase_complete('add'))
self.assertFalse(mig.phase_complete('simdrop'))
self.assertFalse(mig.phase_complete('drop'))
def test_normal_add_pass(self):
v = pygration.migration.VersionNumber("v1-5")
state = pygration.db.PygrationState(str(v)
, MockStep.step_id, MockStep.step_name)
mig = StepMigrator(v, MockStep(), state)
db = MockDB("pass")
result = mig.migrate(db, "add")
self.assertEqual("P", result.add_state)
def test_phase_complete(self):
v = pygration.migration.VersionNumber("v1-5")
state = pygration.db.PygrationState(str(v), MockStep.step_id,
MockStep.step_name)
mig = StepMigrator(v, MockStep(), state)
self.assertFalse(mig.phase_complete('add'))
self.assertFalse(mig.phase_complete('simdrop'))
self.assertFalse(mig.phase_complete('drop'))
def test_normal_add_pass(self):
v = pygration.migration.VersionNumber("v1-5")
state = pygration.db.PygrationState(str(v), MockStep.step_id,
MockStep.step_name)
mig = StepMigrator(v, MockStep(), state)
db = MockDB("pass")
result = mig.migrate(db, "add")
self.assertEqual("P", result.add_state)
def test_normal_rollback_add(self):
v = pygration.migration.VersionNumber("v1-5")
state = pygration.db.PygrationState(str(v)
, MockStep.step_id, MockStep.step_name)
mig = StepMigrator(v, MockStep(), state)
db = MockDB("pass")
result = mig.rollback(db, "add")
self.assertEqual("RB", mig._state.add_state)
self.assertEqual(2, len(db.command))
self.assertEqual("drop mock;", db.command[0])
self.assertEqual("commit()", db.command[1])
def test_normal_rollback_add(self):
v = pygration.migration.VersionNumber("v1-5")
state = pygration.db.PygrationState(str(v), MockStep.step_id,
MockStep.step_name)
mig = StepMigrator(v, MockStep(), state)
db = MockDB("pass")
result = mig.rollback(db, "add")
self.assertEqual("RB", mig._state.add_state)
self.assertEqual(2, len(db.command))
self.assertEqual("drop mock;", db.command[0])
self.assertEqual("commit()", db.command[1])
