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_stats.rb

78 lines
2.3 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.flushdb
end
class DumbWorker
include Sidekiq::Worker
def perform(arg)
raise 'bang' if arg == nil
end
end
it 'updates global stats in the success case' do
msg = { '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])
# adds to the workers set upon initialize
set = conn.smembers('workers')
assert_equal 1, set.size
assert_match(/#{Regexp.escape(`hostname`.strip)}/, set.first)
assert_equal 0, conn.get('stat:failed').to_i
assert_equal 0, conn.get('stat:processed').to_i
assert_equal 0, conn.get("stat:processed:#{processor}").to_i
processor.process(msg, 'xyzzy')
processor.process(msg, 'xyzzy')
processor.process(msg, 'xyzzy')
set = conn.smembers('workers')
assert_equal 1, set.size
assert_match(/#{Regexp.escape(`hostname`.strip)}/, set.first)
assert_equal 0, conn.get('stat:failed').to_i
assert_equal 3, conn.get('stat:processed').to_i
assert_equal 3, conn.get("stat:processed:#{processor}").to_i
end
end
it 'updates global stats in the error case' do
msg = { 'class' => DumbWorker.to_s, 'args' => [nil] }
boss = MiniTest::Mock.new
assert_equal [], @redis.smembers('workers')
assert_equal 0, @redis.get('stat:failed').to_i
assert_equal 0, @redis.get('stat:processed').to_i
processor = Sidekiq::Processor.new(boss)
assert_equal 1, @redis.smembers('workers').size
pstr = processor.to_s
assert_raises RuntimeError do
processor.process(msg, 'xyzzy')
end
set = @redis.smembers('workers')
assert_equal 0, set.size
assert_equal 1, @redis.get('stat:failed').to_i
assert_equal 1, @redis.get('stat:processed').to_i
assert_equal nil, @redis.get("stat:processed:#{pstr}")
end
end
end