adding unittest and following pep8 standards

cbhimavarapu-datalicious · cbhimavarapu-datalicious · commit d1d850d68af3 · 2016-08-03T11:48:01.000+05:30
diff --git a/bigquery/client.py b/bigquery/client.py
@@ -1079,9 +1079,14 @@ def write_to_table(
         write_disposition : str, optional
             One of the JOB_WRITE_* constants
         use_legacy_sql: bool, optional
-            If False, the query will use BigQuery's standard SQL (https://cloud.google.com/bigquery/sql-reference/)
+            If False, the query will use BigQuery's standard SQL
+            (https://cloud.google.com/bigquery/sql-reference/)
         maximum_billing_tier : integer, optional
-            Limits the billing tier for this job. Queries that have resource usage beyond this tier will fail (without incurring a charge). If unspecified, this will be set to your project default. For more information, see https://cloud.google.com/bigquery/pricing#high-compute
+            Limits the billing tier for this job. Queries that have resource
+            usage beyond this tier will fail (without incurring a charge). If
+            unspecified, this will be set to your project default. For more
+            information,
+            see https://cloud.google.com/bigquery/pricing#high-compute
 
         Returns
         -------
diff --git a/bigquery/tests/test_client.py b/bigquery/tests/test_client.py
@@ -1117,6 +1117,7 @@ def setUp(self):
         self.project_id = 'project'
         self.dataset_id = 'dataset'
         self.table_id = 'table'
+        self.maximum_billing_tier = 1000
         self.external_udf_uris = ['gs://bucket/external_udf.js']
         self.use_query_cache = False
         self.priority = "INTERACTIVE"
@@ -1162,6 +1163,44 @@ def test_write(self):
 
         self.assertEqual(result, expected_result)
 
+    def test_write_maxbilltier(self):
+        """ Ensure that write is working when maximumBillingTier is set"""
+        expected_result = {
+            'status': {'state': u'RUNNING'},
+        }
+
+        body = {
+            "configuration": {
+                "query": {
+                    "destinationTable": {
+                        "projectId": self.project_id,
+                        "datasetId": self.dataset_id,
+                        "tableId": self.table_id
+                    },
+                    "query": self.query,
+                    "userDefinedFunctionResources": [{
+                        "resourceUri": self.external_udf_uris[0]
+                    }],
+                    "useQueryCache": self.use_query_cache,
+                    "priority": self.priority,
+                    "maximumBillingTier": self.maximum_billing_tier
+                }
+            }
+        }
+
+        self.mock_api.jobs().insert().execute.return_value = expected_result
+        result = self.client.write_to_table(
+            self.query, self.dataset_id, self.table_id, priority=self.priority,
+            external_udf_uris=self.external_udf_uris, use_query_cache=False,
+            maximum_billing_tier=self.maximum_billing_tier)
+
+        self.mock_api.jobs().insert.assert_called_with(
+            projectId=self.project_id,
+            body=body
+        )
+
+        self.assertEqual(result, expected_result)
+
     def test_write_http_error(self):
         """ Test write with http error"""
         expected_result = {
