Add Queue().list_raw_jobs() method

Author: sylvinus

mrq/queue.py

@@ -117,23 +117,31 @@ def list_job_ids(self, skip=0, limit=20):
         if self.is_raw:
             raise Exception("Can't list job ids from a raw queue")
 
-        # ZSET
+        return self.unserialize_job_ids(self._get_queue_content(skip, limit))
+
+    def list_raw_jobs(self, skip=0, limit=20):
+
+        if not self.is_raw:
+            raise Exception("Queue is not raw")
+
+        return self._get_queue_content(skip, limit)
+
+    def _get_queue_content(self, skip, limit):
         if self.is_sorted:
-            return self.unserialize_job_ids(context.connections.redis.zrange(
+            return context.connections.redis.zrange(
                 self.redis_key,
                 skip,
-                skip + limit - 1))
+                skip + limit - 1)
         # SET
         elif self.is_set:
-            return self.unserialize_job_ids(
-                context.connections.redis.srandmember(self.redis_key, limit)
-            )
+            return context.connections.redis.srandmember(self.redis_key, limit)
+
         # LIST
         else:
-            return self.unserialize_job_ids(context.connections.redis.lrange(
+            return context.connections.redis.lrange(
                 self.redis_key,
                 skip,
-                skip + limit - 1))
+                skip + limit - 1)
 
     def get_sorted_graph(
             self,

tests/test_raw.py

@@ -49,8 +49,10 @@ def test_raw_sorted(worker, p_queue, p_pushback, p_timed, p_flags):
 
     if p_pushback:
         assert Queue(p_queue).size() == 3
+        assert set(Queue(p_queue).list_raw_jobs()) == set(["bbb", "ccc", "aaa"])
     else:
         assert Queue(p_queue).size() == 2
+        assert set(Queue(p_queue).list_raw_jobs()) == set(["bbb", "ccc"])
 
     # The second one should not yet even exist in mrq_jobs
     assert jobs_collection.count() == 1