queue: Add some public methods for easier testing

skroutz · Jul 17, 2020 · b69b8e1 · b69b8e1
1 parent 9f1e6fb
commit b69b8e1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/lib/rspecq/queue.rb b/lib/rspecq/queue.rb
@@ -57,6 +57,8 @@ class Queue
     STATUS_INITIALIZING = "initializing".freeze
     STATUS_READY = "ready".freeze
 
+    attr_reader :redis
+
     def initialize(build_id, worker_id, redis_host)
       @build_id = build_id
       @worker_id = worker_id
@@ -157,6 +159,14 @@ def processed_jobs_count
       @redis.scard(key_queue_processed)
     end
 
+    def processed_jobs
+      @redis.smembers(key_queue_processed)
+    end
+
+    def requeued_jobs
+      @redis.hgetall(key_requeues)
+    end
+
     def become_master
       @redis.setnx(key_queue_status, STATUS_INITIALIZING)
     end