Python MockMetricsManagerClient Examples

Example #1

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

Example #2

File: metricsmgr_client_unittest.py Project: billonahill/heron

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

Example #3

 def setUp(self):
     self.metrics_client = MockMetricsManagerClient()

Example #4

File: metricsmgr_client_unittest.py Project: billonahill/heron

 def setUp(self):
   self.metrics_client = MockMetricsManagerClient()