def test_abort(self): set_migration_started('yellow') self.assertFalse(get_migration_complete('yellow')) self.assertEqual(get_migration_status('yellow'), MigrationStatus.IN_PROGRESS) set_migration_not_started('yellow') self.assertFalse(get_migration_complete('yellow')) self.assertEqual(get_migration_status('yellow'), MigrationStatus.NOT_STARTED)
def test_not_started(self): self.assertFalse(get_migration_complete('red')) self.assertEqual(get_migration_status('red'), MigrationStatus.NOT_STARTED)
def test_complete(self): set_migration_complete('green') self.assertEqual(get_migration_status('green'), MigrationStatus.COMPLETE) self.assertTrue(get_migration_complete('green'))
def test_in_progress(self): set_migration_started('yellow') self.assertFalse(get_migration_complete('yellow')) self.assertEqual(get_migration_status('yellow'), MigrationStatus.IN_PROGRESS)