def test_worker_ttl(self): """Worker ttl.""" w = Worker([]) w.register_birth() # ugly: our test should only call public APIs [worker_key] = self.testconn.smembers(Worker.redis_workers_keys) self.assertIsNotNone(self.testconn.ttl(worker_key)) w.register_death()
def test_gevent_num(self): """ensure gevent num is registered.""" q = Queue() w = GeventWorker(q) w.heartbeat() self.assertEqual(self.testconn.hget(w.key, 'curr_pool_len'), '0') job = q.enqueue(say_hello) w.execute_job(job, q) w.heartbeat() self.assertEqual(self.testconn.hget(w.key, 'curr_pool_len'), '1') w.register_death()