def test_metr_handlers_inheritance(self):
handler = mock.MagicMock()
metr_1 = metrics.get_metr("a")
metr_1.add_handler(handler)
metr_2 = metrics.get_metr("a.b")
metr_2.rec(8)
record = handler.handle.call_args_list[0][0][0]
self.assertEqual(8, record.value)
self.assertEqual("a.b", record.tag)
def test_handlers_closing(self):
cursor = mock.MagicMock()
cursor.__enter__.return_value = cursor
connection = mock.MagicMock()
connection.cursor.return_value = cursor
metr_1 = metrics.get_metr("test_metr_1")
metr_2 = metrics.get_metr("test_metr_2")
handler = metrics.RDSHandler(connection)
metr_1.add_handler(handler)
metr_2.add_handler(handler)
metrics.shutdown_handlers()
connection.close.assert_called_once()
def test_int_record(self):
handler = mock.MagicMock()
metr = metrics.get_metr("test_int_record")
metr.add_handler(handler)
metr.rec(7)
record = handler.handle.call_args_list[0][0][0]
self.assertEqual(7, record.value)
def test_counter_record_context(self):
handler = mock.MagicMock()
metr = metrics.get_metr("test_counter_record_context")
metr.add_handler(handler)
with metr.rec_counter() as counter:
counter.add(2)
counter.add(3)
record = handler.handle.call_args_list[0][0][0]
self.assertEqual(5, record.value)
def test_counter_record(self):
handler = mock.MagicMock()
metr = metrics.get_metr("test_counter_record")
metr.add_handler(handler)
counter = metr.rec_counter()
counter.add(8)
counter.add(7)
counter.close()
record = handler.handle.call_args_list[0][0][0]
self.assertEqual(15, record.value)
def test_exception_record(self):
handler = mock.MagicMock()
metr = metrics.get_metr("test_exception_record")
metr.add_handler(handler)
@metr.rec_exception(ValueError)
def f():
raise ValueError
try:
f()
except ValueError:
record = handler.handle.call_args_list[0][0][0]
self.assertEqual(1, record.value)
else:
self.fail()