Add test for expiration_time option

hagino3000 · hagino3000 · commit 958a54d3168d · 2015-07-21T16:21:37.000+09:00
diff --git a/bigquery/tests/test_client.py b/bigquery/tests/test_client.py
@@ -1482,6 +1482,7 @@ def setUp(self):
                 'tableId': self.table, 'projectId': self.project,
                 'datasetId': self.dataset}
         }
+        self.expiration_time = 1437513693000
 
     def test_table_create_failed(self):
         """Ensure that if creating the table fails, False is returned,
@@ -1535,6 +1536,26 @@ def test_table_create_success(self):
 
         self.mock_tables.insert.return_value.execute.assert_called_with()
 
+    def test_table_create_body_with_expiration_time(self):
+        """Ensure that if expiration_time has specified,
+        it passed to the body."""
+
+        self.mock_tables.insert.return_value.execute.side_effect = [{
+            'status': 'foo'}, {'status': 'bar'}]
+
+        actual = self.client.create_table(self.dataset, self.table,
+                                          self.schema, self.expiration_time)
+
+        body = self.body.copy()
+        body.update({
+            'expirationTime': self.expiration_time
+        })
+
+        self.mock_tables.insert.assert_called_with(
+            projectId=self.project, datasetId=self.dataset, body=body)
+
+        self.mock_tables.insert.return_value.execute.assert_called_with()
+
 
 class TestCreateView(unittest.TestCase):
 
