class MetricsMgrClientTest(unittest.TestCase): def setUp(self): self.metrics_client = MockMetricsManagerClient() def tearDown(self): self.metrics_client = None def test_add_task(self): # _add_metrics_client_tasks() should have been already called self.assertEqual(len(self.metrics_client.looper.wakeup_tasks), 1) self.assertEqual(self.metrics_client.looper.wakeup_tasks[0].__name__, "_send_metrics_messages") def test_on_connect_ok(self): self.metrics_client.on_connect(StatusCode.OK) self.assertTrue(self.metrics_client.register_req_called) def test_on_connect_error(self): self.metrics_client.on_connect(StatusCode.CONNECT_ERROR) self.assertEqual(len(self.metrics_client.looper.timer_tasks), 1) self.assertEqual(self.metrics_client.looper.timer_tasks[0][1].__name__, 'start_connect') def test_on_error(self): self.metrics_client.on_error() self.assertEqual(len(self.metrics_client.looper.timer_tasks), 1) self.assertEqual(self.metrics_client.looper.timer_tasks[0][1].__name__, 'start_connect')
class MetricsMgrClientTest(unittest.TestCase): def setUp(self): self.metrics_client = MockMetricsManagerClient() def tearDown(self): self.metrics_client = None def test_add_task(self): # _add_metrics_client_tasks() should have been already called self.assertEqual(len(self.metrics_client.looper.wakeup_tasks), 1) self.assertEqual(self.metrics_client.looper.wakeup_tasks[0].__name__, "_send_metrics_messages") def test_on_connect_ok(self): self.metrics_client.on_connect(StatusCode.OK) self.assertTrue(self.metrics_client.register_req_called) def test_on_connect_error(self): self.metrics_client.on_connect(StatusCode.CONNECT_ERROR) self.assertEqual(len(self.metrics_client.looper.timer_tasks), 3) self.assertEqual(self.metrics_client.looper.timer_tasks[0][1].__name__, '_update_in_out_stream_metrics_tasks') self.assertEqual(self.metrics_client.looper.timer_tasks[1][1].__name__, '_update_py_metrics') self.assertEqual(self.metrics_client.looper.timer_tasks[2][1].__name__, 'start_connect') def test_on_error(self): self.metrics_client.on_error() self.assertEqual(len(self.metrics_client.looper.timer_tasks), 3) self.assertEqual(self.metrics_client.looper.timer_tasks[0][1].__name__, '_update_in_out_stream_metrics_tasks') self.assertEqual(self.metrics_client.looper.timer_tasks[1][1].__name__, '_update_py_metrics') self.assertEqual(self.metrics_client.looper.timer_tasks[2][1].__name__, 'start_connect')
def setUp(self): self.metrics_client = MockMetricsManagerClient()