def test_get_table_migration_status_only_destination(self):
with self.assertRaises(shinkansen.UnrecoverableError):
status.get_table_migration_status(None, 'dst', 42)
def test_get_table_migration_status_no_partition(self):
with self.assertRaises(shinkansen.UnrecoverableError):
status.get_table_migration_status('src', 'dst', None)
def test_get_table_migration_status_only_source(self):
with self.assertRaises(shinkansen.UnrecoverableError):
status.get_table_migration_status('src', None, 42)
def _test_get_table_migration_status(self, table_status_recs, chunk_status_recs, expected):
self._test_get_migration_status_base(table_status_recs, chunk_status_recs)
result = status.get_table_migration_status('src', 'dst', 43)
self.assertEqual(result, expected)
