@@ -174,13 +174,15 @@ def test_complete_callbacks_are_called_on_rack_response_finished
174
174
completed = false
175
175
executor . to_complete { completed = true }
176
176
177
- env = { "rack.response_finished" => [ ] }
177
+ env = Rack ::MockRequest . env_for
178
+ env [ "rack.response_finished" ] = [ ]
179
+
178
180
call_and_return_body ( env )
179
181
180
182
assert_not completed
181
183
182
184
assert_equal 1 , env [ "rack.response_finished" ] . size
183
- env [ "rack.response_finished" ] . first . call ( { } , 200 , { } , nil )
185
+ env [ "rack.response_finished" ] . first . call ( env , 200 , { } , nil )
184
186
185
187
assert completed
186
188
end
@@ -189,7 +191,8 @@ def test_complete_callbacks_are_called_once_on_rack_response_finished_when_excep
189
191
completed_count = 0
190
192
executor . to_complete { completed_count += 1 }
191
193
192
- env = { "rack.response_finished" => [ ] }
194
+ env = Rack ::MockRequest . env_for
195
+ env [ "rack.response_finished" ] = [ ]
193
196
194
197
begin
195
198
call_and_return_body ( env ) do
@@ -199,15 +202,14 @@ def test_complete_callbacks_are_called_once_on_rack_response_finished_when_excep
199
202
end
200
203
201
204
assert_equal 1 , env [ "rack.response_finished" ] . size
202
- env [ "rack.response_finished" ] . first . call ( { } , 200 , { } , nil )
205
+ env [ "rack.response_finished" ] . first . call ( env , 200 , { } , nil )
203
206
204
207
assert_equal 1 , completed_count
205
208
end
206
209
207
210
private
208
- def call_and_return_body ( env = { } , &block )
211
+ def call_and_return_body ( env = Rack :: MockRequest . env_for , &block )
209
212
app = block || proc { [ 200 , { } , [ ] ] }
210
- env = Rack ::MockRequest . env_for ( "" , env )
211
213
_ , _ , body = middleware ( app ) . call ( env )
212
214
body
213
215
end
0 commit comments