resque-mock 0.1.0

data/Rakefile

@@ -0,0 +1,10 @@
+begin
+  require 'rspec/core/rake_task'
+  RSpec::Core::RakeTask.new do |t|
+    t.rspec_opts = %w[ -c ]
+    t.pattern = 'spec/**/*_spec.rb'
+  end
+
+  task :default => :spec
+rescue LoadError
+end

data/lib/resque/mock.rb

@@ -0,0 +1,84 @@
+require 'resque'
+
+module Resque
+  def self.mock!
+    extend MockExt
+  end
+
+  module MockExt
+    def async
+      @async = true
+      create_worker_manager
+      yield
+    ensure
+      wait_for_worker_manager
+      @async = false
+    end
+
+    def enqueue(klass, *args)
+      puts "Mock enqueue: async=#{!!@async}, stack_depth=#{caller.size}, #{klass}, #{args.inspect}" if ENV['VERBOSE']
+      defer(klass, args)
+    end
+
+    def enqueue_in(delay, klass, *args)
+      puts "Mock enqueue in #{delay}: async=#{!!@async}, stack_depth=#{caller.size}, #{klass}, #{args.inspect}" if ENV['VERBOSE']
+      defer(klass, args, delay)
+    end
+
+    def defer(klass, args, delay = nil)
+      if @async
+        add_job('payload' => { 'class' => klass, 'args' => args }, 'delay' => delay)
+      else
+        sleep delay if delay
+        klass.perform(*roundtrip(args))
+      end
+    end
+
+    def create_worker_manager
+      @worker_manager = Thread.new do
+        Thread.current.abort_on_exception = true
+        worker_threads = []
+
+        while true
+          break if Thread.current[:exit] && worker_threads.empty? && Thread.current[:jobs].empty?
+
+          worker_threads.reject! {|t| !t.alive? }
+
+          while Thread.current[:jobs] && job_data = Thread.current[:jobs].shift
+            worker_threads << create_worker_thread_for(job_data)
+          end
+
+          sleep 0.5
+        end
+      end.tap {|t| t[:jobs] = [] }
+    end
+
+    def wait_for_worker_manager
+      @worker_manager[:exit] = true
+      @worker_manager.join
+      @worker_manager = nil
+    end
+
+    def create_worker_thread_for(data)
+      Thread.new(data) do |data|
+        Thread.current.abort_on_exception = true
+        if delay = data['delay']
+          sleep delay
+        end
+
+        klass = data['payload']['class']
+        puts "Mock perform: #{klass}.perform(*#{data['payload']['args'].inspect})" if ENV['VERBOSE']
+        klass.perform(*roundtrip(data['payload']['args']))
+        puts "Mock exit: #{klass}.perform(*#{data['payload']['args'].inspect})" if ENV['VERBOSE']
+      end
+    end
+
+    def roundtrip(args)
+      decode(encode(args))
+    end
+
+    def add_job(data)
+      @worker_manager[:jobs] << data
+    end
+  end
+end

data/spec/mock_spec.rb

@@ -0,0 +1,111 @@
+#require 'spec_helper'
+require 'resque/mock'
+
+Resque.mock!
+
+class Performer
+  def self.run?
+    !!@args
+  end
+
+  def self.args
+    @args
+  end
+
+  def self.runs
+    @runs || 0
+  end
+
+  def self.perform(*args)
+    @runs += 1
+    if Hash === (options = args.first)
+      if runs_left = options['runs']
+        runs_left -= 1
+        if runs_left > 0
+          Resque.enqueue(self, 'runs' => runs_left)
+        end
+      end
+    end
+    @args = args
+  end
+
+  def self.reset!
+    @args = nil
+    @runs = 0
+  end
+end
+
+class BadPerformer < Performer
+  def self.perform(*args)
+    raise 'hello'
+  end
+end
+
+describe Resque do
+  before { Performer.reset! }
+
+  describe "synchronously" do
+    it "performs jobs without delay" do
+      Resque.enqueue(Performer, 'hello', 'there')
+      Performer.should be_run
+      Performer.args.should == ['hello', 'there']
+    end
+
+    it "performs jobs with a delay" do
+      Resque.should_receive(:sleep).with(5)
+      Resque.enqueue_in(5, Performer, 'hello', 'there')
+      Performer.should be_run
+      Performer.args.should == ['hello', 'there']
+    end
+
+    it "can perform more jobs that are queued" do
+      Resque.enqueue(Performer, 'runs' => 3)
+      Performer.runs.should == 3
+    end
+
+    it "roundtrips arguments" do
+      Resque.enqueue(Performer, :hello => :there)
+      Performer.args.should == [{ 'hello' => 'there' }]
+    end
+  end
+
+  describe "asynchronously" do
+    it "performs jobs without delay" do
+      Resque.async do
+        Resque.enqueue(Performer, 'hello', 'there')
+      end
+
+      Performer.should be_run
+      Performer.args.should == ['hello', 'there']
+    end
+
+    it "performs jobs with delay" do
+      # not immediately sure how to mock this
+
+      Resque.async do
+        Resque.enqueue_in(5, Performer, 'hello', 'there')
+      end
+
+      Performer.should be_run
+      Performer.args.should == ['hello', 'there']
+    end
+
+    it "can perform more jobs that are queued" do
+      Resque.async { Resque.enqueue(Performer, 'runs' => 3) }
+      Performer.runs.should == 3
+    end
+
+    it "roundtrips arguments" do
+      Resque.async { Resque.enqueue(Performer, :hello => :there) }
+      Performer.args.should == [{ 'hello' => 'there' }]
+    end
+
+    it "raises errors encountered inside the block" do
+      expect { Resque.async { raise 'hello' } }.to raise_error
+    end
+
+    it "raises errors encountered by jobs" do
+      expect { Resque.async { Resque.enqueue(BadPerformer, 5) } }.to raise_error
+    end
+  end
+end

metadata

@@ -0,0 +1,85 @@
+--- !ruby/object:Gem::Specification
+name: resque-mock
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Dan Peterson
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-06-16 00:00:00.000000000 -07:00
+default_executable: 
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: resque
+  requirement: &2169208800 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2169208800
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: &2169208360 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *2169208360
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &2169207940 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *2169207940
+description: Mock resque with threads
+email:
+- dpiddy@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/resque/mock.rb
+- Rakefile
+- spec/mock_spec.rb
+has_rdoc: true
+homepage: https://github.com/dpiddy/resque-mock
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.5.2
+signing_key: 
+specification_version: 3
+summary: Mock resque with threads
+test_files:
+- Rakefile
+- spec/mock_spec.rb