resque_manager 3.3.0 → 3.3.1

data/test/unit/overrides/resque_scheduler/resque_scheduler_test.rb

@@ -0,0 +1,104 @@
+require 'test/unit'
+require File.expand_path(File.dirname(__FILE__) + '/../../../test_helper')
+
+class DummyClass
+  include ResqueScheduler
+end
+
+class ResqueSchedulerTest < Test::Unit::TestCase
+  context 'ResqueSchedule' do
+    setup { @dummy = DummyClass.new }
+
+    context '#schedule=' do
+      should 'always raise a RunTimeError' do
+        err = assert_raise(RuntimeError) { @dummy.schedule = {} }
+        assert_match /not implemented/, err.message
+      end
+    end
+
+    context '#schedule' do
+      should 'return a hash' do
+        hash = { make_tea: { every: '1m' }, about: { name: 'green' } }
+        Resque.stubs(:list_range).returns([hash])
+        hash_response = @dummy.schedule
+        assert_equal hash[:make_tea],  hash_response[:make_tea]
+        assert_equal hash[:about],     hash_response[:about]
+      end
+    end
+
+    context '.start' do
+      should 'run rake task resque:scheduler when Rails.env is test' do
+        Thread.expects(:new)
+        ResqueScheduler.start('0.0.0.0')
+      end
+
+      should 'run rake task resque:scheduler when Rails.env is anything other than test or development' do
+        Rails.expects(:env).returns('prod')
+        Thread.expects(:new).with('0.0.0.0')
+        ResqueScheduler.start('0.0.0.0')
+      end
+    end
+
+    context '.quit' do
+      should 'run the rake task resque:quit scheduler when Rails.env is anything other than test or development' do
+        ResqueScheduler.expects(:system).with('rake resque:quit_scheduler')
+        ResqueScheduler.quit('0.0.0.0')
+      end
+
+      should 'run the rake task resque:quit scheduler when Rails.env is test' do
+        Rails.expects(:env).returns('prod').at_least_once
+        ResqueScheduler.expects(:system).with("cd #{Rails.root}; bundle exec cap #{Rails.env} resque:quit_scheduler host=0.0.0.0")
+        ResqueScheduler.quit('0.0.0.0')
+      end
+    end
+
+    context '.restart' do
+      should 'run quit then start' do
+        ip = '0.0.0.0'
+        ResqueScheduler.expects(:quit).with(ip)
+        ResqueScheduler.expects(:start).with(ip)
+        ResqueScheduler.restart(ip)
+      end
+    end
+
+    context '.farm_status' do
+      should 'set the local host status to Stopped' do
+        status = ResqueScheduler.farm_status
+        assert_equal 'Stopped', status['localhost']
+      end
+
+      should 'set the local host status to running' do
+        ResqueScheduler.stubs(:pids).returns('pid')
+        status = ResqueScheduler.farm_status
+        assert_equal 'Running', status['localhost']
+      end
+      
+      should 'set the status to stopped for ip 0.0.0.0 when cap ' do
+        ResqueScheduler.expects(:`).returns('')
+        Rails.expects(:env).returns('prod').at_least_once
+        Resque.expects(:schedule).returns({ job: { 'ip' => '0.0.0.0' } }).at_least_once
+        status = ResqueScheduler.farm_status
+        assert_equal 'Stopped', status['0.0.0.0']
+      end
+
+      should 'set the status to running for ip 0.0.0.0' do
+        ResqueScheduler.expects(:`).returns('resque:scheduler is up')
+        Rails.expects(:env).returns('prod').at_least_once
+        Resque.expects(:schedule).returns({ job: { 'ip' => '0.0.0.0' } }).at_least_once
+        status = ResqueScheduler.farm_status
+        assert_equal 'Running', status['0.0.0.0']
+      end
+    end
+
+    context '.pids' do
+      should 'return an array of pids' do
+        pids = <<-eos
+          123
+          1234
+        eos
+        ResqueScheduler.expects(:`).returns(pids)
+        assert_equal %w(123 1234), ResqueScheduler.pids
+      end
+    end
+  end
+end

data/test/unit/overrides/resque_status/chained_status_test.rb

@@ -0,0 +1,63 @@
+require 'test/unit'
+require File.expand_path(File.dirname(__FILE__) + '/../../../test_helper')
+
+# Testing chained status mixin through SingleRecordLoader
+
+class ChainedStatusTest < Test::Unit::TestCase
+  context 'Resque::Plugins::ChainedStatus' do
+    setup { @uuid   = Resque::Plugins::Status::Hash.generate_uuid }
+
+    context '#included' do
+      should 'have Resque::Plugins::Status and InstanceOverrides included' do
+        assert_includes SingleRecordLoader.included_modules, Resque::Plugins::ChainedStatus::InstanceOverrides, SingleRecordLoader.included_modules.inspect
+        assert_includes SingleRecordLoader.included_modules, Resque::Plugins::Status, SingleRecordLoader.included_modules.inspect
+        assert_includes SingleRecordLoader.included_modules, Resque::Plugins::ChainedStatus, SingleRecordLoader.included_modules.inspect
+      end
+    end
+
+    context 'InstanceOverrides' do
+      setup do
+        @worker = Resque::Worker.new(:data_contribution_file)
+        @single_record_loader  = SingleRecordLoader.new(@uuid, @worker)
+      end
+
+      context '#name' do
+        should 'return nil for no status.name' do
+          assert_nil @single_record_loader.name
+        end
+
+        should 'return the status name' do
+          @single_record_loader.stubs(:status).returns(Resque::Plugins::Status::Hash.new().merge('uuid' => @uuid, 'name' => 'single_record_loader'))
+          assert_equal 'single_record_loader', @single_record_loader.name, @single_record_loader.name.inspect
+        end
+      end
+
+      context '#completed' do
+        should 'add custom messages' do
+          response = @single_record_loader.completed(message: 'test', message2: 'testing')
+          assert_equal 'test',    response.last[:message], response.inspect
+          assert_equal 'testing', response.last[:message2], response.inspect
+        end
+      end
+    end
+
+    context 'ClassOverrides' do
+      context '.enqueu_to' do
+        should 'raise an ArgumentError for a missing UUID' do
+          assert_raises(ArgumentError) { SingleRecordLoader.enqueue_to(:data_contribution_file, 'SingleRecordLoader') }
+        end
+
+        should 'return a uuid and call Resque.enqueue_to' do
+          Resque.expects(:enqueue_to).with(:data_contribution_file, 'SingleRecordLoader', @uuid, { 'uuid' => @uuid }).returns(true)
+          assert_equal @uuid,  SingleRecordLoader.enqueue_to(:data_contribution_file, 'SingleRecordLoader', { 'uuid' => @uuid })
+        end
+
+        should 'return nil' do
+          Resque.expects(:enqueue_to).with(:data_contribution_file, 'SingleRecordLoader', @uuid, { 'uuid' => @uuid }).returns(false)
+          Resque::Plugins::Status::Hash.expects(:remove).with(@uuid)
+          assert_nil SingleRecordLoader.enqueue_to(:data_contribution_file, 'SingleRecordLoader', { 'uuid' => @uuid })
+        end
+      end
+    end
+  end
+end

data/test/unit/overrides/resque_status/status_test.rb

@@ -0,0 +1,193 @@
+require 'test/unit'
+require File.expand_path(File.dirname(__FILE__) + '/../../../test_helper')
+# Testing resque status mixin through DataContributionFile
+
+class StatusTest < Test::Unit::TestCase
+  context 'Resque::Plugins::Status' do
+    setup do
+      Resque.redis.flushdb # flush redis
+      @uuid   = Resque::Plugins::Status::Hash.generate_uuid
+      @worker = Resque::Worker.new(:data_contribution_file)
+    end
+
+    context 'base include' do
+      should 'have the attr_reader :workers' do
+        data_contribution_file = DataContributionFile.new(@uuid)
+        assert_includes data_contribution_file.instance_variable_names, '@worker', data_contribution_file.instance_variable_names.inspect
+      end
+    end
+    
+    context '#initialize' do
+      should 'set the instance variables @uuid, @options, @worker' do
+        options = { options: 'option' }
+        data_contribution_file = DataContributionFile.new(@uuid, @worker, options)
+
+        assert_equal options,  data_contribution_file.options
+        assert_equal @uuid,    data_contribution_file.uuid
+        assert_equal @worker,  data_contribution_file.worker
+      end
+    end
+    
+    context '.enqueue_to' do
+      should 'return a uuid' do
+        assert_not_nil DataContributionFile.enqueue_to(:data_contribution_file, 'SingleRecordLoader')
+      end
+
+      should 'return nil' do
+        Resque.expects(:enqueue_to).returns(false)
+        assert_nil DataContributionFile.enqueue_to(:data_contribution_file, 'SingleRecordLoader')
+      end
+    end
+
+    context '.perform' do
+      should 'return an instance of DataContributionFile with a UUID' do
+        response = DataContributionFile.perform
+        assert_kind_of DataContributionFile, response
+        assert response.uuid.present?
+      end
+
+      should 'set options, worker and UUID' do
+        options  = { options: 'Option' }
+        response = DataContributionFile.perform(@uuid, options) do
+          :single_record_loader
+        end
+        assert_equal @uuid,                 response.uuid
+        assert_equal options,               response.options
+        assert_equal :single_record_loader, response.worker
+      end
+    end
+
+    context '.counter_key' do
+      should 'return a formatted counter key' do
+        assert_equal "data_contribution:#{@uuid}", DataContributionFile.counter_key('data_contribution', @uuid)
+      end
+    end
+
+    context '.remove' do
+      should 'remove from redis' do
+        assert_equal 1, DataContributionFile.incr_counter('data_contribution', @uuid)
+        DataContributionFile.remove(@uuid)
+        assert_equal 0, DataContributionFile.counter('data_contribution', @uuid)
+      end
+    end
+
+    context '.counter' do
+      should 'return a count of 0' do
+        assert_equal 0, DataContributionFile.counter('data_contribution', @uuid)
+      end
+
+      should 'return a count of 1' do
+        DataContributionFile.incr_counter('data_contribution', @uuid)
+        assert_equal 1, DataContributionFile.counter('data_contribution', @uuid)
+      end
+    end
+
+    context '.incr_counter' do
+      should 'increment the counter to one' do
+        assert_equal 1, DataContributionFile.incr_counter('data_contribution', @uuid)
+      end
+    end
+
+    context '#tick' do
+      setup do
+        @data_contribution = DataContributionFile.new(@uuid, @worker)
+        @data_contribution.send(:set_status) # Set the status
+      end
+
+      should 'raise Killed when should_kill? is true' do
+        Resque::Plugins::Status::Hash.expects(:should_kill?).returns(true)
+        assert_raises(Resque::Plugins::Status::Killed) { @data_contribution.tick }
+        end
+
+      should 'raise Killed when status.killed? is true' do
+        # Stub on the hash not the actual status because it will change and wont be stubbed on the right status object
+        Resque::Plugins::Status::Hash.expects(:should_kill?).returns(false)
+        Resque::Plugins::Status::Hash.any_instance.expects(:killed?).returns(true)
+        assert_raises(Resque::Plugins::Status::Killed) { @data_contribution.tick }
+      end
+
+      should 'set status to working' do
+        @data_contribution.status.stubs(:completed?).returns(true)
+        @data_contribution.worker.expects(:paused?).returns(false).at_least_once # break so we dont hit the sleep for 60 seconds
+        @data_contribution.tick
+        assert_equal 'working', @data_contribution.status['status'], @data_contribution.status.inspect
+      end
+    end
+
+    context '#safe_perform!' do
+      setup do
+        @data_contribution = DataContributionFile.new(@uuid, @worker)
+        @status = Resque::Plugins::Status::Hash.new().merge('uuid' => @uuid)
+        @data_contribution.stubs(:status).returns(@status)
+        @now = Time.now
+        Time.stubs(:now).returns(@now)
+      end
+
+      should 'rescue Killed' do
+        # Stub on the hash not the actual status because it will change and wont be stubbed on the right status object
+        Resque::Plugins::Status::Hash.expects(:should_kill?).returns(true)
+        Rails.logger.expects(:info).with("Job #{@data_contribution} Killed at #{@now}")
+        Resque::Plugins::Status::Hash.expects(:killed).with(@uuid)
+        assert_nothing_raised(Resque::Plugins::Status::Killed) { @data_contribution.safe_perform! }
+      end
+
+      should 'rescue an exception and call on failure' do
+        e = Exception.new('exception')
+        Resque::Plugins::Status::Hash.expects(:should_kill?).returns(true)
+        @data_contribution.stubs(:kill!).raises(e)
+        Rails.logger.expects(:error).with(e)
+        @data_contribution.expects(:on_failure).with(e)
+        assert_nothing_raised(Exception) { @data_contribution.safe_perform! }
+      end
+
+      should 'rescue an exception and re-raise when the object does not respond to on failure' do
+        e = Exception.new('exception')
+        Resque::Plugins::Status::Hash.expects(:should_kill?).returns(true)
+        @data_contribution.stubs(:kill!).raises(e)
+        Rails.logger.expects(:error).with(e)
+        @data_contribution.expects(:respond_to?).returns(false)
+        assert_raises(Exception) { @data_contribution.safe_perform! }
+      end
+
+      should 'set the status to working then completed when !status.completed?' do
+        Resque::Plugins::Status::Hash.expects(:should_kill?).returns(false)
+        @status.expects(:failed?).returns(false)
+        @status.expects(:completed?).returns(false)
+        @data_contribution.expects(:set_status).with({'status' => 'working'})
+        @data_contribution.expects(:set_status).with({'status' => 'completed', 'message' => "Completed at #{@now}" })
+        @data_contribution.safe_perform!
+      end
+
+      should 'call on_failure when status.failed?' do
+        @status.expects(:failed?).returns(true)
+        @data_contribution.expects(:on_failure).with(@status.message)
+        @data_contribution.safe_perform!
+      end
+
+      should 'call on_success' do
+        Resque::Plugins::Status::Hash.expects(:should_kill?).returns(false)
+        @status.expects(:failed?).returns(false)
+        @status.expects(:completed?).returns(false)
+        @data_contribution.expects(:on_success)
+        @data_contribution.safe_perform!
+      end
+    end
+
+    context '#pause!' do
+      should 'set the status to paused' do
+        data_contribution = DataContributionFile.new(@uuid, @worker)
+        data_contribution.pause!
+        assert_equal 'paused', data_contribution.status['status'], data_contribution.status.inspect
+        assert_match "#{@worker} paused at", data_contribution.status['message'], data_contribution.status.inspect
+      end
+    end
+
+    context '#overview_message=' do
+      should 'set the overview_message for the worker' do
+        @data_contribution = DataContributionFile.new(@uuid, @worker)
+        @data_contribution.overview_message = 'Test'
+        assert_equal 'Test', @worker.overview_message, @worker.inspect
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: resque_manager
 version: !ruby/object:Gem::Version
-  version: 3.3.0
+  version: 3.3.1
 platform: ruby
 authors:
 - Kevin Tyll
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-08-28 00:00:00.000000000 Z
+date: 2013-10-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -94,6 +94,48 @@ dependencies:
     - - '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: shoulda
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: shoulda-matchers
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: 
 email:
 - kevintyll@gmail.com
@@ -162,6 +204,8 @@ files:
 - test/dummy/app/assets/stylesheets/application.css
 - test/dummy/app/controllers/application_controller.rb
 - test/dummy/app/helpers/application_helper.rb
+- test/dummy/app/models/data_contribution_file.rb
+- test/dummy/app/models/single_record_loader.rb
 - test/dummy/app/views/layouts/application.html.erb
 - test/dummy/config/application.rb
 - test/dummy/config/boot.rb
@@ -172,13 +216,18 @@ files:
 - test/dummy/config/initializers/backtrace_silencers.rb
 - test/dummy/config/initializers/inflections.rb
 - test/dummy/config/initializers/mime_types.rb
+- test/dummy/config/initializers/resque_manager.rb
 - test/dummy/config/initializers/secret_token.rb
 - test/dummy/config/initializers/session_store.rb
 - test/dummy/config/initializers/wrap_parameters.rb
 - test/dummy/config/locales/en.yml
+- test/dummy/config/redis.yml
+- test/dummy/config/resque_manager.yml
 - test/dummy/config/routes.rb
 - test/dummy/config.ru
+- test/dummy/lib/tasks/resque_setup.rake
 - test/dummy/log/development.log
+- test/dummy/log/test.log
 - test/dummy/public/404.html
 - test/dummy/public/422.html
 - test/dummy/public/500.html
@@ -215,12 +264,16 @@ files:
 - test/dummy/tmp/cache/assets/E04/890/sprockets%2F2f5173deea6c795b8fdde723bb4b63af
 - test/dummy/tmp/cache/assets/E2B/7A0/sprockets%2Fd44ef07be0aa6d5b5dea4d37d7f72b4f
 - test/functional/resque_manager/resque_controller_test.rb
-- test/integration/navigation_test.rb
 - test/resque_manager_test.rb
 - test/test_helper.rb
 - test/unit/helpers/resque_manager/resque_helper_test.rb
+- test/unit/overrides/resque/worker_test.rb
+- test/unit/overrides/resque_scheduler/resque_scheduler_test.rb
+- test/unit/overrides/resque_status/chained_status_test.rb
+- test/unit/overrides/resque_status/status_test.rb
 homepage: https://github.com/kevintyll/resque_manager
-licenses: []
+licenses:
+- MIT
 metadata: {}
 post_install_message: 
 rdoc_options: []
@@ -248,6 +301,8 @@ test_files:
 - test/dummy/app/assets/stylesheets/application.css
 - test/dummy/app/controllers/application_controller.rb
 - test/dummy/app/helpers/application_helper.rb
+- test/dummy/app/models/data_contribution_file.rb
+- test/dummy/app/models/single_record_loader.rb
 - test/dummy/app/views/layouts/application.html.erb
 - test/dummy/config/application.rb
 - test/dummy/config/boot.rb
@@ -258,13 +313,18 @@ test_files:
 - test/dummy/config/initializers/backtrace_silencers.rb
 - test/dummy/config/initializers/inflections.rb
 - test/dummy/config/initializers/mime_types.rb
+- test/dummy/config/initializers/resque_manager.rb
 - test/dummy/config/initializers/secret_token.rb
 - test/dummy/config/initializers/session_store.rb
 - test/dummy/config/initializers/wrap_parameters.rb
 - test/dummy/config/locales/en.yml
+- test/dummy/config/redis.yml
+- test/dummy/config/resque_manager.yml
 - test/dummy/config/routes.rb
 - test/dummy/config.ru
+- test/dummy/lib/tasks/resque_setup.rake
 - test/dummy/log/development.log
+- test/dummy/log/test.log
 - test/dummy/public/404.html
 - test/dummy/public/422.html
 - test/dummy/public/500.html
@@ -301,7 +361,10 @@ test_files:
 - test/dummy/tmp/cache/assets/E04/890/sprockets%2F2f5173deea6c795b8fdde723bb4b63af
 - test/dummy/tmp/cache/assets/E2B/7A0/sprockets%2Fd44ef07be0aa6d5b5dea4d37d7f72b4f
 - test/functional/resque_manager/resque_controller_test.rb
-- test/integration/navigation_test.rb
 - test/resque_manager_test.rb
 - test/test_helper.rb
 - test/unit/helpers/resque_manager/resque_helper_test.rb
+- test/unit/overrides/resque/worker_test.rb
+- test/unit/overrides/resque_scheduler/resque_scheduler_test.rb
+- test/unit/overrides/resque_status/chained_status_test.rb
+- test/unit/overrides/resque_status/status_test.rb