mirror of
https://github.com/mperham/sidekiq.git
synced 2022-11-09 13:52:34 -05:00
c93a75802c
Useful for other libraries, like hirefire. Implemented specifically to clean up a6de8329. Will close #386 when merged.
103 lines
2.8 KiB
Ruby
103 lines
2.8 KiB
Ruby
require 'helper'
|
|
require 'sidekiq'
|
|
require 'sidekiq/processor'
|
|
|
|
class TestStats < MiniTest::Unit::TestCase
|
|
describe 'with redis' do
|
|
before do
|
|
@redis = Sidekiq.redis = REDIS
|
|
Sidekiq.redis {|c| c.flushdb }
|
|
end
|
|
|
|
class DumbWorker
|
|
include Sidekiq::Worker
|
|
sidekiq_options :queue => 'dumbq'
|
|
|
|
def perform(arg)
|
|
raise 'bang' if arg == nil
|
|
end
|
|
end
|
|
|
|
it 'updates global stats in the success case' do
|
|
msg = Sidekiq.dump_json({ 'class' => DumbWorker.to_s, 'args' => [""] })
|
|
boss = MiniTest::Mock.new
|
|
|
|
@redis.with do |conn|
|
|
|
|
set = conn.smembers('workers')
|
|
assert_equal 0, set.size
|
|
|
|
processor = Sidekiq::Processor.new(boss)
|
|
boss.expect(:processor_done!, nil, [processor])
|
|
boss.expect(:processor_done!, nil, [processor])
|
|
boss.expect(:processor_done!, nil, [processor])
|
|
|
|
assert_equal 0, Sidekiq::Stats.failed
|
|
assert_equal 0, Sidekiq::Stats.processed
|
|
|
|
processor.process(msg, 'xyzzy')
|
|
processor.process(msg, 'xyzzy')
|
|
processor.process(msg, 'xyzzy')
|
|
|
|
assert_equal 0, Sidekiq::Stats.failed
|
|
assert_equal 3, Sidekiq::Stats.processed
|
|
end
|
|
end
|
|
|
|
it 'updates global stats in the error case' do
|
|
msg = Sidekiq.dump_json({ 'class' => DumbWorker.to_s, 'args' => [nil] })
|
|
boss = MiniTest::Mock.new
|
|
|
|
@redis.with do |conn|
|
|
assert_equal [], conn.smembers('workers')
|
|
assert_equal 0, Sidekiq::Stats.failed
|
|
assert_equal 0, Sidekiq::Stats.processed
|
|
|
|
processor = Sidekiq::Processor.new(boss)
|
|
|
|
pstr = processor.to_s
|
|
assert_raises RuntimeError do
|
|
processor.process(msg, 'xyzzy')
|
|
end
|
|
|
|
assert_equal 1, Sidekiq::Stats.failed
|
|
assert_equal 1, Sidekiq::Stats.processed
|
|
end
|
|
end
|
|
|
|
describe "counting methods" do
|
|
before do
|
|
@redis.with do |conn|
|
|
conn.rpush 'queue:foo', '{}'
|
|
conn.sadd 'queues', 'foo'
|
|
|
|
conn.rpush 'queue:bar', '{}'
|
|
conn.rpush 'queue:bar', '{}'
|
|
conn.sadd 'queues', 'bar'
|
|
end
|
|
end
|
|
|
|
describe "queues_with_counts" do
|
|
it "returns queue names and corresponding job counts" do
|
|
assert_equal [["foo", 1], ["bar", 2]], Sidekiq::Stats.queues_with_sizes
|
|
end
|
|
end
|
|
|
|
describe "backlog" do
|
|
it "returns count of all jobs yet to be processed" do
|
|
assert_equal 3, Sidekiq::Stats.backlog
|
|
end
|
|
end
|
|
|
|
describe "size" do
|
|
it "returns size of queues" do
|
|
assert_equal 1, Sidekiq::Stats.size(:foo)
|
|
assert_equal 1, Sidekiq::Stats.size("foo")
|
|
assert_equal 3, Sidekiq::Stats.size("foo", "bar")
|
|
assert_equal 3, Sidekiq::Stats.size
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|