def __get_source_table_entity(backup_entity):
source_table_entity = Table.get_table_from_backup(backup_entity)
if not source_table_entity:
error_message = "Backup ancestor doesn't exists: '{}:{}'. " \
.format(backup_entity.dataset_id,
backup_entity.table_id)
raise ParameterValidationException(error_message)
return source_table_entity
def test_should_retrieve_table_using_backup(self):
# given
table = Table(project_id='example-proj-name',
dataset_id='example-dataset-name',
table_id='example-table-name',
last_checked=datetime(2017, 02, 1, 16, 30))
table.put()
backup = Backup(parent=table.key,
last_modified=datetime(2017, 02, 1, 16, 30),
created=datetime(2017, 02, 1, 16, 30),
dataset_id='targetDatasetId',
table_id='targetTableId',
numBytes=1234)
backup.put()
# then
backup_entity = Backup.get_by_key(backup.key)
table_entity = Table.get_table_from_backup(backup_entity)
self.assertEqual(table_entity, table)
