mirror of
https://github.com/mperham/sidekiq.git
synced 2022-11-09 13:52:34 -05:00
48 lines
2 KiB
Ruby
48 lines
2 KiB
Ruby
require 'helper'
|
|
require 'sidekiq/scheduled'
|
|
|
|
class TestScheduled < MiniTest::Unit::TestCase
|
|
class ScheduledWorker
|
|
include Sidekiq::Worker
|
|
def perform(x)
|
|
end
|
|
end
|
|
|
|
describe 'poller' do
|
|
before do
|
|
Sidekiq.redis = REDIS
|
|
Sidekiq.redis do |conn|
|
|
conn.flushdb
|
|
end
|
|
end
|
|
|
|
it 'should empty the retry and scheduled queues up to the current time' do
|
|
Sidekiq.redis do |conn|
|
|
error_1 = Sidekiq.dump_json('class' => ScheduledWorker.name, 'args' => ["error_1"], 'queue' => 'queue_1')
|
|
error_2 = Sidekiq.dump_json('class' => ScheduledWorker.name, 'args' => ["error_2"], 'queue' => 'queue_2')
|
|
error_3 = Sidekiq.dump_json('class' => ScheduledWorker.name, 'args' => ["error_3"], 'queue' => 'queue_3')
|
|
future_1 = Sidekiq.dump_json('class' => ScheduledWorker.name, 'args' => ["future_1"], 'queue' => 'queue_4')
|
|
future_2 = Sidekiq.dump_json('class' => ScheduledWorker.name, 'args' => ["future_2"], 'queue' => 'queue_5')
|
|
future_3 = Sidekiq.dump_json('class' => ScheduledWorker.name, 'args' => ["future_3"], 'queue' => 'queue_6')
|
|
|
|
conn.zadd("retry", (Time.now - 60).to_f.to_s, error_1)
|
|
conn.zadd("retry", (Time.now - 50).to_f.to_s, error_2)
|
|
conn.zadd("retry", (Time.now + 60).to_f.to_s, error_3)
|
|
conn.zadd("schedule", (Time.now - 60).to_f.to_s, future_1)
|
|
conn.zadd("schedule", (Time.now - 50).to_f.to_s, future_2)
|
|
conn.zadd("schedule", (Time.now + 60).to_f.to_s, future_3)
|
|
|
|
poller = Sidekiq::Scheduled::Poller.new
|
|
poller.poll
|
|
poller.terminate
|
|
|
|
assert_equal [error_1], conn.lrange("queue:queue_1", 0, -1)
|
|
assert_equal [error_2], conn.lrange("queue:queue_2", 0, -1)
|
|
assert_equal [error_3], conn.zrange("retry", 0, -1)
|
|
assert_equal [future_1], conn.lrange("queue:queue_4", 0, -1)
|
|
assert_equal [future_2], conn.lrange("queue:queue_5", 0, -1)
|
|
assert_equal [future_3], conn.zrange("schedule", 0, -1)
|
|
end
|
|
end
|
|
end
|
|
end
|