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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
require 'helper'
require 'sidekiq'
require 'active_record'
require 'action_mailer'
require 'sidekiq/extensions/action_mailer'
require 'sidekiq/extensions/active_record'
require 'sidekiq/rails'
Sidekiq.hook_rails!
class TestExtensions < Sidekiq::Test
describe 'sidekiq extensions' do
before do
Sidekiq.redis = REDIS
Sidekiq.redis {|c| c.flushdb }
end
class MyModel < ActiveRecord::Base
def self.long_class_method
raise "Should not be called!"
end
end
it 'allows delayed execution of ActiveRecord class methods' do
assert_equal [], Sidekiq::Queue.all.map(&:name)
assert_equal 0, Sidekiq.redis {|c| c.llen('queue:default') }
MyModel.delay.long_class_method
assert_equal ['default'], Sidekiq::Queue.all.map(&:name)
assert_equal 1, Sidekiq.redis {|c| c.llen('queue:default') }
end
it 'uses and stringifies specified options' do
assert_equal [], Sidekiq::Queue.all.map(&:name)
assert_equal 0, Sidekiq.redis {|c| c.llen('queue:notdefault') }
MyModel.delay(queue: :notdefault).long_class_method
assert_equal ['notdefault'], Sidekiq::Queue.all.map(&:name)
assert_equal 1, Sidekiq.redis {|c| c.llen('queue:notdefault') }
end
it 'allows delayed scheduling of AR class methods' do
assert_equal 0, Sidekiq.redis {|c| c.zcard('schedule') }
MyModel.delay_for(5.days).long_class_method
assert_equal 1, Sidekiq.redis {|c| c.zcard('schedule') }
end
it 'allows until delayed scheduling of AR class methods' do
assert_equal 0, Sidekiq.redis {|c| c.zcard('schedule') }
MyModel.delay_until(1.day.from_now).long_class_method
assert_equal 1, Sidekiq.redis {|c| c.zcard('schedule') }
end
class UserMailer < ActionMailer::Base
def greetings(a, b)
raise "Should not be called!"
end
end
it 'allows delayed delivery of ActionMailer mails' do
assert_equal [], Sidekiq::Queue.all.map(&:name)
assert_equal 0, Sidekiq.redis {|c| c.llen('queue:default') }
UserMailer.delay.greetings(1, 2)
assert_equal ['default'], Sidekiq::Queue.all.map(&:name)
assert_equal 1, Sidekiq.redis {|c| c.llen('queue:default') }
end
it 'allows delayed scheduling of AM mails' do
assert_equal 0, Sidekiq.redis {|c| c.zcard('schedule') }
UserMailer.delay_for(5.days).greetings(1, 2)
assert_equal 1, Sidekiq.redis {|c| c.zcard('schedule') }
end
it 'allows until delay scheduling of AM mails' do
assert_equal 0, Sidekiq.redis {|c| c.zcard('schedule') }
UserMailer.delay_until(5.days.from_now).greetings(1, 2)
assert_equal 1, Sidekiq.redis {|c| c.zcard('schedule') }
end
class SomeClass
def self.doit(arg)
end
end
it 'allows delay of any ole class method' do
assert_equal 0, queue_size
SomeClass.delay.doit(Date.today)
assert_equal 1, queue_size
end
module SomeModule
def self.doit(arg)
end
end
it 'allows delay of any module class method' do
assert_equal 0, queue_size
SomeModule.delay.doit(Date.today)
assert_equal 1, queue_size
end
def queue_size(name='default')
Sidekiq::Queue.new(name).size
end
it 'allows removing of the #delay methods' do
Sidekiq.remove_delay!
assert_equal 0, queue_size
assert_raises NoMethodError do
SomeModule.delay.doit(Date.today)
end
Sidekiq.instance_eval { remove_instance_variable :@delay_removed }
# Reload modified modules
load 'sidekiq/extensions/action_mailer.rb'
load 'sidekiq/extensions/active_record.rb'
load 'sidekiq/extensions/generic_proxy.rb'
load 'sidekiq/extensions/class_methods.rb'
end
end
end
|
