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)