def test_context_manager(self, mock_time):
timer = metrics.Timer(self.transport, b"example")
mock_time.return_value = 1000
with timer:
mock_time.return_value = 1003
self.assertEqual(self.transport.send.call_count, 1)
self.assertEqual(self.transport.send.call_args,
mock.call(b"example:3000|ms"))
def test_basic_operation(self, mock_time):
timer = metrics.Timer(self.transport, b"example")
with self.assertRaises(Exception):
timer.stop()
mock_time.return_value = 1000
timer.start()
with self.assertRaises(Exception):
timer.start()
self.assertEqual(self.transport.send.call_count, 0)
mock_time.return_value = 1004
timer.stop()
self.assertEqual(self.transport.send.call_count, 1)
self.assertEqual(self.transport.send.call_args,
mock.call(b"example:4000|ms"))
with self.assertRaises(Exception):
timer.start()
with self.assertRaises(Exception):
timer.stop()
def test_send(self):
timer = metrics.Timer(self.transport, b"example")
timer.send(3.14)
self.assertEqual(self.transport.send.call_count, 1)
self.assertEqual(self.transport.send.call_args,
mock.call(b"example:3140|ms"))