def test_any_migrations_in_progress(self):
self.assertFalse(any_migrations_in_progress('purple'))
set_migration_started('purple', self.slug)
self.assertTrue(any_migrations_in_progress('purple'))
set_migration_started('purple', 'other_slug')
set_migration_not_started('purple', self.slug)
self.assertTrue(any_migrations_in_progress('purple'))
set_migration_complete('purple', 'other_slug')
self.assertFalse(any_migrations_in_progress('purple'))
def test_abort(self):
set_migration_started('yellow', self.slug)
self.assertFalse(get_migration_complete('yellow', self.slug))
self.assertEqual(get_migration_status('yellow', self.slug),
MigrationStatus.IN_PROGRESS)
set_migration_not_started('yellow', self.slug)
self.assertFalse(get_migration_complete('yellow', self.slug))
self.assertEqual(get_migration_status('yellow', self.slug),
MigrationStatus.NOT_STARTED)
def set_couch_sql_migration_not_started(domain):
set_migration_not_started(domain, COUCH_TO_SQL_SLUG)
def set_tz_migration_not_started(domain):
return set_migration_not_started(domain, TZMIGRATION_SLUG)
