def test_timers_data(self): shelf = StatsShelf() self.assertEqual(dict(), shelf.timers_data()) shelf.add(Timer("query.user", 3.223)) shelf.add(Timer("query.user", 4.12)) shelf.add(Timer("api.auth", 9.78)) shelf.add(Timer("api.auth", 8.45)) expected = {"query.user": [3.223, 4.12], "api.auth": [9.78, 8.45]} self.assertEqual(expected, shelf.timers_data())
def test_clear_all_metrics(self): shelf = StatsShelf() shelf.add(Set("users", "me")) shelf.add(Counter("mymetric", 3)) shelf.add(Timer("query", 4.12)) shelf.add(Gauge("cpu%", 38)) shelf.clear() self.assertEqual(dict(), shelf.counters()) self.assertEqual(dict(), shelf.sets()) self.assertEqual(dict(), shelf.timers_data()) self.assertEqual(dict(), shelf.gauges())