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