From 2844ba365d902ac5c6729d06e8e87752b3b875a3 Mon Sep 17 00:00:00 2001 From: Selwin Ong Date: Sun, 8 Dec 2019 08:59:11 +0700 Subject: Simplify worker.queue_names() and worker.queue_keys() --- rq/worker.py | 4 ++-- 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') -- cgit v1.2.1