Merge pull request #171 from panthomakos/pt/inline

Added inline testing functionality.

Changes.md

@@ -29,6 +29,9 @@ end
 - In test environments add the `#drain` class method to workers. This method
   executes all previously queued jobs. (panthomakos)
 
+- Sidekiq workers can be run inline during tests by requiring the
+  `sidekiq/testing/inline` file. (panthomakos)
+
 1.1.4
 -----------
 

lib/sidekiq/testing/inline.rb

@@ -0,0 +1,37 @@
+module Sidekiq
+  module Worker
+
+    ##
+    # The Sidekiq inline infrastructure overrides the perform_async so that it
+    # actually calls perform instead. This allows workers to be run inline in a
+    # testing environment.
+    #
+    # This is similar to `Resque.inline = true` functionality.
+    #
+    # Example:
+    #
+    #   require 'sidekiq/testing/inline'
+    #
+    #   $external_variable = 0
+    #
+    #   class ExternalWorker
+    #     include Sidekiq::Worker
+    #
+    #     def perform
+    #       $external_variable = 1
+    #     end
+    #   end
+    #
+    #   assert_equal 0, $external_variable
+    #   ExternalWorker.perform_async
+    #   assert_equal 1, $external_variable
+    #
+    module ClassMethods
+      alias_method :perform_async_old, :perform_async
+      def perform_async(*args)
+        new.perform(*args)
+        true
+      end
+    end
+  end
+end

test/test_testing_inline.rb

@@ -0,0 +1,75 @@
+require 'helper'
+require 'sidekiq'
+require 'sidekiq/worker'
+require 'active_record'
+require 'action_mailer'
+require 'sidekiq/rails'
+require 'sidekiq/extensions/action_mailer'
+require 'sidekiq/extensions/active_record'
+
+Sidekiq.hook_rails!
+
+class TestInline < MiniTest::Unit::TestCase
+  describe 'sidekiq inline testing' do
+    class InlineError < RuntimeError; end
+
+    class InlineWorker
+      include Sidekiq::Worker
+      def perform(pass)
+        raise InlineError unless pass
+      end
+    end
+
+    class InlineFooMailer < ActionMailer::Base
+      def bar(str)
+        raise InlineError
+      end
+    end
+
+    class InlineFooModel < ActiveRecord::Base
+      def self.bar(str)
+        raise InlineError
+      end
+    end
+
+    before do
+      load 'sidekiq/testing/inline.rb'
+    end
+
+    after do
+      Sidekiq::Worker::ClassMethods.class_eval do
+        remove_method :perform_async
+        alias_method :perform_async, :perform_async_old
+        remove_method :perform_async_old
+      end
+    end
+
+    it 'stubs the async call when in testing mode' do
+      assert InlineWorker.perform_async(true)
+
+      assert_raises InlineError do
+        InlineWorker.perform_async(false)
+      end
+    end
+
+    it 'stubs the delay call on mailers' do
+      assert_raises InlineError do
+        InlineFooMailer.delay.bar('three')
+      end
+    end
+
+    it 'stubs the delay call on models' do
+      assert_raises InlineError do
+        InlineFooModel.delay.bar('three')
+      end
+    end
+
+    it 'stubs the enqueue call when in testing mode' do
+      assert Sidekiq::Client.enqueue(InlineWorker, true)
+
+      assert_raises InlineError do
+        Sidekiq::Client.enqueue(InlineWorker, false)
+      end
+    end
+  end
+end