-
Notifications
You must be signed in to change notification settings - Fork 417
/
scheduled_task_spec.rb
259 lines (217 loc) · 6.71 KB
/
scheduled_task_spec.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
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
require 'spec_helper'
require 'timecop'
require_relative 'obligation_shared'
require_relative 'runnable_shared'
module Concurrent
describe ScheduledTask do
context 'behavior' do
# runnable
subject { ScheduledTask.new(0.5){ nil } }
it_should_behave_like :runnable
# obligation
let!(:fulfilled_value) { 10 }
let!(:rejected_reason) { StandardError.new('mojo jojo') }
let(:pending_subject) do
task = ScheduledTask.new(1){ fulfilled_value }
task.run!
task
end
let(:fulfilled_subject) do
task = ScheduledTask.new(0.1){ fulfilled_value }
task.run
task
end
let(:rejected_subject) do
task = ScheduledTask.new(0.1){ raise rejected_reason }
task.run
task
end
it_should_behave_like :obligation
end
context '#initialize' do
it 'accepts a number of seconds (from now) as the shcedule time' do
Timecop.freeze do
now = Time.now
task = ScheduledTask.new(60){ nil }
task.schedule_time.to_i.should eq now.to_i + 60
end
end
it 'accepts a time object as the schedule time' do
schedule = Time.now + (60*10)
task = ScheduledTask.new(schedule){ nil }
task.schedule_time.should eq schedule
end
it 'raises an exception when seconds is less than zero' do
expect {
ScheduledTask.new(-1){ nil }
}.to raise_error(ArgumentError)
end
it 'raises an exception when schedule time is in the past' do
expect {
ScheduledTask.new(Time.now - 60){ nil }
}.to raise_error(ArgumentError)
end
it 'raises an exception when no block given' do
expect {
ScheduledTask.new(1)
}.to raise_error(ArgumentError)
end
it 'sets the initial state to :pending' do
task = ScheduledTask.new(1){ nil }
task.should be_pending
end
end
context '#cancel' do
it 'returns false if the task has already been performed' do
task = ScheduledTask.new(0.1){ 42 }
task.run!
sleep(0.2)
task.cancel.should be_false
end
it 'returns false if the task is already in progress' do
task = ScheduledTask.new(0.1){ sleep(1); 42 }
task.run!
sleep(0.2)
task.cancel.should be_false
end
it 'cancels the task if it has not yet started' do
@expected = true
task = ScheduledTask.new(0.3){ @expected = false }
task.run!
sleep(0.1)
task.cancel
sleep(0.5)
@expected.should be_true
end
it 'returns true on success' do
task = ScheduledTask.new(0.3){ @expected = false }
task.run!
sleep(0.1)
task.cancel.should be_true
end
it 'sets the state to :cancelled when cancelled' do
task = ScheduledTask.new(10){ 42 }
task.run!
sleep(0.1)
task.cancel
task.should be_cancelled
end
it 'stops the runnable' do
task = ScheduledTask.new(0.2){ 42 }
task.run!
sleep(0.1)
task.cancel
sleep(0.2)
task.should_not be_running
end
end
context 'execution' do
it 'sets the state to :in_progress when the task is running' do
task = ScheduledTask.new(0.1){ sleep(1); 42 }
task.run!
sleep(0.2)
task.should be_in_progress
end
it 'stops itself on task completion' do
task = ScheduledTask.new(0.1){ 42 }
task.run!
sleep(0.2)
task.should_not be_running
end
end
context 'observation' do
let(:clazz) do
Class.new do
attr_reader :value
attr_reader :reason
attr_reader :count
define_method(:update) do |time, value, reason|
@count = @count.to_i + 1
@value = value
@reason = reason
end
end
end
let(:observer) { clazz.new }
it 'returns true for an observer added while :pending' do
task = ScheduledTask.new(1){ 42 }
task.run!
task.add_observer(observer).should be_true
end
it 'returns true for an observer added while :in_progress' do
task = ScheduledTask.new(0.1){ sleep(1); 42 }
task.run!
sleep(0.2)
task.add_observer(observer).should be_true
end
it 'returns true for an observer added while not running' do
task = ScheduledTask.new(1){ 42 }
task.add_observer(observer).should be_true
end
it 'returns false for an observer added once :cancelled' do
task = ScheduledTask.new(1){ 42 }
task.run!
sleep(0.1)
task.cancel
sleep(0.1)
task.add_observer(observer).should be_false
end
it 'returns false for an observer added once :fulfilled' do
task = ScheduledTask.new(0.1){ 42 }
task.run!
sleep(0.2)
task.add_observer(observer).should be_false
end
it 'returns false for an observer added once :rejected' do
task = ScheduledTask.new(0.1){ raise StandardError }
task.run!
sleep(0.2)
task.add_observer(observer).should be_false
end
it 'notifies all observers on fulfillment' do
task = ScheduledTask.new(0.1){ 42 }
task.add_observer(observer)
task.run!
sleep(0.2)
task.value.should == 42
task.reason.should be_nil
observer.value.should == 42
observer.reason.should be_nil
end
it 'notifies all observers on rejection' do
task = ScheduledTask.new(0.1){ raise StandardError }
task.add_observer(observer)
task.run!
sleep(0.2)
task.value.should be_nil
task.reason.should be_a(StandardError)
observer.value.should be_nil
observer.reason.should be_a(StandardError)
end
it 'does not notify an observer added after fulfillment' do
observer.should_not_receive(:update).with(any_args())
task = ScheduledTask.new(0.1){ 42 }
sleep(0.2)
task.add_observer(observer)
sleep(0.1)
end
it 'does not notify an observer added after rejection' do
observer.should_not_receive(:update).with(any_args())
task = ScheduledTask.new(0.1){ raise StandardError }
sleep(0.2)
task.add_observer(observer)
sleep(0.1)
end
it 'does not notify an observer added after cancellation' do
observer.should_not_receive(:update).with(any_args())
task = ScheduledTask.new(0.5){ 42 }
task.run!
sleep(0.1)
task.cancel
sleep(0.1)
task.add_observer(observer)
sleep(0.5)
end
end
end
end