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())