Merge pull request tylertreat#1 from tylertreat/add_table_expiration_time

Takashi Nishibayashi · Takashi Nishibayashi · commit 795404f6baa5 · 2015-07-22T10:30:28.000+09:00
Fix unit tests
diff --git a/bigquery/tests/test_client.py b/bigquery/tests/test_client.py
@@ -65,7 +65,7 @@ def test_initialize_readonly(self, mock_build, mock_return_cred):
         mock_return_cred.assert_called_once_with()
         mock_cred.assert_called_once_with(service_account, key,
                                           scope=BIGQUERY_SCOPE_READ_ONLY)
-        mock_cred.authorize.assert_called_once()
+        self.assertTrue(mock_cred.return_value.authorize.called)
         mock_build.assert_called_once_with('bigquery', 'v2', http=mock_http)
         self.assertEquals(mock_bq, bq_client.bigquery)
         self.assertEquals(project_id, bq_client.project_id)
@@ -95,7 +95,7 @@ def test_initialize_read_write(self, mock_build, mock_return_cred):
         mock_return_cred.assert_called_once_with()
         mock_cred.assert_called_once_with(service_account, key,
                                           scope=BIGQUERY_SCOPE)
-        mock_cred.authorize.assert_called_once()
+        self.assertTrue(mock_cred.return_value.authorize.called)
         mock_build.assert_called_once_with('bigquery', 'v2', http=mock_http)
         self.assertEquals(mock_bq, bq_client.bigquery)
         self.assertEquals(project_id, bq_client.project_id)
@@ -130,7 +130,7 @@ def test_initialize_key_file(self, mock_open, mock_build,
         mock_return_cred.assert_called_once_with()
         mock_cred.assert_called_once_with(service_account, key,
                                           scope=BIGQUERY_SCOPE)
-        mock_cred.authorize.assert_called_once()
+        self.assertTrue(mock_cred.return_value.authorize.called)
         mock_build.assert_called_once_with('bigquery', 'v2', http=mock_http)
         self.assertEquals(mock_bq, bq_client.bigquery)
         self.assertEquals(project_id, bq_client.project_id)
@@ -378,7 +378,7 @@ def test_get_response(self):
             projectId=self.project_id, jobId=job_id, startIndex=offset,
             maxResults=limit, pageToken=page_token, timeoutMs=1000)
 
-        mock_query_job.execute.assert_called_once()
+        mock_query_job.execute.assert_called_once_with()
         self.assertEquals(actual, mock_query_reply)
 
 
