test/plugins/job/throttle_test.rb in rocketjob-3.0.4 vs test/plugins/job/throttle_test.rb in rocketjob-3.0.5

@@ -5,20 +5,20 @@
   module Job
     class ThrottleTest < Minitest::Test
 
       class ThrottleJob < RocketJob::Job
         # Only allow one to be processed at a time
-        self.throttle_max_workers = 1
+        self.throttle_running_jobs = 1
 
         def perform
           21
         end
       end
 
-      describe RocketJob::Plugins::Job::Logger do
+      describe RocketJob::Plugins::Job::Throttle do
         before do
-          ThrottleJob.delete_all
+          RocketJob::Job.delete_all
         end
 
         describe '#throttle_exceeded?' do
           it 'does not exceed throttle when no other jobs are running' do
             ThrottleJob.create!
@@ -45,9 +45,48 @@
             job1 = ThrottleJob.new
             job1.start
             job1.fail!
             job2 = ThrottleJob.new
             refute job2.throttle_exceeded?
+          end
+        end
+
+        describe '.rocket_job_next_job' do
+          before do
+            @worker_name = 'worker:123'
+          end
+
+          after do
+            @job.destroy if @job && !@job.new_record?
+          end
+
+          it 'return nil when no jobs available' do
+            assert_nil RocketJob::Job.rocket_job_next_job(@worker_name)
+          end
+
+          it 'return the job when others are queued, paused, failed, or complete' do
+            @job = ThrottleJob.create!
+            ThrottleJob.create!(state: :failed)
+            ThrottleJob.create!(state: :complete)
+            ThrottleJob.create!(state: :paused)
+            assert job = RocketJob::Job.rocket_job_next_job(@worker_name), -> { ThrottleJob.all.to_a.ai }
+            assert_equal @job.id, job.id, -> { ThrottleJob.all.to_a.ai }
+          end
+
+          it 'return nil when other jobs are running' do
+            ThrottleJob.create!
+            @job = ThrottleJob.new
+            @job.start!
+            assert_nil RocketJob::Job.rocket_job_next_job(@worker_name), -> { ThrottleJob.all.to_a.ai }
+          end
+
+          it 'add job to filter when other jobs are running' do
+            ThrottleJob.create!
+            @job = ThrottleJob.new
+            @job.start!
+            filter = {}
+            assert_nil RocketJob::Job.rocket_job_next_job(@worker_name, filter), -> { ThrottleJob.all.to_a.ai }
+            assert_equal 1, filter.size
           end
         end
 
       end
     end