lib/ci/queue/static.rb in ci-queue-0.1.0 vs lib/ci/queue/static.rb in ci-queue-0.2.0

@@ -1,14 +1,16 @@
 module CI
   module Queue
     class Static
       attr_reader :progress, :total
 
-      def initialize(tests)
+      def initialize(tests, max_requeues: 0, requeue_tolerance: 0.0)
         @queue = tests
         @progress = 0
         @total = tests.size
+        @max_requeues = max_requeues
+        @global_max_requeues = (tests.size * requeue_tolerance).ceil
       end
 
       def to_a
         @queue.dup
       end
@@ -24,9 +26,35 @@
         end
       end
 
       def empty?
         @queue.empty?
+      end
+
+      def acknowledge(test, success)
+        if !success && should_requeue?(test)
+          requeue(test)
+          return false
+        end
+        
+        true
+      end
+
+      private
+
+      attr_reader :max_requeues, :global_max_requeues
+
+      def should_requeue?(test)
+        requeues[test] < max_requeues && requeues.values.inject(0, :+) < global_max_requeues
+      end
+
+      def requeue(test)
+        requeues[test] += 1
+        @queue.unshift(test)
+      end
+
+      def requeues
+        @requeues ||= Hash.new(0)
       end
     end
   end
 end