Add job factory: 'Client.extract_table_to_storage'.

tseaver · tseaver · commit 4bcaaf540b76 · 2015-08-17T23:19:36.000-04:00
diff --git a/gcloud/bigquery/client.py b/gcloud/bigquery/client.py
@@ -19,6 +19,7 @@
 from gcloud.bigquery.connection import Connection
 from gcloud.bigquery.dataset import Dataset
 from gcloud.bigquery.job import CopyJob
+from gcloud.bigquery.job import ExtractTableToStorageJob
 from gcloud.bigquery.job import LoadTableFromStorageJob
 
 
@@ -133,3 +134,22 @@ def copy_table(self, name, destination, *sources):
         :returns: a new ``CopyJob`` instance
         """
         return CopyJob(name, destination, sources, client=self)
+
+    def extract_table_to_storage(self, name, source, *destination_uris):
+        """Construct a job for copying one or more tables into another table.
+
+        :type name: string
+        :param name: Name of the job.
+
+        :type source: :class:`gcloud.bigquery.table.Table`
+        :param source: table to be extracted.
+
+        :type destination_uris: sequence of string
+        :param destination_uris: URIs of CloudStorage file(s) into which
+                                 table data is to be extracted.
+
+        :rtype: :class:`gcloud.bigquery.job.ExtractTableToStorageJob`
+        :returns: a new ``ExtractTableToStorageJob`` instance
+        """
+        return ExtractTableToStorageJob(name, source, destination_uris,
+                                        client=self)
diff --git a/gcloud/bigquery/test_client.py b/gcloud/bigquery/test_client.py
@@ -167,6 +167,25 @@ def test_copy_table(self):
         self.assertEqual(list(job.sources), [source])
         self.assertTrue(job.destination is destination)
 
+    def test_extract_table_to_storage(self):
+        from gcloud.bigquery.job import ExtractTableToStorageJob
+        PROJECT = 'PROJECT'
+        JOB = 'job_name'
+        DATASET = 'dataset_name'
+        SOURCE = 'source_table'
+        DESTINATION = 'gs://bucket_name/object_name'
+        creds = _Credentials()
+        http = object()
+        client = self._makeOne(project=PROJECT, credentials=creds, http=http)
+        dataset = client.dataset(DATASET)
+        source = dataset.table(SOURCE)
+        job = client.extract_table_to_storage(JOB, source, DESTINATION)
+        self.assertTrue(isinstance(job, ExtractTableToStorageJob))
+        self.assertTrue(job._client is client)
+        self.assertEqual(job.name, JOB)
+        self.assertEqual(job.source, source)
+        self.assertEqual(list(job.destination_uris), [DESTINATION])
+
 
 class _Credentials(object):
 
