def test_in_progress(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) self.assertEqual(get_migration_status('yellow', 'otherslug'), MigrationStatus.NOT_STARTED)
def test_dry_run(self): set_migration_started('yellow', self.slug, True) self.assertFalse(get_migration_complete('yellow', self.slug)) self.assertFalse(migration_in_progress('yellow', self.slug)) self.assertTrue(migration_in_progress('yellow', self.slug, True)) self.assertEqual(get_migration_status('yellow', self.slug), MigrationStatus.DRY_RUN) self.assertEqual(get_migration_status('yellow', 'otherslug'), MigrationStatus.NOT_STARTED)
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 test_dry_run(self): set_migration_started('yellow', self.slug, dry_run=True) self.assertFalse(get_migration_complete('yellow', self.slug)) self.assertFalse(migration_in_progress('yellow', self.slug)) self.assertTrue(migration_in_progress('yellow', self.slug, True)) self.assertEqual(get_migration_status('yellow', self.slug), MigrationStatus.DRY_RUN) self.assertEqual(get_migration_status('yellow', 'otherslug'), MigrationStatus.NOT_STARTED) progress = self.get_progress('yellow') self.assertIsNotNone(progress.started_on) self.assertLessEqual(progress.started_on, datetime.utcnow())
def test_complete(self): set_migration_complete('green', self.slug) self.assertEqual(get_migration_status('green', self.slug), MigrationStatus.COMPLETE) self.assertTrue(get_migration_complete('green', self.slug)) self.assertFalse(get_migration_complete('green', 'otherslug')) progress = self.get_progress('green') self.assertIsNotNone(progress.completed_on) self.assertLessEqual(progress.completed_on, datetime.utcnow())
def get_tz_migration_status(domain, strict=False): return get_migration_status(domain, TZMIGRATION_SLUG, strict=strict)
def test_not_started(self): self.assertFalse(get_migration_complete('red', self.slug)) self.assertEqual(get_migration_status('red', self.slug), MigrationStatus.NOT_STARTED)
def test_complete(self): set_migration_complete('green', self.slug) self.assertEqual(get_migration_status('green', self.slug), MigrationStatus.COMPLETE) self.assertTrue(get_migration_complete('green', self.slug)) self.assertFalse(get_migration_complete('green', 'otherslug'))
def tzmigration_status(request): if request.domain: return get_migration_status(request.domain, TZMIGRATION_SLUG) else: return None
def couch_sql_migration_in_progress(domain): return get_migration_status(domain, COUCH_TO_SQL_SLUG, strict=True) == MigrationStatus.IN_PROGRESS
def get_couch_sql_migration_status(domain): return get_migration_status(domain, COUCH_TO_SQL_SLUG)