feat: add destination_expiration_time property to copy job (#1277)

* feat: add destination_expiration_time property to copy job * update test * refactor test * remove unused import * Update google/cloud/bigquery/job/copy_.py Co-authored-by: Anthonios Partheniou <partheniou@google.com> * Update google/cloud/bigquery/job/copy_.py Co-authored-by: Anthonios Partheniou <partheniou@google.com> Co-authored-by: Anthonios Partheniou <partheniou@google.com>
googleapis · Jun 17, 2022 · 728b07c · 728b07c
1 parent 51c1f37
commit 728b07c
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 0 deletions.
diff --git a/google/cloud/bigquery/job/copy_.py b/google/cloud/bigquery/job/copy_.py
@@ -126,6 +126,20 @@ def operation_type(self, value: Optional[str]):
             value = OperationType.OPERATION_TYPE_UNSPECIFIED
         self._set_sub_prop("operationType", value)
 
+    @property
+    def destination_expiration_time(self) -> str:
+        """google.cloud.bigquery.job.DestinationExpirationTime: The time when the
+        destination table expires. Expired tables will be deleted and their storage reclaimed.
+
+        See
+        https://cloud.google.com/bigquery/docs/reference/rest/v2/Job#JobConfigurationTableCopy.FIELDS.destination_expiration_time
+        """
+        return self._get_sub_prop("destinationExpirationTime")
+
+    @destination_expiration_time.setter
+    def destination_expiration_time(self, value: str):
+        self._set_sub_prop("destinationExpirationTime", value)
+
 
 class CopyJob(_AsyncJob):
     """Asynchronous job: copy data into a table from other tables.

diff --git a/tests/system/test_client.py b/tests/system/test_client.py
@@ -2153,6 +2153,11 @@ def test_table_snapshots(dataset_id):
     copy_config = CopyJobConfig()
     copy_config.operation_type = OperationType.SNAPSHOT
 
+    today = datetime.date.today()
+    destination_expiration_time = f"{today.year + 1}-01-01T00:00:00Z"
+
+    copy_config.destination_expiration_time = destination_expiration_time
+
     copy_job = client.copy_table(
         sources=source_table_path,
         destination=snapshot_table_path,

diff --git a/tests/unit/job/test_copy.py b/tests/unit/job/test_copy.py
@@ -19,6 +19,8 @@
 from .helpers import _Base
 from .helpers import _make_client
 
+import datetime
+
 
 class TestCopyJobConfig(_Base):
     JOB_TYPE = "copy"
@@ -36,6 +38,7 @@ def test_ctor_defaults(self):
 
         assert config.create_disposition is None
         assert config.write_disposition is None
+        assert config.destination_expiration_time is None
         assert config.destination_encryption_configuration is None
         assert config.operation_type == OperationType.OPERATION_TYPE_UNSPECIFIED
 
@@ -48,15 +51,22 @@ def test_ctor_w_properties(self):
         write_disposition = WriteDisposition.WRITE_TRUNCATE
         snapshot_operation = OperationType.SNAPSHOT
 
+        today = datetime.date.today()
+        destination_expiration_time = f"{today.year + 1}-01-01T00:00:00Z"
+
         config = self._get_target_class()(
             create_disposition=create_disposition,
             write_disposition=write_disposition,
             operation_type=snapshot_operation,
+            destination_expiration_time=destination_expiration_time,
         )
 
         self.assertEqual(config.create_disposition, create_disposition)
         self.assertEqual(config.write_disposition, write_disposition)
         self.assertEqual(config.operation_type, snapshot_operation)
+        self.assertEqual(
+            config.destination_expiration_time, destination_expiration_time
+        )
 
     def test_to_api_repr_with_encryption(self):
         from google.cloud.bigquery.encryption_configuration import (