mperham--sidekiq/test/test_testing.rb

# frozen_string_literal: true
require_relative 'helper'

describe 'Sidekiq::Testing' do
  describe 'require/load sidekiq/testing.rb' do
    before do
      require 'sidekiq/testing'
    end

    after do
      Sidekiq::Testing.disable!
    end

    it 'enables fake testing' do
      Sidekiq::Testing.fake!
      assert Sidekiq::Testing.enabled?
      assert Sidekiq::Testing.fake?
      refute Sidekiq::Testing.inline?
    end

    it 'enables fake testing in a block' do
      Sidekiq::Testing.disable!
      assert Sidekiq::Testing.disabled?
      refute Sidekiq::Testing.fake?

      Sidekiq::Testing.fake! do
        assert Sidekiq::Testing.enabled?
        assert Sidekiq::Testing.fake?
        refute Sidekiq::Testing.inline?
      end

      refute Sidekiq::Testing.enabled?
      refute Sidekiq::Testing.fake?
    end

    it 'disables testing in a block' do
      Sidekiq::Testing.fake!
      assert Sidekiq::Testing.fake?

      Sidekiq::Testing.disable! do
        refute Sidekiq::Testing.fake?
        assert Sidekiq::Testing.disabled?
      end

      assert Sidekiq::Testing.fake?
      assert Sidekiq::Testing.enabled?
    end
  end

  describe 'require/load sidekiq/testing/inline.rb' do
    before do
      require 'sidekiq/testing/inline'
    end

    after do
      Sidekiq::Testing.disable!
    end

    it 'enables inline testing' do
      Sidekiq::Testing.inline!
      assert Sidekiq::Testing.enabled?
      assert Sidekiq::Testing.inline?
      refute Sidekiq::Testing.fake?
    end

    it 'enables inline testing in a block' do
      Sidekiq::Testing.disable!
      assert Sidekiq::Testing.disabled?
      refute Sidekiq::Testing.fake?

      Sidekiq::Testing.inline! do
        assert Sidekiq::Testing.enabled?
        assert Sidekiq::Testing.inline?
      end

      refute Sidekiq::Testing.enabled?
      refute Sidekiq::Testing.inline?
      refute Sidekiq::Testing.fake?
    end
  end

  describe 'with middleware' do
    before do
      require 'sidekiq/testing'
    end

    after do
      Sidekiq::Testing.disable!
    end

    class AttributeWorker
      include Sidekiq::Worker
      sidekiq_class_attribute :count
      self.count = 0
      attr_accessor :foo

      def perform
        self.class.count += 1 if foo == :bar
      end
    end

    class AttributeMiddleware
      def call(worker, msg, queue)
        worker.foo = :bar if worker.respond_to?(:foo=)
        yield
      end
    end

    it 'wraps the inlined worker with middleware' do
      Sidekiq::Testing.server_middleware do |chain|
        chain.add AttributeMiddleware
      end

      begin
        Sidekiq::Testing.fake! do
          AttributeWorker.perform_async
          assert_equal 0, AttributeWorker.count
        end

        AttributeWorker.perform_one
        assert_equal 1, AttributeWorker.count

        Sidekiq::Testing.inline! do
          AttributeWorker.perform_async
          assert_equal 2, AttributeWorker.count
        end
      ensure
        Sidekiq::Testing.server_middleware.clear
      end
    end
  end

end