mirror of
https://github.com/mperham/sidekiq.git
synced 2022-11-09 13:52:34 -05:00
59 lines
1.5 KiB
Ruby
59 lines
1.5 KiB
Ruby
# frozen_string_literal: true
|
|
require_relative 'helper'
|
|
require 'sidekiq/scheduled'
|
|
require 'active_support/core_ext/integer/time'
|
|
|
|
describe 'job scheduling' do
|
|
describe 'middleware' do
|
|
class SomeScheduledWorker
|
|
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 SomeScheduledWorker.perform_in(600, 'mike')
|
|
assert_equal 1, ss.size
|
|
|
|
assert SomeScheduledWorker.perform_in(1.month, 'mike')
|
|
assert_equal 2, ss.size
|
|
|
|
assert SomeScheduledWorker.perform_in(5.days.from_now, 'mike')
|
|
assert_equal 3, ss.size
|
|
|
|
q = Sidekiq::Queue.new("custom_queue")
|
|
qs = q.size
|
|
assert SomeScheduledWorker.perform_in(-300, 'mike')
|
|
assert_equal 3, ss.size
|
|
assert_equal qs+1, q.size
|
|
|
|
assert Sidekiq::Client.push_bulk('class' => SomeScheduledWorker, 'args' => [['mike'], ['mike']], 'at' => 600)
|
|
assert_equal 5, ss.size
|
|
|
|
assert SomeScheduledWorker.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 SomeScheduledWorker.perform_in(1.month, 'mike')
|
|
job = ss.first
|
|
assert job['created_at']
|
|
refute job['enqueued_at']
|
|
end
|
|
end
|
|
|
|
end
|