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)