rocketjob 3.5.2 → 4.0.0

data/test/plugins/restart_test.rb

@@ -1,193 +0,0 @@
-require_relative '../test_helper'
-
-# Unit Test for RocketJob::Job
-module Plugins
-  class RestartTest < Minitest::Test
-    class RestartableJob < RocketJob::Job
-      include RocketJob::Plugins::Restart
-
-      field :start_at, type: Date
-      field :end_at, type: Date
-
-      def perform
-        self.start_at = Date.today
-        self.end_at   = Date.today
-        'DONE'
-      end
-    end
-
-    class RestartablePausableJob < RocketJob::Job
-      include RocketJob::Plugins::Restart
-
-      field :start_at, type: Date
-      field :end_at, type: Date
-
-      # Job will reload itself during process to check if it was paused.
-      self.pausable = true
-
-      def perform
-        self.start_at = Date.today
-        self.end_at   = Date.today
-        'DONE'
-      end
-    end
-
-    describe RocketJob::Plugins::Restart do
-      before do
-        RestartableJob.delete_all
-        RestartablePausableJob.delete_all
-      end
-
-      after do
-        @job.delete if @job && !@job.new_record?
-      end
-
-      describe '#create!' do
-        it 'queues a new job' do
-          @job = RestartableJob.create!
-          assert @job.valid?
-          refute @job.new_record?
-        end
-      end
-
-      describe '#save!' do
-        it 'queues a new job' do
-          @job = RestartableJob.new
-          @job.save!
-          assert @job.valid?
-          refute @job.new_record?
-        end
-      end
-
-      describe '#abort!' do
-        it 'queues a new job on abort' do
-          @job = RestartableJob.create!
-          @job.abort!
-          assert_equal 2, RestartableJob.count
-          assert other = RestartableJob.where(:id.ne => @job.id).first
-          refute_equal @job.id, other.id
-          assert other.queued?
-        end
-
-        it 'does not queue a new job when expired' do
-          @job = RestartableJob.create!(expires_at: Time.now - 1.day)
-          assert @job.expired?
-          @job.abort!
-          assert_equal 1, RestartableJob.count
-          assert_nil RestartableJob.where(:id.ne => @job.id).first
-        end
-      end
-
-      describe '#complete' do
-        it 'queues a new job when destroy_on_complete' do
-          assert_equal 0, RestartableJob.count
-          @job = RestartableJob.create!(destroy_on_complete: true)
-          @job.perform_now
-          assert @job.completed?, @job.attributes.ai
-          assert_equal 1, RestartableJob.count, RestartableJob.all.to_a.ai
-        end
-
-        it 'queues a new job when not destroy_on_complete' do
-          @job = RestartableJob.create!(destroy_on_complete: false)
-          @job.perform_now
-          assert @job.completed?
-          assert_equal 2, RestartableJob.count
-        end
-
-        it 'does not queue a new job when expired' do
-          @job = RestartableJob.create!(expires_at: Time.now - 1.day)
-          @job.perform_now
-          assert @job.expired?
-          assert @job.completed?
-          assert_equal 0, RestartableJob.count
-        end
-      end
-
-      describe '#pause' do
-        it 'does not enqueue a new job when paused' do
-          @job = RestartablePausableJob.new
-          @job.start
-          @job.pause!
-          assert @job.paused?
-          assert_equal 1, RestartablePausableJob.count
-        end
-      end
-
-      describe '#fail' do
-        it 'aborts from queued' do
-          @job = RestartableJob.new
-          assert @job.queued?
-          @job.fail
-          assert @job.aborted?
-        end
-
-        it 'aborts from running' do
-          @job = RestartableJob.new
-          @job.start
-          assert @job.running?
-          @job.fail
-          assert @job.aborted?
-        end
-
-        it 'aborts from paused' do
-          @job = RestartablePausableJob.new
-          @job.start
-          @job.pause
-          assert @job.paused?
-          @job.fail
-          assert @job.aborted?
-        end
-
-        it 'does not queue a new job when expired' do
-          @job = RestartableJob.new(expires_at: Time.now - 1.day)
-          @job.start!
-          assert @job.running?
-          assert @job.expired?
-          assert_equal 1, RestartableJob.count
-          assert_nil RestartableJob.where(:id.ne => @job.id).first
-        end
-      end
-
-      describe '#create_new_instance' do
-        it 'sets job back to queued state' do
-          @job = RestartableJob.create!(destroy_on_complete: true)
-          @job.perform_now
-          assert_equal 1, RestartableJob.count
-          assert job2 = RestartableJob.where(:id.ne => @job.id).first
-          assert job2.queued?, job2.attributes.ai
-        end
-
-        it 'excludes attributes related to running jobs' do
-          @job = RestartableJob.create!(destroy_on_complete: true, expires_at: Time.now + 1.day)
-          refute @job.expired?
-          @job.perform_now
-          assert_equal 1, RestartableJob.count
-          assert job2 = RestartableJob.where(:id.ne => @job.id).first
-          assert job2.queued?, job2.attributes.ai
-
-          assert RestartableJob.rocket_job_restart_attributes.include?(:priority)
-          assert RestartableJob.rocket_job_restart_attributes.exclude?(:start_at)
-          assert RestartableJob.rocket_job_restart_attributes.exclude?(:end_at)
-          assert RestartableJob.rocket_job_restart_attributes.exclude?(:run_at)
-
-          # Copy across all attributes, except
-          RestartableJob.rocket_job_restart_attributes.each do |key|
-            assert_equal @job.send(key).to_s, job2.send(key).to_s, "Attributes are supposed to be copied across. For #{key}"
-          end
-
-          assert_nil job2.start_at
-          assert_nil job2.end_at
-          assert_equal :queued, job2.state
-          assert job2.created_at
-          assert_nil job2.started_at
-          assert_nil job2.completed_at
-          assert_equal 0, job2.failure_count
-          assert_nil job2.worker_name
-          assert_equal 0, job2.percent_complete
-          assert_nil job2.exception
-          refute job2.result
-        end
-      end
-    end
-  end
-end

data/test/plugins/retry_test.rb

@@ -1,88 +0,0 @@
-require_relative '../test_helper'
-
-module Plugins
-  class RetryTest < Minitest::Test
-    class RetryJob < RocketJob::Job
-      include RocketJob::Plugins::Retry
-
-      # Fails 5 times before succeeding
-      def perform
-        raise 'Oh No' unless rocket_job_failure_count >= 5
-        'DONE'
-      end
-    end
-
-    describe RocketJob::Plugins::Retry do
-      before do
-        RetryJob.delete_all
-      end
-
-      after do
-        @job.delete if @job && !@job.new_record?
-      end
-
-      describe '#perform' do
-        it 're-queues job on failure' do
-          @job = RetryJob.create!
-          assert created_at = @job.created_at
-          assert_equal 0, @job.failed_at_list.size
-
-          assert_raises RuntimeError do
-            @job.perform_now
-          end
-
-          assert @job.queued?, -> { @job.attributes.ai }
-
-          # Includes failure time
-          assert_equal 1, @job.rocket_job_failure_count
-          assert failed_at = @job.failed_at_list.first
-          assert failed_at >= created_at
-
-          assert next_time = @job.run_at
-          assert next_time > failed_at
-        end
-
-        it 're-queues until it succeeds' do
-          @job = RetryJob.create!
-
-          # 5 retries
-          5.times do |i|
-            assert_raises RuntimeError do
-              @job.perform_now
-            end
-            assert @job.queued?, -> { @job.attributes.ai }
-            assert_equal (i + 1), @job.rocket_job_failure_count
-          end
-
-          assert_equal 5, @job.rocket_job_failure_count
-
-          # Should succeed on the 6th attempt
-          @job.perform_now
-          assert @job.completed?, -> { @job.attributes.ai }
-          assert_equal 5, @job.rocket_job_failure_count
-        end
-
-        it 'stops re-queueing after limit is reached' do
-          @job = RetryJob.create!(retry_limit: 3)
-
-          # 3 attempts are retried
-          3.times do |i|
-            assert_raises RuntimeError do
-              @job.perform_now
-            end
-            assert @job.queued?, -> { @job.attributes.ai }
-            assert_equal (i + 1), @job.rocket_job_failure_count
-          end
-
-          # Should fail on the 4th attempt
-          assert_equal 3, @job.rocket_job_failure_count
-          assert_raises RuntimeError do
-            @job.perform_now
-          end
-
-          assert @job.failed?, -> { @job.attributes.ai }
-        end
-      end
-    end
-  end
-end

data/test/plugins/singleton_test.rb

@@ -1,92 +0,0 @@
-require_relative '../test_helper'
-
-module Plugins
-  # Unit Test for RocketJob::Job
-  class SingletonTest < Minitest::Test
-    class SingletonJob < RocketJob::Job
-      include RocketJob::Plugins::Singleton
-
-      def perform
-      end
-    end
-
-    describe RocketJob::Plugins::Singleton do
-      before do
-        SingletonJob.delete_all
-      end
-
-      after do
-        @job.destroy if @job && !@job.new_record?
-      end
-
-      describe '#rocket_job_singleton_active?' do
-        it 'returns false if no jobs of this class are active' do
-          @job = SingletonJob.new
-          assert_equal false, @job.rocket_job_singleton_active?
-        end
-
-        it 'excludes self when queued from check' do
-          @job = SingletonJob.create
-          assert @job.queued?
-          assert_equal false, @job.rocket_job_singleton_active?
-        end
-
-        it 'excludes self when started from check' do
-          @job = SingletonJob.new
-          @job.start!
-          assert @job.running?
-          assert_equal false, @job.rocket_job_singleton_active?
-        end
-
-        it 'returns true when other jobs of this class are queued' do
-          @job = SingletonJob.create!
-          job2 = SingletonJob.new
-          assert_equal true, job2.rocket_job_singleton_active?
-        end
-
-        it 'returns true when other jobs of this class are running' do
-          @job = SingletonJob.new
-          @job.start!
-          job2 = SingletonJob.new
-          assert_equal true, job2.rocket_job_singleton_active?
-        end
-
-        it 'returns false when other jobs of this class are not active' do
-          @job = SingletonJob.new
-          @job.perform_now
-          @job.save!
-          assert @job.completed?
-          job2 = SingletonJob.new
-          assert_equal false, job2.rocket_job_singleton_active?
-        end
-      end
-
-      describe 'validation' do
-        it 'passes if another job is not active' do
-          @job = SingletonJob.new
-          @job.perform_now
-          @job.save!
-          job2 = SingletonJob.new
-          assert_equal true, job2.valid?
-        end
-
-        it 'fails if another job is active' do
-          @job = SingletonJob.new
-          @job.start!
-          job2 = SingletonJob.new
-          refute job2.valid?
-          assert_equal ['Another instance of Plugins::SingletonTest::SingletonJob is already running, queued, or paused'], job2.errors.messages[:state]
-        end
-
-        it 'passes if another job is active, but this job is not' do
-          @job = SingletonJob.new
-          @job.start!
-          job2 = SingletonJob.new
-          job2.abort
-          assert job2.valid?
-          job2.save!
-        end
-      end
-    end
-  end
-end

data/test/plugins/state_machine_event_callbacks_test.rb

@@ -1,102 +0,0 @@
-require_relative '../test_helper'
-
-module Plugins
-  # Unit Test for RocketJob::Job
-  class StateMachineEventCallbacksTest < Minitest::Test
-    # This job adds each event callback as they run into an array
-    # [:start, :complete, :fail, :retry, :pause, :resume, :abort, :requeue]
-    class PositivePathJob < RocketJob::Job
-      field :call_list, type: Array, default: []
-
-      before_complete do
-        call_list << 'before_complete_block'
-      end
-
-      after_complete do
-        call_list << 'after_complete_block'
-      end
-
-      before_complete :before_complete_method
-
-      before_start do
-        call_list << 'before_start_block'
-      end
-
-      before_start :before_start_method
-
-      before_start do
-        call_list << 'before_start2_block'
-      end
-
-      after_start :after_start_method
-      after_complete :after_complete_method
-
-      before_complete do
-        call_list << 'before_complete2_block'
-      end
-
-      after_start do
-        call_list << 'after_start_block'
-      end
-
-      after_complete do
-        call_list << 'after_complete2_block'
-      end
-
-      after_start :after_start_method
-
-      after_start do
-        call_list << 'after_start2_block'
-      end
-
-      before_start :before_start_method2
-      before_complete :before_complete_method2
-
-      def perform
-        call_list << 'perform'
-      end
-
-      private
-
-      def before_start_method
-        call_list << 'before_start_method'
-      end
-
-      def before_start_method2
-        call_list << 'before_start_method2'
-      end
-
-      def after_start_method
-        call_list << 'after_start_method'
-      end
-
-      def before_complete_method
-        call_list << 'before_complete_method'
-      end
-
-      def before_complete_method2
-        call_list << 'before_complete_method2'
-      end
-
-      def after_complete_method
-        call_list << 'after_complete_method'
-      end
-    end
-
-    describe RocketJob::Plugins::StateMachine do
-      after do
-        @job.destroy if @job && !@job.new_record?
-      end
-
-      describe 'before_start after_start & before_complete after_complete' do
-        it 'runs blocks and functions' do
-          @job = PositivePathJob.new
-          @job.perform_now
-          assert @job.completed?, @job.attributes.ai
-          expected = %w[before_start_block before_start_method before_start2_block before_start_method2 after_start2_block after_start_method after_start_block perform before_complete_block before_complete_method before_complete2_block before_complete_method2 after_complete2_block after_complete_method after_complete_block]
-          assert_equal expected, @job.call_list, 'Sequence of before_perform callbacks is incorrect'
-        end
-      end
-    end
-  end
-end