/
test_unique_jobs.rb
83 lines (69 loc) · 2.62 KB
/
test_unique_jobs.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
require 'securerandom'
require 'helper'
require 'sidekiq/client'
require 'sidekiq/worker'
require 'sidekiq/processor'
require 'sidekiq-middleware'
class TestUniqueJobs < MiniTest::Unit::TestCase
describe 'with real redis' do
before do
@boss = MiniTest::Mock.new
@processor = ::Sidekiq::Processor.new(@boss)
Sidekiq.redis = REDIS
Sidekiq.redis {|c| c.flushdb }
end
class UniqueWorker
include Sidekiq::Worker
sidekiq_options queue: :unique_queue, unique: true
def perform(x)
end
end
it 'does not duplicate messages with enabled unique option' do
5.times { UniqueWorker.perform_async('args') }
assert_equal 1, Sidekiq.redis { |c| c.llen('queue:unique_queue') }
end
it 'discards non critical information about the message' do
5.times { Sidekiq::Client.push('class' => UniqueWorker, 'args' => ['critical'], 'sent_at' => Time.now.to_f, 'non' => 'critical') }
assert_equal 1, Sidekiq.redis { |c| c.llen('queue:unique_queue') }
end
class NotUniqueWorker
include Sidekiq::Worker
sidekiq_options queue: :not_unique_queue, unique: false
def perform(x)
end
end
it 'duplicates messages with disabled unique option' do
5.times { NotUniqueWorker.perform_async('args') }
assert_equal 5, Sidekiq.redis { |c| c.llen('queue:not_unique_queue') }
end
class UniqueScheduledWorker
include Sidekiq::Worker
sidekiq_options queue: :unique_scheduled_queue, unique: :all, forever: true
def perform(x)
UniqueScheduledWorker.perform_in(60, x)
end
end
it 'does not duplicate scheduled messages with enabled unique option' do
5.times { |t| UniqueScheduledWorker.perform_in((t+1)*60, 'args') }
assert_equal 1, Sidekiq.redis { |c| c.zcard('schedule') }
end
it 'allows the job to reschedule itself with enabled forever option' do
5.times {
msg = Sidekiq.dump_json('class' => UniqueScheduledWorker.to_s, 'args' => ['something'])
@boss.expect(:processor_done!, nil, [@processor])
@processor.process(msg, 'default')
@boss.verify
}
assert_equal 1, Sidekiq.redis { |c| c.zcard('schedule') }
end
it 'discards non critical information about the message' do
5.times {|i|
msg = Sidekiq.dump_json('class' => UniqueScheduledWorker.to_s, 'args' => ['something'], 'sent_at' => (Time.now + i*60).to_f)
@boss.expect(:processor_done!, nil, [@processor])
@processor.process(msg, 'default')
@boss.verify
}
assert_equal 1, Sidekiq.redis { |c| c.zcard('schedule') }
end
end
end