1 files changed, 30 insertions, 6 deletions
diff --git a/railties/test/rails_info_controller_test.rb b/railties/test/rails_info_controller_test.rb
index 8d61af4972..c51503c2b7 100644
--- a/railties/test/rails_info_controller_test.rb
+++ b/railties/test/rails_info_controller_test.rb
@@ -1,5 +1,4 @@
 require 'abstract_unit'
-require 'mocha/setup' # FIXME: stop using mocha
 
 module ActionController
   class Base
@@ -15,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)
+    Rails::InfoController.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
@@ -54,4 +53,29 @@ class InfoControllerTest < ActionController::TestCase
     assert_response :success
   end
 
+  test "info controller returns exact matches" do
+    exact_count = -> { JSON(response.body)['exact'].size }
+
+    get :routes, params: { path: 'rails/info/route' }
+    assert exact_count.call == 0, 'should not match incomplete routes'
+
+    get :routes, params: { path: 'rails/info/routes' }
+    assert exact_count.call == 1, 'should match complete routes'
+
+    get :routes, params: { path: 'rails/info/routes.html' }
+    assert exact_count.call == 1, 'should match complete routes with optional parts'
+  end
+
+  test "info controller returns fuzzy matches" do
+    fuzzy_count = -> { JSON(response.body)['fuzzy'].size }
+
+    get :routes, params: { path: 'rails/info' }
+    assert fuzzy_count.call == 2, 'should match incomplete routes'
+
+    get :routes, params: { path: 'rails/info/routes' }
+    assert fuzzy_count.call == 1, 'should match complete routes'
+
+    get :routes, params: { path: 'rails/info/routes.html' }
+    assert fuzzy_count.call == 0, 'should match optional parts of route literally'
+  end
 end