Skip to content

Commit ef21678

Browse files
committed
Remove unused options and simplify methods for fork supervisor tests
1 parent 93eb0cf commit ef21678

File tree

2 files changed

+19
-21
lines changed

2 files changed

+19
-21
lines changed

test/test_helper.rb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def assert_no_pending_jobs
4747

4848
def run_supervisor_as_fork(**options)
4949
fork do
50-
SolidQueue::Supervisor.start(**options)
50+
SolidQueue::Supervisor.start(mode: :fork, **options)
5151
end
5252
end
5353

test/unit/fork_supervisor_test.rb

Lines changed: 18 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ class ForkSupervisorTest < ActiveSupport::TestCase
2121
wait_for_registered_processes(4)
2222

2323
assert_registered_supervisor(pid)
24-
assert_registered_workers(2, supervisor_pid: pid)
24+
assert_registered_workers(count: 2, supervisor_pid: pid)
2525
assert_registered_dispatcher(supervisor_pid: pid)
2626

2727
terminate_process(pid)
@@ -35,7 +35,7 @@ class ForkSupervisorTest < ActiveSupport::TestCase
3535
wait_for_registered_processes(2) # supervisor + dispatcher
3636

3737
assert_registered_supervisor(pid)
38-
assert_registered_workers(0)
38+
assert_registered_workers(count: 0)
3939
assert_registered_dispatcher(supervisor_pid: pid)
4040

4141
terminate_process(pid)
@@ -46,7 +46,7 @@ class ForkSupervisorTest < ActiveSupport::TestCase
4646
test "create and delete pidfile" do
4747
assert_not File.exist?(@pidfile)
4848

49-
pid = run_supervisor_as_fork(mode: :all)
49+
pid = run_supervisor_as_fork
5050
wait_for_registered_processes(4)
5151

5252
assert File.exist?(@pidfile)
@@ -61,7 +61,7 @@ class ForkSupervisorTest < ActiveSupport::TestCase
6161
FileUtils.mkdir_p(File.dirname(@pidfile))
6262
File.write(@pidfile, ::Process.pid.to_s)
6363

64-
pid = run_supervisor_as_fork(mode: :all)
64+
pid = run_supervisor_as_fork
6565
wait_for_registered_processes(4)
6666

6767
assert File.exist?(@pidfile)
@@ -71,7 +71,7 @@ class ForkSupervisorTest < ActiveSupport::TestCase
7171
end
7272

7373
test "delete previous pidfile if the owner is dead" do
74-
pid = run_supervisor_as_fork(mode: :all)
74+
pid = run_supervisor_as_fork
7575
wait_for_registered_processes(4)
7676

7777
terminate_process(pid, signal: :KILL)
@@ -81,7 +81,7 @@ class ForkSupervisorTest < ActiveSupport::TestCase
8181

8282
wait_for_registered_processes(0)
8383

84-
pid = run_supervisor_as_fork(mode: :all)
84+
pid = run_supervisor_as_fork
8585
wait_for_registered_processes(4)
8686

8787
assert File.exist?(@pidfile)
@@ -121,24 +121,22 @@ class ForkSupervisorTest < ActiveSupport::TestCase
121121
end
122122

123123
private
124-
def assert_registered_workers(count, supervisor_pid: nil, **metadata)
125-
skip_active_record_query_cache do
126-
workers = find_processes_registered_as("Worker")
127-
assert_equal count, workers.count
124+
def assert_registered_workers(supervisor_pid: nil, count: 1)
125+
assert_registered_processes(kind: "Worker", count: count, supervisor_pid: supervisor_pid)
126+
end
128127

129-
workers.each do |process|
130-
assert_equal supervisor_pid, process.supervisor.pid
131-
assert metadata < process.metadata.symbolize_keys
132-
end
133-
end
128+
def assert_registered_dispatcher(supervisor_pid: nil)
129+
assert_registered_processes(kind: "Dispatcher", count: 1, supervisor_pid: supervisor_pid)
134130
end
135131

136-
def assert_registered_dispatcher(supervisor_pid:, **metadata)
132+
def assert_registered_processes(kind:, supervisor_pid:, count: 1)
137133
skip_active_record_query_cache do
138-
processes = find_processes_registered_as("Dispatcher")
139-
assert_equal 1, processes.count
140-
assert_equal supervisor_pid, processes.first.supervisor.pid
141-
assert metadata < processes.first.metadata.symbolize_keys
134+
processes = find_processes_registered_as(kind)
135+
assert_equal count, processes.count
136+
137+
processes.each do |process|
138+
assert_equal supervisor_pid, process.supervisor.pid
139+
end
142140
end
143141
end
144142

0 commit comments

Comments
 (0)