def test_handler(self):
http = HttpMockSequence([
({
'status': '200'
}, self.bigquery_discovery),
({
'status': '200'
}, self.bigquery_running),
({
'status': '200'
}, self.bigquery_done),
])
gs = build("bigquery", "v2", http=http)
with patch("clumpy.google.service", gs):
load_bigquery.handler(self.clean_data, FakeContext())
def test_missing_dataset(self):
data = copy(self.clean_data)
data["queryStringParameters"].pop("datasetId")
with self.assertRaises(KeyError):
ret = load_bigquery.handler(data, FakeContext())
def test_invalid_fields(self):
data = copy(self.clean_data)
data["queryStringParameters"]["fields"] = "klsjdf"
with self.assertRaises(TypeError):
ret = load_bigquery.handler(data, FakeContext())