Add 'Client.copy_table' factory.

tseaver · tseaver · commit 16dba9dcf3bc · 2015-08-17T18:35:19.000-04:00
diff --git a/gcloud/bigquery/client.py b/gcloud/bigquery/client.py
@@ -18,6 +18,7 @@
 from gcloud.client import JSONClient
 from gcloud.bigquery.connection import Connection
 from gcloud.bigquery.dataset import Dataset
+from gcloud.bigquery.job import CopyJob
 from gcloud.bigquery.job import LoadTableFromStorageJob
 
 
@@ -115,3 +116,20 @@ def load_table_from_storage(self, name, destination, *source_uris):
         """
         return LoadTableFromStorageJob(name, destination, source_uris,
                                        client=self)
+
+    def copy_table(self, name, destination, *sources):
+        """Construct a job for copying one or more tables into another table.
+
+        :type name: string
+        :param name: Name of the job.
+
+        :type destination: :class:`gcloud.bigquery.table.Table`
+        :param destination: Table into which data is to be copied.
+
+        :type sources: sequence of :class:`gcloud.bigquery.table.Table`
+        :param sources: tables to be copied.
+
+        :rtype: :class:`gcloud.bigquery.job.CopyJob`
+        :returns: a new ``CopyJob`` instance
+        """
+        return CopyJob(name, destination, sources, client=self)
diff --git a/gcloud/bigquery/test_client.py b/gcloud/bigquery/test_client.py
@@ -147,6 +147,26 @@ def test_load_table_from_storage(self):
         self.assertEqual(list(job.source_uris), [SOURCE_URI])
         self.assertTrue(job.destination is destination)
 
+    def test_copy_table(self):
+        from gcloud.bigquery.job import CopyJob
+        PROJECT = 'PROJECT'
+        JOB = 'job_name'
+        DATASET = 'dataset_name'
+        SOURCE = 'source_table'
+        DESTINATION = 'destination_table'
+        creds = _Credentials()
+        http = object()
+        client = self._makeOne(project=PROJECT, credentials=creds, http=http)
+        dataset = client.dataset(DATASET)
+        source = dataset.table(SOURCE)
+        destination = dataset.table(DESTINATION)
+        job = client.copy_table(JOB, destination, source)
+        self.assertTrue(isinstance(job, CopyJob))
+        self.assertTrue(job._client is client)
+        self.assertEqual(job.name, JOB)
+        self.assertEqual(list(job.sources), [source])
+        self.assertTrue(job.destination is destination)
+
 
 class _Credentials(object):
 
