def test_send_tagged(self):
tags = {"test": "true"}
timer = metrics.Timer(self.transport, b"example", tags)
timer.send(3.14)
self.assertEqual(self.transport.send.call_count, 1)
self.assertEqual(self.transport.send.call_args,
mock.call(b"example,test=true:3140|ms"))
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"))
def test_update_tags(self):
tags = {"test": "true"}
timer = metrics.Timer(self.transport, b"example", tags)
new_tags = {"test2": "false"}
timer.update_tags(new_tags)
self.assertNotEqual(timer.tags, new_tags)
def test_init_with_tags(self):
tags = {"test": "true"}
timer = metrics.Timer(self.transport, b"example", tags)
self.assertEqual(timer.tags, tags)