/
response.rb
114 lines (99 loc) · 4.31 KB
/
response.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
# frozen_string_literal: true
# :markup: markdown
module ActionDispatch
module Assertions
# A small suite of assertions that test responses from Rails applications.
module ResponseAssertions
RESPONSE_PREDICATES = { # :nodoc:
success: :successful?,
missing: :not_found?,
redirect: :redirection?,
error: :server_error?,
}
# Asserts that the response is one of the following types:
#
# * `:success` - Status code was in the 200-299 range
# * `:redirect` - Status code was in the 300-399 range
# * `:missing` - Status code was 404
# * `:error` - Status code was in the 500-599 range
#
#
# You can also pass an explicit status number like `assert_response(501)` or its
# symbolic equivalent `assert_response(:not_implemented)`. See
# `Rack::Utils::SYMBOL_TO_STATUS_CODE` for a full list.
#
# # Asserts that the response was a redirection
# assert_response :redirect
#
# # Asserts that the response code was status code 401 (unauthorized)
# assert_response 401
def assert_response(type, message = nil)
message ||= generate_response_message(type)
if RESPONSE_PREDICATES.key?(type)
assert @response.public_send(RESPONSE_PREDICATES[type]), message
else
assert_equal AssertionResponse.new(type).code, @response.response_code, message
end
end
# Asserts that the response is a redirect to a URL matching the given options.
#
# # Asserts that the redirection was to the "index" action on the WeblogController
# assert_redirected_to controller: "weblog", action: "index"
#
# # Asserts that the redirection was to the named route login_url
# assert_redirected_to login_url
#
# # Asserts that the redirection was to the URL for @customer
# assert_redirected_to @customer
#
# # Asserts that the redirection matches the regular expression
# assert_redirected_to %r(\Ahttp://example.org)
#
# # Asserts that the redirection has the HTTP status code 301 (Moved
# # Permanently).
# assert_redirected_to "/some/path", status: :moved_permanently
def assert_redirected_to(url_options = {}, options = {}, message = nil)
options, message = {}, options unless options.is_a?(Hash)
status = options[:status] || :redirect
assert_response(status, message)
return true if url_options === @response.location
redirect_is = normalize_argument_to_redirection(@response.location)
redirect_expected = normalize_argument_to_redirection(url_options)
message ||= "Expected response to be a redirect to <#{redirect_expected}> but was a redirect to <#{redirect_is}>"
assert_operator redirect_expected, :===, redirect_is, message
end
private
# Proxy to to_param if the object will respond to it.
def parameterize(value)
value.respond_to?(:to_param) ? value.to_param : value
end
def normalize_argument_to_redirection(fragment)
if Regexp === fragment
fragment
else
handle = @controller || ActionController::Redirecting
handle._compute_redirect_to_location(@request, fragment)
end
end
def generate_response_message(expected, actual = @response.response_code)
(+"Expected response to be a <#{code_with_name(expected)}>,"\
" but was a <#{code_with_name(actual)}>").concat(location_if_redirected).concat(response_body_if_short)
end
def response_body_if_short
return "" if @response.body.size > 500
"\nResponse body: #{@response.body}"
end
def location_if_redirected
return "" unless @response.redirection? && @response.location.present?
location = normalize_argument_to_redirection(@response.location)
" redirect to <#{location}>"
end
def code_with_name(code_or_name)
if RESPONSE_PREDICATES.value?("#{code_or_name}?".to_sym)
code_or_name = RESPONSE_PREDICATES.invert["#{code_or_name}?".to_sym]
end
AssertionResponse.new(code_or_name).code_and_name
end
end
end
end