def test_set_key(self): tc = ThreadedCounter(start_time=1442671176) self.assertEqual('default_counter', tc.get_key()) tc.roll = True tc.key = 'myKey' tc._set_key() self.assertEqual('myKey:2015-09-19_13:59:36', tc.get_key())
def test_purge_old(self): # test removing old keys tc = ThreadedCounter(keep_max=1) tc.keep_max = 1 tc.redis_conn = self.redis_conn self.redis_conn.set('default_counter:2015-09', 'stuff') self.redis_conn.set('default_counter:2015-10', 'stuff2') tc.purge_old() self.assertEqual(['default_counter:2015-10'], self.redis_conn.keys(tc.get_key() + ':*')) self.redis_conn.delete('default_counter:2015-10')