def test_sets(self): shelf = StatsShelf() self.assertEqual(dict(), shelf.sets()) shelf.add(Set("users", "me")) shelf.add(Set("users", "me")) shelf.add(Set("users", "you")) shelf.add(Set("say.what?", "nothing")) shelf.add(Set("users", "me")) shelf.add(Set("say.what?", "nothing")) shelf.add(Set("say.what?", "ok")) expected = {"users": {"me", "you"}, "say.what?": {"nothing", "ok"}} self.assertEqual(expected, shelf.sets()) sets = shelf.sets() sets["sets should"] = set("not change") self.assertEqual(expected, shelf.sets())
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())