Skip to content

Commit 3628a6f

Browse files
committed
More idiomatic hook definition, tests no longer use globals, and are correctly cased
1 parent 52e61b6 commit 3628a6f

File tree

2 files changed

+20
-18
lines changed

2 files changed

+20
-18
lines changed

lib/resque.rb

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,8 @@ def after_fork(&block)
124124
# The `before_pause` hook will be run in the parent process before the
125125
# worker has paused processing (via #pause_processing or SIGUSR2).
126126
def before_pause(&block)
127-
block ? (@before_pause = block) : @before_pause
127+
@before_pause = block if block_given?
128+
@before_pause
128129
end
129130

130131
# Set the after_pause proc.
@@ -133,7 +134,8 @@ def before_pause(&block)
133134
# The `after_pause` hook will be run in the parent process after the
134135
# worker has paused (via SIGCONT).
135136
def after_pause(&block)
136-
block ? (@after_pause = block) : @after_pause
137+
@after_pause = block if block_given?
138+
@after_pause
137139
end
138140

139141
# Set the after_continue proc.

test/worker_test.rb

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -470,50 +470,50 @@ def self.perform
470470
assert_not_equal original_connection, Resque.redis.client.connection.instance_variable_get("@sock")
471471
end
472472

473-
it "Will call before_pause before it is paused" do
474-
$BEFORE_PAUSE_CALLED = false
475-
$CAPTURED_WORKER = nil
473+
it "will call before_pause before it is paused" do
474+
before_pause_called = false
475+
captured_worker = nil
476476

477477
Resque.before_pause do |worker|
478-
$BEFORE_PAUSE_CALLED = true
479-
$CAPTURED_WORKER = worker
478+
before_pause_called = true
479+
captured_worker = worker
480480
end
481481

482482
@worker.pause_processing
483483

484-
assert !$BEFORE_PAUSE_CALLED
484+
assert !before_pause_called
485485

486486
t = Thread.start { sleep(0.1); Process.kill('CONT', @worker.pid) }
487487

488488
@worker.work(0)
489489

490490
t.join
491491

492-
assert $BEFORE_PAUSE_CALLED
493-
assert_equal @worker, $CAPTURED_WORKER
492+
assert before_pause_called
493+
assert_equal @worker, captured_worker
494494
end
495495

496-
it "Will call after_pause after it is paused" do
497-
$AFTER_PAUSED_CALLED = false
498-
$CAPTURED_WORKER = nil
496+
it "will call after_pause after it is paused" do
497+
after_pause_called = false
498+
captured_worker = nil
499499

500500
Resque.after_pause do |worker|
501-
$AFTER_PAUSED_CALLED = true
502-
$CAPTURED_WORKER = worker
501+
after_pause_called = true
502+
captured_worker = worker
503503
end
504504

505505
@worker.pause_processing
506506

507-
assert !$AFTER_PAUSED_CALLED
507+
assert !after_pause_called
508508

509509
t = Thread.start { sleep(0.1); Process.kill('CONT', @worker.pid) }
510510

511511
@worker.work(0)
512512

513513
t.join
514514

515-
assert $AFTER_PAUSED_CALLED
516-
assert_equal @worker, $CAPTURED_WORKER
515+
assert after_pause_called
516+
assert_equal @worker, captured_worker
517517
end
518518

519519
if !defined?(RUBY_ENGINE) || defined?(RUBY_ENGINE) && RUBY_ENGINE != "jruby"

0 commit comments

Comments
 (0)