Added tests for export data formats

elibixby · elibixby · commit d0bc45f2928c · 2015-06-30T13:01:57.000-07:00
diff --git a/bigquery/samples/export_data_to_cloud_storage.py b/bigquery/samples/export_data_to_cloud_storage.py
@@ -29,10 +29,10 @@ def export_table(service, cloud_storage_path,
         service: initialized and authorized bigquery
         google-api-client object,
         cloud_storage_path: fully qualified
-        path to a Google Cloud Storage location,
-        e.g. gs://mybucket/myfolder/
-        export_format: format to export in;
-        "CSV", "NEWLINE_DELIMITED_JSON", or "AVRO".
+            path to a Google Cloud Storage location,
+            e.g. gs://mybucket/myfolder/
+            export_format: format to export in;
+            "CSV", "NEWLINE_DELIMITED_JSON", or "AVRO".
 
     Returns: an extract job resource representing the
     job, see https://cloud.google.com/bigquery/docs/reference/v2/jobs
@@ -65,7 +65,7 @@ def export_table(service, cloud_storage_path,
 # [START run]
 def run(cloud_storage_path,
         projectId, datasetId, tableId,
-        num_retries, interval):
+        num_retries, interval, export_format="CSV"):
 
     bigquery = get_service()
     resource = export_table(bigquery, cloud_storage_path,
@@ -74,6 +74,7 @@ def run(cloud_storage_path,
              resource['jobReference']['projectId'],
              resource['jobReference']['jobId'],
              interval,
+             export_format,
              num_retries)
 # [END run]
 
diff --git a/bigquery/tests/test_export_data_to_cloud_storage.py b/bigquery/tests/test_export_data_to_cloud_storage.py
@@ -22,14 +22,32 @@
 
 class TestExportTableToGCS(CloudBaseTest):
 
-    def test_export_table(self):
+    def test_export_table_csv(self):
         run(self.constants['cloudStorageInputURI'],
             self.constants['projectId'],
             self.constants['datasetId'],
             self.constants['newTableId'],
             5,
-            5)
+            5,
+            export_format="CSV")
+
+    def test_export_table_json(self):
+        run(self.constants['cloudStorageInputURI'],
+            self.constants['projectId'],
+            self.constants['datasetId'],
+            self.constants['newTableId'],
+            5,
+            5,
+            export_format="NEWLINE_DELIMITED_JSON")
 
+    def test_export_table_avro(self):
+        run(self.constants['cloudStorageInputURI'],
+            self.constants['projectId'],
+            self.constants['datasetId'],
+            self.constants['newTableId'],
+            5,
+            5,
+            export_format="AVRO")
 
 if __name__ == '__main__':
     unittest.main()
