vendor/rails/actionpack/test/controller/rack_test.rb

   1 require 'abstract_unit'
   2 require 'action_controller/rack_process'
   3
   4 class BaseRackTest < Test::Unit::TestCase
   5   def setup
   6     @env = {
   7       "HTTP_MAX_FORWARDS" => "10",
   8       "SERVER_NAME" => "glu.ttono.us:8007",
   9       "FCGI_ROLE" => "RESPONDER",
  10       "AUTH_TYPE" => "Basic",
  11       "HTTP_X_FORWARDED_HOST" => "glu.ttono.us",
  12       "HTTP_ACCEPT_CHARSET" => "UTF-8",
  13       "HTTP_ACCEPT_ENCODING" => "gzip, deflate",
  14       "HTTP_CACHE_CONTROL" => "no-cache, max-age=0",
  15       "HTTP_PRAGMA" => "no-cache",
  16       "HTTP_USER_AGENT" => "Mozilla/5.0 (Macintosh; U; PPC Mac OS X; en)",
  17       "PATH_INFO" => "/homepage/",
  18       "HTTP_ACCEPT_LANGUAGE" => "en",
  19       "HTTP_NEGOTIATE" => "trans",
  20       "HTTP_HOST" => "glu.ttono.us:8007",
  21       "HTTP_REFERER" => "http://www.google.com/search?q=glu.ttono.us",
  22       "HTTP_FROM" => "googlebot",
  23       "SERVER_PROTOCOL" => "HTTP/1.1",
  24       "REDIRECT_URI" => "/dispatch.fcgi",
  25       "SCRIPT_NAME" => "/dispatch.fcgi",
  26       "SERVER_ADDR" => "207.7.108.53",
  27       "REMOTE_ADDR" => "207.7.108.53",
  28       "REMOTE_HOST" => "google.com",
  29       "REMOTE_IDENT" => "kevin",
  30       "REMOTE_USER" => "kevin",
  31       "SERVER_SOFTWARE" => "lighttpd/1.4.5",
  32       "HTTP_COOKIE" => "_session_id=c84ace84796670c052c6ceb2451fb0f2; is_admin=yes",
  33       "HTTP_X_FORWARDED_SERVER" => "glu.ttono.us",
  34       "REQUEST_URI" => "/admin",
  35       "DOCUMENT_ROOT" => "/home/kevinc/sites/typo/public",
  36       "PATH_TRANSLATED" => "/home/kevinc/sites/typo/public/homepage/",
  37       "SERVER_PORT" => "8007",
  38       "QUERY_STRING" => "",
  39       "REMOTE_PORT" => "63137",
  40       "GATEWAY_INTERFACE" => "CGI/1.1",
  41       "HTTP_X_FORWARDED_FOR" => "65.88.180.234",
  42       "HTTP_ACCEPT" => "*/*",
  43       "SCRIPT_FILENAME" => "/home/kevinc/sites/typo/public/dispatch.fcgi",
  44       "REDIRECT_STATUS" => "200",
  45       "REQUEST_METHOD" => "GET"
  46     }
  47     @request = ActionController::RackRequest.new(@env)
  48     # some Nokia phone browsers omit the space after the semicolon separator.
  49     # some developers have grown accustomed to using comma in cookie values.
  50     @alt_cookie_fmt_request = ActionController::RackRequest.new(@env.merge({"HTTP_COOKIE"=>"_session_id=c84ace847,96670c052c6ceb2451fb0f2;is_admin=yes"}))
  51   end
  52
  53   def default_test; end
  54
  55   private
  56
  57   def set_content_data(data)
  58     @request.env['REQUEST_METHOD'] = 'POST'
  59     @request.env['CONTENT_LENGTH'] = data.length
  60     @request.env['CONTENT_TYPE'] = 'application/x-www-form-urlencoded; charset=utf-8'
  61     @request.env['RAW_POST_DATA'] = data
  62   end
  63 end
  64
  65 class RackRequestTest < BaseRackTest
  66   def test_proxy_request
  67     assert_equal 'glu.ttono.us', @request.host_with_port(true)
  68   end
  69
  70   def test_http_host
  71     @env.delete "HTTP_X_FORWARDED_HOST"
  72     @env['HTTP_HOST'] = "rubyonrails.org:8080"
  73     assert_equal "rubyonrails.org", @request.host(true)
  74     assert_equal "rubyonrails.org:8080", @request.host_with_port(true)
  75
  76     @env['HTTP_X_FORWARDED_HOST'] = "www.firsthost.org, www.secondhost.org"
  77     assert_equal "www.secondhost.org", @request.host(true)
  78   end
  79
  80   def test_http_host_with_default_port_overrides_server_port
  81     @env.delete "HTTP_X_FORWARDED_HOST"
  82     @env['HTTP_HOST'] = "rubyonrails.org"
  83     assert_equal "rubyonrails.org", @request.host_with_port(true)
  84   end
  85
  86   def test_host_with_port_defaults_to_server_name_if_no_host_headers
  87     @env.delete "HTTP_X_FORWARDED_HOST"
  88     @env.delete "HTTP_HOST"
  89     assert_equal "glu.ttono.us:8007", @request.host_with_port(true)
  90   end
  91
  92   def test_host_with_port_falls_back_to_server_addr_if_necessary
  93     @env.delete "HTTP_X_FORWARDED_HOST"
  94     @env.delete "HTTP_HOST"
  95     @env.delete "SERVER_NAME"
  96     assert_equal "207.7.108.53", @request.host(true)
  97     assert_equal 8007, @request.port(true)
  98     assert_equal "207.7.108.53:8007", @request.host_with_port(true)
  99   end
 100
 101   def test_host_with_port_if_http_standard_port_is_specified
 102     @env['HTTP_X_FORWARDED_HOST'] = "glu.ttono.us:80"
 103     assert_equal "glu.ttono.us", @request.host_with_port(true)
 104   end
 105
 106   def test_host_with_port_if_https_standard_port_is_specified
 107     @env['HTTP_X_FORWARDED_PROTO'] = "https"
 108     @env['HTTP_X_FORWARDED_HOST'] = "glu.ttono.us:443"
 109     assert_equal "glu.ttono.us", @request.host_with_port(true)
 110   end
 111
 112   def test_host_if_ipv6_reference
 113     @env.delete "HTTP_X_FORWARDED_HOST"
 114     @env['HTTP_HOST'] = "[2001:1234:5678:9abc:def0::dead:beef]"
 115     assert_equal "[2001:1234:5678:9abc:def0::dead:beef]", @request.host(true)
 116   end
 117
 118   def test_host_if_ipv6_reference_with_port
 119     @env.delete "HTTP_X_FORWARDED_HOST"
 120     @env['HTTP_HOST'] = "[2001:1234:5678:9abc:def0::dead:beef]:8008"
 121     assert_equal "[2001:1234:5678:9abc:def0::dead:beef]", @request.host(true)
 122   end
 123
 124   def test_cgi_environment_variables
 125     assert_equal "Basic", @request.auth_type
 126     assert_equal 0, @request.content_length
 127     assert_equal nil, @request.content_type
 128     assert_equal "CGI/1.1", @request.gateway_interface
 129     assert_equal "*/*", @request.accept
 130     assert_equal "UTF-8", @request.accept_charset
 131     assert_equal "gzip, deflate", @request.accept_encoding
 132     assert_equal "en", @request.accept_language
 133     assert_equal "no-cache, max-age=0", @request.cache_control
 134     assert_equal "googlebot", @request.from
 135     assert_equal "glu.ttono.us", @request.host
 136     assert_equal "trans", @request.negotiate
 137     assert_equal "no-cache", @request.pragma
 138     assert_equal "http://www.google.com/search?q=glu.ttono.us", @request.referer
 139     assert_equal "Mozilla/5.0 (Macintosh; U; PPC Mac OS X; en)", @request.user_agent
 140     assert_equal "/homepage/", @request.path_info
 141     assert_equal "/home/kevinc/sites/typo/public/homepage/", @request.path_translated
 142     assert_equal "", @request.query_string
 143     assert_equal "207.7.108.53", @request.remote_addr
 144     assert_equal "google.com", @request.remote_host
 145     assert_equal "kevin", @request.remote_ident
 146     assert_equal "kevin", @request.remote_user
 147     assert_equal :get, @request.request_method
 148     assert_equal "/dispatch.fcgi", @request.script_name
 149     assert_equal "glu.ttono.us:8007", @request.server_name
 150     assert_equal 8007, @request.server_port
 151     assert_equal "HTTP/1.1", @request.server_protocol
 152     assert_equal "lighttpd", @request.server_software
 153   end
 154
 155   def test_cookie_syntax_resilience
 156     cookies = @request.cookies
 157     assert_equal ["c84ace84796670c052c6ceb2451fb0f2"], cookies["_session_id"], cookies.inspect
 158     assert_equal ["yes"], cookies["is_admin"], cookies.inspect
 159
 160     alt_cookies = @alt_cookie_fmt_request.cookies
 161     assert_equal ["c84ace847,96670c052c6ceb2451fb0f2"], alt_cookies["_session_id"], alt_cookies.inspect
 162     assert_equal ["yes"], alt_cookies["is_admin"], alt_cookies.inspect
 163   end
 164 end
 165
 166 class RackRequestParamsParsingTest < BaseRackTest
 167   def test_doesnt_break_when_content_type_has_charset
 168     set_content_data 'flamenco=love'
 169
 170     assert_equal({"flamenco"=> "love"}, @request.request_parameters)
 171   end
 172
 173   def test_doesnt_interpret_request_uri_as_query_string_when_missing
 174     @request.env['REQUEST_URI'] = 'foo'
 175     assert_equal({}, @request.query_parameters)
 176   end
 177 end
 178
 179 class RackRequestContentTypeTest < BaseRackTest
 180   def test_html_content_type_verification
 181     @request.env['CONTENT_TYPE'] = Mime::HTML.to_s
 182     assert @request.content_type.verify_request?
 183   end
 184
 185   def test_xml_content_type_verification
 186     @request.env['CONTENT_TYPE'] = Mime::XML.to_s
 187     assert !@request.content_type.verify_request?
 188   end
 189 end
 190
 191 class RackRequestMethodTest < BaseRackTest
 192   def test_get
 193     assert_equal :get, @request.request_method
 194   end
 195
 196   def test_post
 197     @request.env['REQUEST_METHOD'] = 'POST'
 198     assert_equal :post, @request.request_method
 199   end
 200
 201   def test_put
 202     set_content_data '_method=put'
 203
 204     assert_equal :put, @request.request_method
 205   end
 206
 207   def test_delete
 208     set_content_data '_method=delete'
 209
 210     assert_equal :delete, @request.request_method
 211   end
 212 end
 213
 214 class RackRequestNeedsRewoundTest < BaseRackTest
 215   def test_body_should_be_rewound
 216     data = 'foo'
 217     @env['rack.input'] = StringIO.new(data)
 218     @env['CONTENT_LENGTH'] = data.length
 219     @env['CONTENT_TYPE'] = 'application/x-www-form-urlencoded; charset=utf-8'
 220
 221     # Read the request body by parsing params.
 222     request = ActionController::RackRequest.new(@env)
 223     request.request_parameters
 224
 225     # Should have rewound the body.
 226     assert_equal 0, request.body.pos
 227   end
 228 end
 229
 230 class RackResponseTest < BaseRackTest
 231   def setup
 232     super
 233     @response = ActionController::RackResponse.new(@request)
 234     @output = StringIO.new('')
 235   end
 236
 237   def test_simple_output
 238     @response.body = "Hello, World!"
 239     @response.prepare!
 240
 241     status, headers, body = @response.out(@output)
 242     assert_equal "200 OK", status
 243     assert_equal({
 244       "Content-Type" => "text/html; charset=utf-8",
 245       "Cache-Control" => "private, max-age=0, must-revalidate",
 246       "ETag" => '"65a8e27d8879283831b664bd8b7f0ad4"',
 247       "Set-Cookie" => [],
 248       "Content-Length" => "13"
 249     }, headers)
 250
 251     parts = []
 252     body.each { |part| parts << part }
 253     assert_equal ["Hello, World!"], parts
 254   end
 255
 256   def test_streaming_block
 257     @response.body = Proc.new do |response, output|
 258       5.times { |n| output.write(n) }
 259     end
 260     @response.prepare!
 261
 262     status, headers, body = @response.out(@output)
 263     assert_equal "200 OK", status
 264     assert_equal({"Content-Type" => "text/html; charset=utf-8", "Cache-Control" => "no-cache", "Set-Cookie" => []}, headers)
 265
 266     parts = []
 267     body.each { |part| parts << part }
 268     assert_equal ["0", "1", "2", "3", "4"], parts
 269   end
 270
 271   def test_set_session_cookie
 272     cookie = CGI::Cookie.new({"name" => "name", "value" => "Josh"})
 273     @request.cgi.send :instance_variable_set, '@output_cookies', [cookie]
 274
 275     @response.body = "Hello, World!"
 276     @response.prepare!
 277
 278     status, headers, body = @response.out(@output)
 279     assert_equal "200 OK", status
 280     assert_equal({
 281       "Content-Type" => "text/html; charset=utf-8",
 282       "Cache-Control" => "private, max-age=0, must-revalidate",
 283       "ETag" => '"65a8e27d8879283831b664bd8b7f0ad4"',
 284       "Set-Cookie" => ["name=Josh; path="],
 285       "Content-Length" => "13"
 286     }, headers)
 287
 288     parts = []
 289     body.each { |part| parts << part }
 290     assert_equal ["Hello, World!"], parts
 291   end
 292 end
 293
 294 class RackResponseHeadersTest < BaseRackTest
 295   def setup
 296     super
 297     @response = ActionController::RackResponse.new(@request)
 298     @output = StringIO.new('')
 299     @response.headers['Status'] = "200 OK"
 300   end
 301
 302   def test_content_type
 303     [204, 304].each do |c|
 304       @response.headers['Status'] = c.to_s
 305       assert !response_headers.has_key?("Content-Type"), "#{c} should not have Content-Type header"
 306     end
 307
 308     [200, 302, 404, 500].each do |c|
 309       @response.headers['Status'] = c.to_s
 310       assert response_headers.has_key?("Content-Type"), "#{c} did not have Content-Type header"
 311     end
 312   end
 313
 314   def test_status
 315     assert !response_headers.has_key?('Status')
 316   end
 317
 318   private
 319     def response_headers
 320       @response.prepare!
 321       @response.out(@output)[1]
 322     end
 323 end