def test_uniques(self): counter = UniqueCounter(key='test_key') counter.setup(redis_conn=self.redis_conn) i = 0 while i < 1000: counter.increment(i % 100) i += 1 self.assertEqual(counter.value(), 100) counter.stop() self.clean_keys(counter.key)
def test_roll_uniques(self): # rough sleep to get us back on track time.sleep(2 - (int(time.time()) % 2)) counter = UniqueCounter(key='test_key', window=2, cycle_time=.1, roll=True) counter.setup(redis_conn=self.redis_conn) i = 0 while i < 100: counter.increment(i % 10) i += 1 self.assertEqual(counter.value(), 10) time.sleep(3) # now the counter should have rolled i = 0 while i < 50: counter.increment(i % 10) i += 1 self.assertEqual(counter.value(), 10) counter.stop() self.clean_keys(counter.key)