2016-08-06 17:16:09 +00:00
|
|
|
require "isolation/abstract_unit"
|
2010-01-05 01:34:42 +00:00
|
|
|
|
|
|
|
module ApplicationTests
|
2012-01-06 01:30:17 +00:00
|
|
|
class MiddlewareTest < ActiveSupport::TestCase
|
2010-01-05 01:34:42 +00:00
|
|
|
include ActiveSupport::Testing::Isolation
|
|
|
|
|
|
|
|
def setup
|
|
|
|
build_app
|
|
|
|
FileUtils.rm_rf "#{app_path}/config/environments"
|
|
|
|
end
|
|
|
|
|
2011-06-06 12:54:05 +00:00
|
|
|
def teardown
|
|
|
|
teardown_app
|
|
|
|
end
|
|
|
|
|
2010-04-26 07:04:04 +00:00
|
|
|
def app
|
|
|
|
@app ||= Rails.application
|
|
|
|
end
|
|
|
|
|
2010-01-05 01:34:42 +00:00
|
|
|
test "default middleware stack" do
|
2013-07-15 08:47:59 +00:00
|
|
|
add_to_config "config.active_record.migration_error = :page_load"
|
|
|
|
|
2010-01-05 01:34:42 +00:00
|
|
|
boot!
|
|
|
|
|
2010-09-15 19:58:49 +00:00
|
|
|
assert_equal [
|
2012-03-03 19:05:45 +00:00
|
|
|
"Rack::Sendfile",
|
2010-09-15 19:58:49 +00:00
|
|
|
"ActionDispatch::Static",
|
2016-02-22 01:25:52 +00:00
|
|
|
"ActionDispatch::Executor",
|
2015-10-02 21:45:31 +00:00
|
|
|
"ActiveSupport::Cache::Strategy::LocalCache",
|
2015-10-03 21:21:31 +00:00
|
|
|
"Rack::Runtime",
|
|
|
|
"Rack::MethodOverride",
|
2011-10-20 07:29:27 +00:00
|
|
|
"ActionDispatch::RequestId",
|
2016-12-30 21:50:35 +00:00
|
|
|
"ActionDispatch::RemoteIp",
|
|
|
|
"Rails::Rack::Logger",
|
2010-09-15 19:58:49 +00:00
|
|
|
"ActionDispatch::ShowExceptions",
|
2011-12-01 19:46:18 +00:00
|
|
|
"ActionDispatch::DebugExceptions",
|
2010-11-24 02:04:05 +00:00
|
|
|
"ActionDispatch::Reloader",
|
2010-09-15 19:58:49 +00:00
|
|
|
"ActionDispatch::Callbacks",
|
2013-07-15 08:47:59 +00:00
|
|
|
"ActiveRecord::Migration::CheckPending",
|
2010-09-15 19:58:49 +00:00
|
|
|
"ActionDispatch::Cookies",
|
|
|
|
"ActionDispatch::Session::CookieStore",
|
|
|
|
"ActionDispatch::Flash",
|
2012-07-23 16:50:48 +00:00
|
|
|
"Rack::Head",
|
2010-09-22 19:40:14 +00:00
|
|
|
"Rack::ConditionalGet",
|
2013-01-28 17:14:28 +00:00
|
|
|
"Rack::ETag"
|
2010-09-15 19:58:49 +00:00
|
|
|
], middleware
|
|
|
|
end
|
|
|
|
|
2015-04-16 17:55:32 +00:00
|
|
|
test "api middleware stack" do
|
|
|
|
add_to_config "config.api_only = true"
|
|
|
|
|
|
|
|
boot!
|
|
|
|
|
|
|
|
assert_equal [
|
|
|
|
"Rack::Sendfile",
|
|
|
|
"ActionDispatch::Static",
|
2016-02-22 01:25:52 +00:00
|
|
|
"ActionDispatch::Executor",
|
2015-04-16 17:55:32 +00:00
|
|
|
"ActiveSupport::Cache::Strategy::LocalCache",
|
2015-10-03 21:21:31 +00:00
|
|
|
"Rack::Runtime",
|
2015-04-16 17:55:32 +00:00
|
|
|
"ActionDispatch::RequestId",
|
2016-12-30 21:48:18 +00:00
|
|
|
"ActionDispatch::RemoteIp",
|
2016-12-30 21:50:35 +00:00
|
|
|
"Rails::Rack::Logger",
|
2015-04-16 17:55:32 +00:00
|
|
|
"ActionDispatch::ShowExceptions",
|
|
|
|
"ActionDispatch::DebugExceptions",
|
|
|
|
"ActionDispatch::Reloader",
|
|
|
|
"ActionDispatch::Callbacks",
|
|
|
|
"Rack::Head",
|
|
|
|
"Rack::ConditionalGet",
|
|
|
|
"Rack::ETag"
|
|
|
|
], middleware
|
|
|
|
end
|
|
|
|
|
2016-12-30 21:50:35 +00:00
|
|
|
test "middleware dependencies" do
|
|
|
|
boot!
|
|
|
|
|
|
|
|
# The following array-of-arrays describes dependencies between
|
|
|
|
# middlewares: the first item in each list depends on the
|
|
|
|
# remaining items (and therefore must occur later in the
|
|
|
|
# middleware stack).
|
|
|
|
|
|
|
|
dependencies = [
|
|
|
|
# Logger needs a fully "corrected" request environment
|
|
|
|
%w(Rails::Rack::Logger Rack::MethodOverride ActionDispatch::RequestId ActionDispatch::RemoteIp),
|
|
|
|
|
|
|
|
# Serving public/ doesn't invoke user code, so it should skip
|
|
|
|
# locks etc
|
|
|
|
%w(ActionDispatch::Executor ActionDispatch::Static),
|
|
|
|
|
|
|
|
# Errors during reload must be reported
|
|
|
|
%w(ActionDispatch::Reloader ActionDispatch::ShowExceptions ActionDispatch::DebugExceptions),
|
|
|
|
|
|
|
|
# Outright dependencies
|
|
|
|
%w(ActionDispatch::Static Rack::Sendfile),
|
|
|
|
%w(ActionDispatch::Flash ActionDispatch::Session::CookieStore),
|
|
|
|
%w(ActionDispatch::Session::CookieStore ActionDispatch::Cookies),
|
|
|
|
]
|
|
|
|
|
|
|
|
require "tsort"
|
|
|
|
sorted = TSort.tsort((middleware | dependencies.flatten).method(:each),
|
|
|
|
lambda { |n, &b| dependencies.each { |m, *ds| ds.each(&b) if m == n } })
|
|
|
|
assert_equal sorted, middleware
|
|
|
|
end
|
|
|
|
|
2012-10-03 21:43:39 +00:00
|
|
|
test "Rack::Cache is not included by default" do
|
2010-09-15 19:58:49 +00:00
|
|
|
boot!
|
|
|
|
|
2016-09-16 16:44:05 +00:00
|
|
|
assert_not_includes middleware, "Rack::Cache", "Rack::Cache is not included in the default stack unless you set config.action_dispatch.rack_cache"
|
2012-10-03 21:43:39 +00:00
|
|
|
end
|
|
|
|
|
2012-10-18 17:31:52 +00:00
|
|
|
test "Rack::Cache is present when action_dispatch.rack_cache is set" do
|
2012-10-03 21:43:39 +00:00
|
|
|
add_to_config "config.action_dispatch.rack_cache = true"
|
|
|
|
|
|
|
|
boot!
|
|
|
|
|
2016-09-16 16:44:05 +00:00
|
|
|
assert_includes middleware, "Rack::Cache"
|
2010-01-05 01:34:42 +00:00
|
|
|
end
|
|
|
|
|
2013-06-08 19:10:15 +00:00
|
|
|
test "ActiveRecord::Migration::CheckPending is present when active_record.migration_error is set to :page_load" do
|
|
|
|
add_to_config "config.active_record.migration_error = :page_load"
|
|
|
|
|
|
|
|
boot!
|
|
|
|
|
2016-09-16 16:44:05 +00:00
|
|
|
assert_includes middleware, "ActiveRecord::Migration::CheckPending"
|
2013-06-08 19:10:15 +00:00
|
|
|
end
|
|
|
|
|
2012-03-17 02:22:25 +00:00
|
|
|
test "ActionDispatch::SSL is present when force_ssl is set" do
|
2011-03-27 16:55:46 +00:00
|
|
|
add_to_config "config.force_ssl = true"
|
|
|
|
boot!
|
2016-09-16 16:44:05 +00:00
|
|
|
assert_includes middleware, "ActionDispatch::SSL"
|
2011-03-27 16:55:46 +00:00
|
|
|
end
|
|
|
|
|
2012-03-17 02:22:25 +00:00
|
|
|
test "ActionDispatch::SSL is configured with options when given" do
|
2011-09-27 02:22:52 +00:00
|
|
|
add_to_config "config.force_ssl = true"
|
2016-06-20 01:35:59 +00:00
|
|
|
add_to_config "config.ssl_options = { redirect: { host: 'example.com' } }"
|
2011-09-27 02:22:52 +00:00
|
|
|
boot!
|
2011-12-20 19:17:17 +00:00
|
|
|
|
2016-06-20 01:35:59 +00:00
|
|
|
assert_equal [{ redirect: { host: "example.com" } }], Rails.application.middleware.first.args
|
2011-09-27 02:22:52 +00:00
|
|
|
end
|
|
|
|
|
2010-07-25 20:08:34 +00:00
|
|
|
test "removing Active Record omits its middleware" do
|
2010-01-05 01:34:42 +00:00
|
|
|
use_frameworks []
|
|
|
|
boot!
|
2016-09-16 16:44:05 +00:00
|
|
|
assert_not_includes middleware, "ActiveRecord::Migration::CheckPending"
|
2010-01-05 01:34:42 +00:00
|
|
|
end
|
|
|
|
|
2016-02-22 01:25:52 +00:00
|
|
|
test "includes executor" do
|
2014-09-29 16:02:42 +00:00
|
|
|
boot!
|
2016-02-22 01:25:52 +00:00
|
|
|
assert_includes middleware, "ActionDispatch::Executor"
|
2014-09-18 20:35:19 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
test "does not include lock if cache_classes is set and so is eager_load" do
|
|
|
|
add_to_config "config.cache_classes = true"
|
|
|
|
add_to_config "config.eager_load = true"
|
|
|
|
boot!
|
2014-09-29 16:02:42 +00:00
|
|
|
assert_not_includes middleware, "Rack::Lock"
|
2010-01-05 01:34:42 +00:00
|
|
|
end
|
|
|
|
|
2014-09-29 16:16:07 +00:00
|
|
|
test "does not include lock if allow_concurrency is set to :unsafe" do
|
|
|
|
add_to_config "config.allow_concurrency = :unsafe"
|
2013-03-03 20:20:44 +00:00
|
|
|
boot!
|
2014-09-29 16:02:42 +00:00
|
|
|
assert_not_includes middleware, "Rack::Lock"
|
|
|
|
end
|
|
|
|
|
|
|
|
test "includes lock if allow_concurrency is disabled" do
|
|
|
|
add_to_config "config.allow_concurrency = false"
|
|
|
|
boot!
|
|
|
|
assert_includes middleware, "Rack::Lock"
|
2013-03-03 20:20:44 +00:00
|
|
|
end
|
|
|
|
|
2015-11-04 21:31:16 +00:00
|
|
|
test "removes static asset server if public_file_server.enabled is disabled" do
|
|
|
|
add_to_config "config.public_file_server.enabled = false"
|
2010-01-05 01:34:42 +00:00
|
|
|
boot!
|
2016-09-16 16:44:05 +00:00
|
|
|
assert_not_includes middleware, "ActionDispatch::Static"
|
2010-01-05 01:34:42 +00:00
|
|
|
end
|
|
|
|
|
2010-06-07 21:17:23 +00:00
|
|
|
test "can delete a middleware from the stack" do
|
|
|
|
add_to_config "config.middleware.delete ActionDispatch::Static"
|
|
|
|
boot!
|
2016-09-16 16:44:05 +00:00
|
|
|
assert_not_includes middleware, "ActionDispatch::Static"
|
2010-06-07 21:17:23 +00:00
|
|
|
end
|
|
|
|
|
2015-02-18 18:35:51 +00:00
|
|
|
test "can delete a middleware from the stack even if insert_before is added after delete" do
|
2015-10-03 21:21:31 +00:00
|
|
|
add_to_config "config.middleware.delete Rack::Runtime"
|
|
|
|
add_to_config "config.middleware.insert_before(Rack::Runtime, Rack::Config)"
|
2015-02-18 18:35:51 +00:00
|
|
|
boot!
|
2016-09-16 16:44:05 +00:00
|
|
|
assert_includes middleware, "Rack::Config"
|
2015-10-03 21:21:31 +00:00
|
|
|
assert_not middleware.include?("Rack::Runtime")
|
2015-02-18 18:35:51 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
test "can delete a middleware from the stack even if insert_after is added after delete" do
|
2015-10-03 21:21:31 +00:00
|
|
|
add_to_config "config.middleware.delete Rack::Runtime"
|
|
|
|
add_to_config "config.middleware.insert_after(Rack::Runtime, Rack::Config)"
|
2015-02-18 18:35:51 +00:00
|
|
|
boot!
|
2016-09-16 16:44:05 +00:00
|
|
|
assert_includes middleware, "Rack::Config"
|
2015-10-03 21:21:31 +00:00
|
|
|
assert_not middleware.include?("Rack::Runtime")
|
2015-02-18 18:35:51 +00:00
|
|
|
end
|
|
|
|
|
2011-12-01 19:46:18 +00:00
|
|
|
test "includes exceptions middlewares even if action_dispatch.show_exceptions is disabled" do
|
2010-04-02 18:54:10 +00:00
|
|
|
add_to_config "config.action_dispatch.show_exceptions = false"
|
|
|
|
boot!
|
2016-09-16 16:44:05 +00:00
|
|
|
assert_includes middleware, "ActionDispatch::ShowExceptions"
|
|
|
|
assert_includes middleware, "ActionDispatch::DebugExceptions"
|
2010-04-02 18:54:10 +00:00
|
|
|
end
|
|
|
|
|
2010-11-24 02:04:05 +00:00
|
|
|
test "removes ActionDispatch::Reloader if cache_classes is true" do
|
|
|
|
add_to_config "config.cache_classes = true"
|
|
|
|
boot!
|
2016-09-16 16:44:05 +00:00
|
|
|
assert_not_includes middleware, "ActionDispatch::Reloader"
|
2010-11-24 02:04:05 +00:00
|
|
|
end
|
|
|
|
|
2010-01-05 01:34:42 +00:00
|
|
|
test "use middleware" do
|
|
|
|
use_frameworks []
|
|
|
|
add_to_config "config.middleware.use Rack::Config"
|
|
|
|
boot!
|
|
|
|
assert_equal "Rack::Config", middleware.last
|
|
|
|
end
|
|
|
|
|
2010-01-05 01:38:56 +00:00
|
|
|
test "insert middleware after" do
|
2013-07-15 05:33:09 +00:00
|
|
|
add_to_config "config.middleware.insert_after Rack::Sendfile, Rack::Config"
|
2010-01-05 01:38:56 +00:00
|
|
|
boot!
|
2010-09-15 19:58:49 +00:00
|
|
|
assert_equal "Rack::Config", middleware.second
|
2010-01-05 01:38:56 +00:00
|
|
|
end
|
|
|
|
|
2016-08-06 17:16:09 +00:00
|
|
|
test "unshift middleware" do
|
|
|
|
add_to_config "config.middleware.unshift Rack::Config"
|
2013-10-09 10:16:22 +00:00
|
|
|
boot!
|
2016-08-06 17:16:09 +00:00
|
|
|
assert_equal "Rack::Config", middleware.first
|
2013-10-09 10:16:22 +00:00
|
|
|
end
|
|
|
|
|
2012-01-17 16:53:09 +00:00
|
|
|
test "Rails.cache does not respond to middleware" do
|
2010-06-24 16:42:09 +00:00
|
|
|
add_to_config "config.cache_store = :memory_store"
|
|
|
|
boot!
|
2015-10-03 21:21:31 +00:00
|
|
|
assert_equal "Rack::Runtime", middleware.fourth
|
2010-06-24 16:42:09 +00:00
|
|
|
end
|
|
|
|
|
2012-01-17 16:53:09 +00:00
|
|
|
test "Rails.cache does respond to middleware" do
|
2010-06-24 16:42:09 +00:00
|
|
|
boot!
|
2015-10-03 21:21:31 +00:00
|
|
|
assert_equal "Rack::Runtime", middleware.fifth
|
2010-06-24 16:42:09 +00:00
|
|
|
end
|
|
|
|
|
2010-01-05 01:38:56 +00:00
|
|
|
test "insert middleware before" do
|
2013-07-15 05:33:09 +00:00
|
|
|
add_to_config "config.middleware.insert_before Rack::Sendfile, Rack::Config"
|
2010-01-05 01:38:56 +00:00
|
|
|
boot!
|
2010-09-15 19:58:49 +00:00
|
|
|
assert_equal "Rack::Config", middleware.first
|
2010-01-05 01:38:56 +00:00
|
|
|
end
|
|
|
|
|
2012-04-20 19:17:03 +00:00
|
|
|
test "can't change middleware after it's built" do
|
|
|
|
boot!
|
|
|
|
assert_raise RuntimeError do
|
|
|
|
app.config.middleware.use Rack::Config
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2010-09-22 19:40:14 +00:00
|
|
|
# ConditionalGet + Etag
|
|
|
|
test "conditional get + etag middlewares handle http caching based on body" do
|
|
|
|
make_basic_app
|
|
|
|
|
|
|
|
class ::OmgController < ActionController::Base
|
|
|
|
def index
|
|
|
|
if params[:nothing]
|
2016-05-21 12:49:38 +00:00
|
|
|
render plain: ""
|
2010-09-22 19:40:14 +00:00
|
|
|
else
|
2016-05-21 12:49:38 +00:00
|
|
|
render plain: "OMG"
|
2010-09-22 19:40:14 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2016-05-06 21:30:42 +00:00
|
|
|
etag = "W/" + "c00862d1c6c1cf7c1b49388306e7b3c1".inspect
|
2010-09-22 19:40:14 +00:00
|
|
|
|
|
|
|
get "/"
|
|
|
|
assert_equal 200, last_response.status
|
|
|
|
assert_equal "OMG", last_response.body
|
2016-05-21 12:49:38 +00:00
|
|
|
assert_equal "text/plain; charset=utf-8", last_response.headers["Content-Type"]
|
2010-09-22 19:40:14 +00:00
|
|
|
assert_equal "max-age=0, private, must-revalidate", last_response.headers["Cache-Control"]
|
|
|
|
assert_equal etag, last_response.headers["Etag"]
|
|
|
|
|
|
|
|
get "/", {}, "HTTP_IF_NONE_MATCH" => etag
|
|
|
|
assert_equal 304, last_response.status
|
|
|
|
assert_equal "", last_response.body
|
2016-12-24 17:29:52 +00:00
|
|
|
assert_nil last_response.headers["Content-Type"]
|
2010-09-22 19:40:14 +00:00
|
|
|
assert_equal "max-age=0, private, must-revalidate", last_response.headers["Cache-Control"]
|
|
|
|
assert_equal etag, last_response.headers["Etag"]
|
|
|
|
|
|
|
|
get "/?nothing=true"
|
|
|
|
assert_equal 200, last_response.status
|
|
|
|
assert_equal "", last_response.body
|
2016-05-21 12:49:38 +00:00
|
|
|
assert_equal "text/plain; charset=utf-8", last_response.headers["Content-Type"]
|
2010-09-22 19:40:14 +00:00
|
|
|
assert_equal "no-cache", last_response.headers["Cache-Control"]
|
2016-12-24 17:29:52 +00:00
|
|
|
assert_nil last_response.headers["Etag"]
|
2010-09-22 19:40:14 +00:00
|
|
|
end
|
|
|
|
|
2011-12-20 19:17:17 +00:00
|
|
|
test "ORIGINAL_FULLPATH is passed to env" do
|
|
|
|
boot!
|
|
|
|
env = ::Rack::MockRequest.env_for("/foo/?something")
|
|
|
|
Rails.application.call(env)
|
|
|
|
|
|
|
|
assert_equal "/foo/?something", env["ORIGINAL_FULLPATH"]
|
|
|
|
end
|
|
|
|
|
2010-01-05 01:34:42 +00:00
|
|
|
private
|
2010-04-26 07:04:04 +00:00
|
|
|
|
2010-01-05 01:34:42 +00:00
|
|
|
def boot!
|
|
|
|
require "#{app_path}/config/environment"
|
|
|
|
end
|
|
|
|
|
|
|
|
def middleware
|
2013-06-10 02:47:07 +00:00
|
|
|
Rails.application.middleware.map(&:klass).map(&:name)
|
2010-01-05 01:34:42 +00:00
|
|
|
end
|
|
|
|
end
|
2011-06-06 12:54:05 +00:00
|
|
|
end
|