Merge 0388706 into c809f68

ocadotechnology · Dec 5, 2018 · 0df9d71 · 0df9d71
2 parents c809f68 + 0388706
commit 0df9d71
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 7 deletions.
diff --git a/src/restore/dataset/dataset_restore_items_generator.py b/src/restore/dataset/dataset_restore_items_generator.py
@@ -34,7 +34,6 @@ def generate_restore_items(cls, project_id, dataset_id, target_project_id,
                         target_project_id,
                         target_dataset_id,
                         table_entity.table_id,
-                        table_entity.partition_id
                     )
 
                     restore_item = RestoreItem.create(source_table_reference,

diff --git a/src/restore/list/backup_list_restore_service.py b/src/restore/list/backup_list_restore_service.py
@@ -141,5 +141,4 @@ def __create_target_table_reference(restore_request, source_entity):
             target_dataset_id = source_entity.dataset_id
         return TableReference(target_project_id,
                               target_dataset_id,
-                              source_entity.table_id,
-                              source_entity.partition_id)
+                              source_entity.table_id)
diff --git a/tests/restore/dataset/test_dataset_restore_items_generator.py b/tests/restore/dataset/test_dataset_restore_items_generator.py
@@ -260,8 +260,7 @@ def __generate_expected_restore_item(
             custom_target_dataset)
         expected_target = TableReference(project_id=target_project_id,
                                          dataset_id=target_dataset,
-                                         table_id=table.table_id,
-                                         partition_id=table.partition_id)
+                                         table_id=table.table_id)
         expected_restore_item = RestoreItem.create(expected_source,
                                                    expected_target)
         return expected_restore_item

diff --git a/tests/restore/list/test_backup_list_restore_service.py b/tests/restore/list/test_backup_list_restore_service.py
@@ -66,7 +66,6 @@ def test_that_restore_service_will_receive_suitable_request(
             "target_project_id",
             "target_dataset_id",
             "source_table_id",
-            "source_partition_id"
         )
         expected_restore_item = RestoreItem.create(
             expected_source_table_reference,
@@ -116,7 +115,6 @@ def test_that_restore_service_will_generate_default_values_if_missing(
             "source_project_id",
             "source_dataset_id",
             "source_table_id",
-            "source_partition_id"
         )
         expected_restore_item = RestoreItem.create(
             expected_source_table_reference, expected_target_table_reference