def test_that_table_backup_is_scheduled_for_not_partitioned_table(
self, backup_start, _, _1):
# given
table_reference = TableReference(project_id="test-project",
dataset_id="test-dataset",
table_id="test-table",
partition_id=None)
# when
TableBackup.start(table_reference)
# then
backup_start.assert_called_once()
def test_that_backup_are_scheduled_for_non_empty_single_partition(
self, _, _1, _2, _3, backup_start):
# given
table_reference = TableReference(project_id="test-project",
dataset_id="test-dataset",
table_id="test-table",
partition_id="20170303")
# when
TableBackup.start(table_reference)
# then
backup_start.assert_called_once()
def test_that_backup_for_partitions_is_scheduled_for_empty_partitioned_table(
self, table_partitions_backup_scheduler, _, _1, _2, _3):
# given
table_reference = TableReference(project_id="test-project",
dataset_id="test-dataset",
table_id="test-table",
partition_id=None)
# when
TableBackup.start(table_reference)
# then
table_partitions_backup_scheduler.assert_called_once()
def test_that_backup_for_partitions_is_scheduled_for_partitioned_table(
self, _, _1, _2, _3, _4):
# given
table_reference = TableReference(project_id="test-project",
dataset_id="test-dataset",
table_id="test-table",
partition_id=None)
# when
TableBackup.start(table_reference)
# then
tasks = self.task_queue_stub.get_filtered_tasks(
queue_names='backup-scheduler')
self.assertEqual(len(tasks), 1)
self.assertEqual(
tasks[0].payload,
'projectId=test-project&tableId=test-table&datasetId=test-dataset')
def get(self, project_id, dataset_id, table_id, partition_id=None): # nopep8 pylint: disable=R0201
table_reference = TableReference(project_id, dataset_id,
urllib2.unquote(table_id), partition_id)
TableBackup.start(table_reference)