def setUp(self): self.client_mock = mock.Mock() self.client_mock.write_to_table.return_value = { 'jobReference': { 'jobId': mock.sentinel.job_id } } self.service = BigQueryService(self.client_mock)
class TestBigQueryService(unittest.TestCase): def setUp(self): self.client_mock = mock.Mock() self.service = BigQueryService(self.client_mock) def test_query_returns_a_biquery_result_object(self): self.client_mock.query.return_value = [None, None] result = self.service.query(mock.sentinel.sql) self.assertIsInstance(result, BigQueryResult)
class TestBigQueryDataset(unittest.TestCase): def setUp(self): self.client_mock = mock.Mock() self.client_mock.write_to_table.return_value = { 'jobReference': { 'jobId': mock.sentinel.job_id } } self.service = BigQueryService(self.client_mock) def test_write_to_table_returns_a_biquery_job_object(self): dataset = self.service.dataset(mock.sentinel.dataset) result = dataset.write_to_table( mock.sentinel.sql, mock.sentinel.table, mock.sentinel.write_disposition ) self.assertIsInstance(result, BigQueryJob) def test_write_to_truncated_table_returns_uses_job_write_truncate(self): dataset = self.service.dataset(mock.sentinel.dataset) dataset.write_to_truncated_table( mock.sentinel.sql, mock.sentinel.table ) self.client_mock.write_to_table.assert_called_with( mock.sentinel.sql, mock.sentinel.dataset, mock.sentinel.table, write_disposition=JOB_WRITE_TRUNCATE )
def setUp(self): self.client_mock = mock.Mock() self.service = BigQueryService(self.client_mock)