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)
