aboutsummaryrefslogblamecommitdiffstats
path: root/actionpack/lib/action_dispatch/testing/test_request.rb
blob: 91b25ec1552f802a443d36a11f4fbf14fe4dfac7 (plain) (tree)
1
2
3
4
5
6
7
8
9

                                                         
 

                             



                                                    
     
 

                                                         
                                                                                                       
                                                                    
                                 

       




                                     
                               
                               


                   
                                   


                     
                                            


                         
                                    


                   
                                   

       
                            
                                                 

       
                                         
                                                         


                            
                                            


                          
                                     


                               
                                               


                           

                                                                            


       
require "active_support/core_ext/hash/indifferent_access"
require "rack/utils"

module ActionDispatch
  class TestRequest < Request
    DEFAULT_ENV = Rack::MockRequest.env_for("/",
      "HTTP_HOST"                => "test.host",
      "REMOTE_ADDR"              => "0.0.0.0",
      "HTTP_USER_AGENT"          => "Rails Testing",
    )

    # Create a new test request with default `env` values
    def self.create(env = {})
      env = Rails.application.env_config.merge(env) if defined?(Rails.application) && Rails.application
      env["rack.request.cookie_hash"] ||= {}.with_indifferent_access
      new(default_env.merge(env))
    end

    def self.default_env
      DEFAULT_ENV
    end
    private_class_method :default_env

    def request_method=(method)
      super(method.to_s.upcase)
    end

    def host=(host)
      set_header("HTTP_HOST", host)
    end

    def port=(number)
      set_header("SERVER_PORT", number.to_i)
    end

    def request_uri=(uri)
      set_header("REQUEST_URI", uri)
    end

    def path=(path)
      set_header("PATH_INFO", path)
    end

    def action=(action_name)
      path_parameters[:action] = action_name.to_s
    end

    def if_modified_since=(last_modified)
      set_header("HTTP_IF_MODIFIED_SINCE", last_modified)
    end

    def if_none_match=(etag)
      set_header("HTTP_IF_NONE_MATCH", etag)
    end

    def remote_addr=(addr)
      set_header("REMOTE_ADDR", addr)
    end

    def user_agent=(user_agent)
      set_header("HTTP_USER_AGENT", user_agent)
    end

    def accept=(mime_types)
      delete_header("action_dispatch.request.accepts")
      set_header("HTTP_ACCEPT", Array(mime_types).collect(&:to_s).join(","))
    end
  end
end