class RedisCountersTestCase(unittest.TestCase):
def setUp(self):
self.redis_mock = FakeRedis()
self.counter = RedisCounters(rootkey='counters', redis_instance=self.redis_mock)
def test_ping_does_an_hset_on_the_root_key(self):
self.redis_mock.hset = Mock(return_value=1)
self.counter.ping("a.b")
self.redis_mock.hset.assert_called_with('counters', 'pings.a.b', int(time.time()))
def test_ping_raises_exception_when_invalid_key(self):
self.redis_mock.hset = Mock(return_value=1)
self.assertRaises(ValueError, self.counter.ping, "a/b")
def test_hit_with_increment_does_an_hincr_on_the_named_key(self):
self.redis_mock.hincrby = Mock(return_value=1)
self.counter.hit('bytes_in', 2019)
self.redis_mock.hincrby.assert_called_with('counters', 'hits.bytes_in', 2019)
def test_hit_does_an_hincr_on_the_named_key(self):
self.redis_mock.hincrby = Mock(return_value=1)
self.counter.hit('pages_crawled')
self.redis_mock.hincrby.assert_called_with('counters', 'hits.pages_crawled', 1)
def setUp(self): self.redis_mock = FakeRedis() self.counter = RedisCounters(rootkey='counters', redis_instance=self.redis_mock)
