mirror of
https://github.com/mperham/sidekiq.git
synced 2022-11-09 13:52:34 -05:00
107 lines
2.8 KiB
Ruby
107 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
|
|
actor = MiniTest::Mock.new
|
|
|
|
@redis.with do |conn|
|
|
|
|
set = conn.smembers('workers')
|
|
assert_equal 0, set.size
|
|
|
|
processor = Sidekiq::Processor.new(boss)
|
|
3.times do
|
|
actor.expect(:processor_done, nil, [processor])
|
|
boss.expect(:async, actor, [])
|
|
end
|
|
|
|
assert_equal 0, Sidekiq.info[:failed]
|
|
assert_equal 0, Sidekiq.info[:processed]
|
|
|
|
processor.process(msg, 'xyzzy')
|
|
processor.process(msg, 'xyzzy')
|
|
processor.process(msg, 'xyzzy')
|
|
|
|
assert_equal 0, Sidekiq.info[:failed]
|
|
assert_equal 3, Sidekiq.info[: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.info[:failed]
|
|
assert_equal 0, Sidekiq.info[:processed]
|
|
|
|
processor = Sidekiq::Processor.new(boss)
|
|
|
|
assert_raises RuntimeError do
|
|
processor.process(msg, 'xyzzy')
|
|
end
|
|
|
|
assert_equal 1, Sidekiq.info[:failed]
|
|
assert_equal 1, Sidekiq.info[:processed]
|
|
end
|
|
end
|
|
|
|
describe "info counts" do
|
|
before do
|
|
@redis.with do |conn|
|
|
conn.rpush 'queue:foo', '{}'
|
|
conn.sadd 'queues', 'foo'
|
|
|
|
3.times { conn.rpush 'queue:bar', '{}' }
|
|
conn.sadd 'queues', 'bar'
|
|
|
|
2.times { conn.rpush 'queue:baz', '{}' }
|
|
conn.sadd 'queues', 'baz'
|
|
end
|
|
end
|
|
|
|
describe "queues_with_sizes" do
|
|
it "returns queue names and corresponding job counts" do
|
|
assert_equal [["foo", 1], ["baz", 2], ["bar", 3]], Sidekiq.info[:queues_with_sizes]
|
|
end
|
|
end
|
|
|
|
describe "backlog" do
|
|
it "returns count of all jobs yet to be processed" do
|
|
assert_equal 6, Sidekiq.info[:backlog]
|
|
end
|
|
end
|
|
|
|
describe "size" do
|
|
it "returns size of queues" do
|
|
assert_equal 0, Sidekiq.size("foox")
|
|
assert_equal 1, Sidekiq.size(:foo)
|
|
assert_equal 1, Sidekiq.size("foo")
|
|
assert_equal 4, Sidekiq.size("foo", "bar")
|
|
assert_equal 6, Sidekiq.size
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|