add a test for prepending SCRIPT_NAME to generated urls

1 files changed, 88 insertions, 0 deletions

diff --git a/railties/test/path_generation_test.rb b/railties/test/path_generation_test.rb
new file mode 100644
index 0000000000..13bf29d3c3
--- /dev/null
+++ b/railties/test/path_generation_test.rb
@@ -0,0 +1,88 @@
+# encoding: utf-8
+require 'abstract_unit'
+require 'active_support/core_ext/object/with_options'
+require 'active_support/core_ext/object/json'
+require 'rails'
+require 'rails/application'
+
+ROUTING = ActionDispatch::Routing
+
+class PathGenerationTest < ActiveSupport::TestCase
+  attr_reader :app
+
+  class TestSet < ROUTING::RouteSet
+    def initialize(block)
+      @block = block
+      super()
+    end
+
+    class Dispatcher < ROUTING::RouteSet::Dispatcher
+      def initialize(defaults, set, block)
+        super(defaults)
+        @block = block
+        @set = set
+      end
+
+      def controller_reference(controller_param)
+        block = @block
+        set = @set
+        Class.new(ActionController::Base) {
+          include set.url_helpers
+          define_method(:process) { |name| block.call(self) }
+          def to_a; [200, {}, []]; end
+        }
+      end
+    end
+
+    def dispatcher defaults
+      TestSet::Dispatcher.new defaults, self, @block
+    end
+  end
+
+  def send_request(uri_or_host, method, path, script_name = nil)
+    host = uri_or_host.host unless path
+    path ||= uri_or_host.path
+
+    params = {'PATH_INFO'      => path,
+              'REQUEST_METHOD' => method,
+              'HTTP_HOST'      => host }
+
+    params['SCRIPT_NAME'] = script_name if script_name
+
+    status, headers, body = app.call(params)
+    new_body = []
+    body.each { |part| new_body << part }
+    body.close if body.respond_to? :close
+    [status, headers, new_body]
+  end
+
+  def test_original_script_name
+    original_logger = Rails.logger
+    Rails.logger    = Logger.new nil
+
+    app = Class.new(Rails::Application) {
+      attr_accessor :controller
+      def initialize
+        super
+        app = self
+        @routes = TestSet.new ->(c) { app.controller = c }
+        secrets.secret_key_base = "foo"
+        secrets.secret_token = "foo"
+      end
+      def app; routes; end
+    }
+
+    @app = app
+    app.routes.draw { resource :blogs }
+
+    url = URI("http://example.org/blogs")
+
+    send_request(url, 'GET', nil, '/FOO')
+    assert_equal '/FOO/blogs', app.instance.controller.blogs_path
+
+    send_request(url, 'GET', nil)
+    assert_equal '/blogs', app.instance.controller.blogs_path
+  ensure
+    Rails.logger = original_logger
+  end
+end