| Index: infra_libs/ts_mon/test/api_monitor_test.py
|
| diff --git a/infra_libs/ts_mon/test/api_monitor_test.py b/infra_libs/ts_mon/test/api_monitor_test.py
|
| deleted file mode 100644
|
| index 8309eeaf2013696a1c919f8e12399744259ffe7f..0000000000000000000000000000000000000000
|
| --- a/infra_libs/ts_mon/test/api_monitor_test.py
|
| +++ /dev/null
|
| @@ -1,76 +0,0 @@
|
| -# Copyright 2015 The Chromium Authors. All rights reserved.
|
| -# Use of this source code is governed by a BSD-style license that can be
|
| -# found in the LICENSE file.
|
| -
|
| -import os
|
| -import unittest
|
| -
|
| -import mock
|
| -
|
| -from monacq import acquisition_api
|
| -
|
| -from infra_libs.ts_mon import api_monitor
|
| -from infra_libs.ts_mon.protos import metrics_pb2
|
| -import infra_libs
|
| -
|
| -
|
| -class ApiMonitorTest(unittest.TestCase):
|
| -
|
| - def test_logging_callback(self):
|
| - """Smoke test for coverage: should not crash."""
|
| - api_monitor._logging_callback(200, "OK")
|
| -
|
| - @mock.patch('infra_libs.ts_mon.api_monitor.acquisition_api', auto_spec=True)
|
| - def test_init(self, fake_api):
|
| - _ = api_monitor.ApiMonitor('/path/to/creds.p8.json', 'https://www.tld/api')
|
| - # ApiMonitor calls abspath on the credential filename before passing it to
|
| - # AcquisitionCredential.Load. That doesn't change anything on Linux, but
|
| - # on Windows it prepends a drive letter and turns /s into \s.
|
| - fake_api.AcquisitionCredential.Load.assert_called_once_with(
|
| - os.path.abspath('/path/to/creds.p8.json'))
|
| - fake_api.AcquisitionApi.assert_called_once_with(
|
| - fake_api.AcquisitionCredential.Load.return_value,
|
| - 'https://www.tld/api')
|
| -
|
| - def test_init_gce_credential(self):
|
| - with self.assertRaises(NotImplementedError):
|
| - api_monitor.ApiMonitor(':gce', 'https://www.tld/api')
|
| -
|
| - @mock.patch('infra_libs.ts_mon.api_monitor.acquisition_api', auto_spec=True)
|
| - def test_send(self, _fake_api):
|
| - m = api_monitor.ApiMonitor('/path/to/creds.p8.json', 'https://www.tld/api')
|
| - metric1 = metrics_pb2.MetricsData(name='m1')
|
| - m.send(metric1)
|
| - metric2 = metrics_pb2.MetricsData(name='m2')
|
| - m.send([metric1, metric2])
|
| - collection = metrics_pb2.MetricsCollection(data=[metric1, metric2])
|
| - m.send(collection)
|
| - self.assertEquals(m._api.Send.call_count, 3)
|
| -
|
| - @mock.patch('infra_libs.ts_mon.api_monitor.acquisition_api', auto_spec=True)
|
| - def test_instrumented(self, fake_api):
|
| - m = api_monitor.ApiMonitor('/path/to/creds.p8.json', 'https://www.tld/api')
|
| - m.send(metrics_pb2.MetricsData(name='m1'))
|
| -
|
| - api = fake_api.AcquisitionApi.return_value
|
| - api.SetHttp.assert_called_once()
|
| - self.assertIsInstance(api.SetHttp.call_args[0][0],
|
| - infra_libs.httplib2_utils.InstrumentedHttp)
|
| -
|
| - @mock.patch('infra_libs.ts_mon.api_monitor.acquisition_api', auto_spec=True)
|
| - def test_not_instrumented(self, fake_api):
|
| - m = api_monitor.ApiMonitor('/path/to/creds.p8.json', 'https://www.tld/api',
|
| - use_instrumented_http=False)
|
| - m.send(metrics_pb2.MetricsData(name='m1'))
|
| -
|
| - self.assertFalse(fake_api.SetHttp.called)
|
| -
|
| - @mock.patch('infra_libs.ts_mon.api_monitor.acquisition_api.'
|
| - 'AcquisitionCredential', auto_spec=True)
|
| - @mock.patch('infra_libs.ts_mon.api_monitor.acquisition_api.AcquisitionApi',
|
| - auto_spec=True)
|
| - def test_failed_request_should_not_crash(self, _fake_api, _fake_creds):
|
| - m = api_monitor.ApiMonitor('/path/to/creds.p8.json', 'https://www.tld/api')
|
| - m._api.Send.side_effect = acquisition_api.AcquisitionApiRequestException()
|
| - m.send(metrics_pb2.MetricsData(name='m1'))
|
| -
|
|
|