diff options
| author | Selwin Ong <selwin.ong@gmail.com> | 2017-11-24 21:45:55 +0700 |
|---|---|---|
| committer | Selwin Ong <selwin.ong@gmail.com> | 2017-11-24 21:45:55 +0700 |
| commit | f45bb6f7b7644bda572f70b45168b34fde785826 (patch) | |
| tree | 8d83178cdad6298aa6e6d094282cc619fe09be91 /tests/test_worker_registry.py | |
| parent | 7d23d752cc59e0700b0d941b62009ef9319cd759 (diff) | |
| download | rq-worker-by-queue.tar.gz | |
Added worker_registration.get_keys().worker-by-queue
Diffstat (limited to 'tests/test_worker_registry.py')
| -rw-r--r-- | tests/test_worker_registry.py | 33 |
1 files changed, 0 insertions, 33 deletions
diff --git a/tests/test_worker_registry.py b/tests/test_worker_registry.py deleted file mode 100644 index 8393348..0000000 --- a/tests/test_worker_registry.py +++ /dev/null @@ -1,33 +0,0 @@ -from tests import RQTestCase - -from rq import Queue, Worker -from rq.worker_registration import register, unregister, workers_by_queue_key - - -class TestWorkerRegistry(RQTestCase): - - def test_worker_registration(self): - """Ensure worker.key is correctly set in Redis.""" - foo_queue = Queue(name='foo') - bar_queue = Queue(name='bar') - worker = Worker([foo_queue, bar_queue]) - - register(worker) - redis = worker.connection - - self.assertTrue(redis.sismember(worker.redis_workers_keys, worker.key)) - self.assertTrue( - redis.sismember(workers_by_queue_key % foo_queue.name, worker.key) - ) - self.assertTrue( - redis.sismember(workers_by_queue_key % bar_queue.name, worker.key) - ) - - unregister(worker) - self.assertFalse(redis.sismember(worker.redis_workers_keys, worker.key)) - self.assertFalse( - redis.sismember(workers_by_queue_key % foo_queue.name, worker.key) - ) - self.assertFalse( - redis.sismember(workers_by_queue_key % bar_queue.name, worker.key) - ) |
