require_relative '../../test_helper'
module Plugins
  module Job
    # Unit Test for RocketJob::Job
    class StateMachineTest < Minitest::Test

      class StateMachineJob < RocketJob::Job
        def perform
        end
      end

      describe RocketJob::Plugins::Job::StateMachine do
        before do
          @job = StateMachineJob.new
        end

        after do
          @job.destroy if @job && !@job.new_record?
        end

        describe '#requeue!' do
          it 'requeue jobs from dead workers' do
            worker_name      = 'server:12345'
            @job.worker_name = worker_name
            @job.start!
            assert @job.running?

            @job.requeue!(worker_name)
            @job.reload

            assert @job.queued?
            assert_nil @job.worker_name
          end
        end

        describe '#requeue' do
          it 'requeue jobs from dead workers' do
            worker_name      = 'server:12345'
            @job.worker_name = worker_name
            assert @job.valid?, @job.errors.messages
            @job.start!
            assert @job.running?, @job.state

            @job.requeue(worker_name)
            assert @job.queued?
            assert_nil @job.worker_name

            @job.reload
            assert @job.running?
            assert_equal worker_name, @job.worker_name
          end
        end

        describe '#after_complete' do
          it 'destroy on complete' do
            @job.destroy_on_complete = true
            @job.perform_now
            assert @job.completed?, @job.state
            assert_equal 0, RocketJob::Job.where(id: @job.id).count
          end
        end

        describe '#fail!' do
          it 'fail with message' do
            @job.start!
            @job.fail!('myworker:2323', 'oh no')
            assert @job.failed?
            assert exc = @job.exception
            assert_equal 'RocketJob::JobException', exc.class_name
            assert_equal 'oh no', exc.message
          end

          it 'fail with no arguments' do
            @job.start!
            @job.fail!
            assert @job.failed?
            assert exc = @job.exception
            assert_equal 'RocketJob::JobException', exc.class_name
            assert_nil exc.message
            assert_nil exc.worker_name
            assert_equal [], exc.backtrace
          end

          it 'fail with exception' do
            @job.start!
            exception = RuntimeError.new('Oh no')
            @job.fail!('myworker:2323', exception)
            assert @job.failed?
            assert exc = @job.exception
            assert_equal exception.class.name, exc.class_name
            assert_equal exception.message, exc.message
            assert_equal [], exc.backtrace
          end
        end

        describe '#retry!' do
          it 'retry failed jobs' do
            worker_name      = 'server:12345'
            @job.worker_name = worker_name
            @job.start!
            assert @job.running?
            assert_equal worker_name, @job.worker_name

            @job.fail!(worker_name, 'oh no')
            assert @job.failed?
            assert_equal 'oh no', @job.exception.message

            @job.retry!
            assert @job.queued?
            assert_nil @job.worker_name
            assert_nil @job.exception
          end
        end

      end
    end
  end
end