aboutsummaryrefslogtreecommitdiffstats
path: root/actionpack/lib/action_dispatch/testing/test_request.rb
blob: 46523a860048d125ca4923194c4ef3811a71e4e5 (plain) (blame)
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
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)
      set_header('REQUEST_METHOD', 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