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])