def test_export_bigquery(self, mock_bq, mock_bq_2, mock_bq_3): with mock_bq, mock_bq_2, mock_bq_3, \ self.assertLogs(level='DEBUG') as log: mock_bq.return_value = [] export(self.data, self.exporters[2]) self.assertEqual(len(log.output), 5) self.assertEqual(len(log.records), 5)
def test_export_bigquery_error(self, mock_bq, mock_bq_2, mock_bq_3): with mock_bq, mock_bq_2, mock_bq_3, \ self.assertLogs(level='DEBUG') as log: mock_bq.return_value = self.load_fixture( filename=f'{cwd}/fixtures/bq_error.json', load_json=True) with self.assertRaises(BigQueryError): export(self.data, self.exporters[2]) self.assertEqual(len(log.output), 4) self.assertEqual(len(log.records), 4)
def test_export_stackdriver(self): with self.assertLogs(level='DEBUG') as log: channel = self.make_grpc_stub() patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel export(self.data, self.exporters[1]) self.assertEqual(len(log.output), 6) self.assertEqual(len(log.records), 6)
def test_export_bigquery(self, mock_bq, mock_bq_2, mock_bq_3): export(SLO_REPORT, EXPORTERS[2])
def test_export_stackdriver(self, mock): export(SLO_REPORT, EXPORTERS[1])
def test_export_pubsub(self, mock_pubsub, mock_pubsub_res): export(SLO_REPORT, EXPORTERS[0])
def test_export_deprecated(self, mock): with self.assertWarns(FutureWarning): export(SLO_REPORT, EXPORTERS[6])
def test_export_dynatrace_error(self, mock): responses = export(SLO_REPORT, EXPORTERS[5]) codes = [r[0]['response']['error']['code'] for r in responses] self.assertTrue(all(code == 429 for code in codes))
def test_export_dynatrace(self, mock): export(SLO_REPORT, EXPORTERS[5])
def test_export_multiple_error(self, *mocks): exporters = [EXPORTERS[1], EXPORTERS[2]] results = export(SLO_REPORT, exporters) self.assertTrue(isinstance(results[-1], BigQueryError))
def test_export_bigquery(self, *mocks): export(SLO_REPORT, EXPORTERS[2])
def test_export_pubsub(self, mock_pubsub, mock_pubsub_res): with mock_pubsub, mock_pubsub_res, \ self.assertLogs(level='DEBUG') as log: export(self.data, self.exporters[0]) self.assertEqual(len(log.output), 5) self.assertEqual(len(log.records), 5)
def main(data, context): data = json.loads(base64.b64decode(data['data'])) compute.export(data, exporters)
def test_export_multiple_error_raise(self, *mocks): exporters = [EXPORTERS[1], EXPORTERS[2]] with self.assertRaises(BigQueryError): export(SLO_REPORT, exporters, raise_on_error=True)
def test_export_bigquery_error(self, mock_bq, mock_bq_2, mock_bq_3): with self.assertRaises(BigQueryError): export(SLO_REPORT, EXPORTERS[2])
def test_export_datadog(self): export(SLO_REPORT, EXPORTERS[4])
def test_export_prometheus(self, mock): export(SLO_REPORT, EXPORTERS[3])
def test_export_bigquery_error(self, *mocks): with self.assertRaises(BigQueryError): export(SLO_REPORT, EXPORTERS[2], raise_on_error=True)