1 files changed, 5 insertions, 5 deletions
diff --git a/railties/test/rails_info_controller_test.rb b/railties/test/rails_info_controller_test.rb
index 08fcddd4bf..d336a2e6c0 100644
--- a/railties/test/rails_info_controller_test.rb
+++ b/railties/test/rails_info_controller_test.rb
@@ -14,26 +14,26 @@ class InfoControllerTest < ActionController::TestCase
       get '/rails/info/properties' => "rails/info#properties"
       get '/rails/info/routes'     => "rails/info#routes"
     end
-    @controller.stubs(:local_request? => true)
     @routes = Rails.application.routes
 
     Rails::InfoController.send(:include, @routes.url_helpers)
+
+    @request.env["REMOTE_ADDR"] = "127.0.0.1"
   end
 
   test "info controller does not allow remote requests" do
-    @controller.stubs(local_request?: false)
+    @request.env["REMOTE_ADDR"] = "example.org"
     get :properties
     assert_response :forbidden
   end
 
   test "info controller renders an error message when request was forbidden" do
-    @controller.stubs(local_request?: false)
+    @request.env["REMOTE_ADDR"] = "example.org"
     get :properties
     assert_select 'p'
   end
 
   test "info controller allows requests when all requests are considered local" do
-    @controller.stubs(local_request?: true)
     get :properties
     assert_response :success
   end
@@ -50,7 +50,7 @@ class InfoControllerTest < ActionController::TestCase
 
   test "info controller renders with routes" do
     get :routes
-    assert_select 'pre'
+    assert_response :success
   end
 
 end