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