1
0
Fork 0
mirror of https://github.com/mperham/sidekiq.git synced 2022-11-09 13:52:34 -05:00
mperham--sidekiq/test/test_manager.rb

90 lines
2.2 KiB
Ruby

require 'helper'
require 'sidekiq/manager'
class TestManager < MiniTest::Unit::TestCase
describe 'manager' do
it 'creates N processor instances' do
mgr = Sidekiq::Manager.new(options)
assert_equal options[:concurrency], mgr.ready.size
assert_equal [], mgr.busy
assert mgr.fetcher
end
it 'fetches upon start' do
mgr = Sidekiq::Manager.new(options)
count = options[:concurrency]
fetch_mock = MiniTest::Mock.new
count.times { fetch_mock.expect(:fetch, nil, []) }
async_mock = MiniTest::Mock.new
count.times { async_mock.expect(:async, fetch_mock, []) }
mgr.fetcher = async_mock
mgr.start
fetch_mock.verify
async_mock.verify
end
it 'assigns work to a processor' do
uow = MiniTest::Mock.new
processor = MiniTest::Mock.new
processor.expect(:async, processor, [])
processor.expect(:process, nil, [uow])
mgr = Sidekiq::Manager.new(options)
mgr.ready << processor
mgr.assign(uow)
assert_equal 1, mgr.busy.size
processor.verify
end
it 'requeues work if stopping' do
uow = MiniTest::Mock.new
uow.expect(:requeue, nil, [])
mgr = Sidekiq::Manager.new(options)
mgr.stop
mgr.assign(uow)
uow.verify
end
it 'shuts down the system' do
mgr = Sidekiq::Manager.new(options)
mgr.stop
assert mgr.busy.empty?
assert mgr.ready.empty?
refute mgr.fetcher.alive?
end
it 'returns finished processors to the ready pool' do
mgr = Sidekiq::Manager.new(options)
init_size = mgr.ready.size
processor = mgr.ready.pop
mgr.busy << processor
mgr.processor_done(processor)
assert_equal 0, mgr.busy.size
assert_equal init_size, mgr.ready.size
end
it 'throws away dead processors' do
mgr = Sidekiq::Manager.new(options)
init_size = mgr.ready.size
processor = mgr.ready.pop
mgr.busy << processor
mgr.processor_died(processor, 'ignored')
assert_equal 0, mgr.busy.size
assert_equal init_size, mgr.ready.size
refute mgr.ready.include?(processor)
end
def options
{ :concurrency => 3, :queues => ['default'] }
end
end
end