diff options
author | Selwin Ong <selwin.ong@gmail.com> | 2019-12-08 08:59:11 +0700 |
---|---|---|
committer | Selwin Ong <selwin.ong@gmail.com> | 2019-12-08 08:59:11 +0700 |
commit | 2844ba365d902ac5c6729d06e8e87752b3b875a3 (patch) | |
tree | b14548a66fc2b28da3ee851e81eb3177a147df3a | |
parent | 52ab7098e1415366fb60b4d55508bbb34fc447b4 (diff) | |
download | rq-job-scheduling.tar.gz |
Simplify worker.queue_names() and worker.queue_keys()job-scheduling
-rw-r--r-- | rq/worker.py | 4 | ||||
-rw-r--r-- | tests/test_worker.py | 3 |
2 files changed, 5 insertions, 2 deletions
diff --git a/rq/worker.py b/rq/worker.py index 0535c92..25b98e8 100644 --- a/rq/worker.py +++ b/rq/worker.py @@ -222,11 +222,11 @@ class Worker(object): def queue_names(self): """Returns the queue names of this worker's queues.""" - return list(map(lambda q: q.name, self.queues)) + return [queue.name for queue in self.queues] def queue_keys(self): """Returns the Redis keys representing this worker's queues.""" - return list(map(lambda q: q.key, self.queues)) + return [queue.key for queue in self.queues] @property def key(self): diff --git a/tests/test_worker.py b/tests/test_worker.py index 309f350..e983bf5 100644 --- a/tests/test_worker.py +++ b/tests/test_worker.py @@ -59,6 +59,9 @@ class TestWorker(RQTestCase): self.assertEqual(w.queues[0].name, 'foo') self.assertEqual(w.queues[1].name, 'bar') + self.assertEqual(w.queue_keys(), [w.queues[0].key, w.queues[1].key]) + self.assertEqual(w.queue_names(), ['foo', 'bar']) + # With iterable of strings w = Worker(iter(['foo', 'bar'])) self.assertEqual(w.queues[0].name, 'foo') |