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
|
# frozen_string_literal: true
require_relative 'helper'
require 'sidekiq/scheduled'
class TestScheduling < Sidekiq::Test
describe 'middleware' do
class ScheduledWorker
include Sidekiq::Worker
sidekiq_options :queue => :custom_queue
def perform(x)
end
end
# Assume we can pass any class as time to perform_in
class TimeDuck
def to_f; 42.0 end
end
it 'schedules jobs' do
ss = Sidekiq::ScheduledSet.new
ss.clear
assert_equal 0, ss.size
assert ScheduledWorker.perform_in(600, 'mike')
assert_equal 1, ss.size
assert ScheduledWorker.perform_in(1.month, 'mike')
assert_equal 2, ss.size
assert ScheduledWorker.perform_in(5.days.from_now, 'mike')
assert_equal 3, ss.size
q = Sidekiq::Queue.new("custom_queue")
qs = q.size
assert ScheduledWorker.perform_in(-300, 'mike')
assert_equal 3, ss.size
assert_equal qs+1, q.size
assert Sidekiq::Client.push_bulk('class' => ScheduledWorker, 'args' => [['mike'], ['mike']], 'at' => 600)
assert_equal 5, ss.size
assert ScheduledWorker.perform_in(TimeDuck.new, 'samwise')
assert_equal 6, ss.size
end
it 'removes the enqueued_at field when scheduling' do
ss = Sidekiq::ScheduledSet.new
ss.clear
assert ScheduledWorker.perform_in(1.month, 'mike')
job = ss.first
assert job['created_at']
refute job['enqueued_at']
end
end
end
|
