Module: Devise::TestHelpers

Defined in:
lib/devise/test_helpers.rb

Overview

Devise::TestHelpers provides a facility to test controllers in isolation when using ActionController::TestCase allowing you to quickly sign_in or sign_out a user. Do not use Devise::TestHelpers in integration tests.

Notice you should not test Warden specific behavior (like Warden callbacks) using Devise::TestHelpers since it is a stub of the actual behavior. Such callbacks should be tested in your integration suite instead.

Instance Method Summary collapse

Instance Method Details

#_catch_warden(&block) ⇒ Object (protected)

Catch warden continuations and handle like the middleware would. Returns nil when interrupted, otherwise the normal result of the block.



74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
 
# File 'lib/devise/test_helpers.rb', line 74

def _catch_warden(&block)
  result = catch(:warden, &block)

  env = @controller.request.env

  result ||= {}

  # Set the response. In production, the rack result is returned
  # from Warden::Manager#call, which the following is modelled on.
  case result
  when Array
    if result.first == 401 && intercept_401?(env) # does this happen during testing?
      _process_unauthenticated(env)
    else
      result
    end
  when Hash
    _process_unauthenticated(env, result)
  else
    result
  end
end

#_process_unauthenticated(env, options = {}) ⇒ Object (protected) 



97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
 
# File 'lib/devise/test_helpers.rb', line 97

def _process_unauthenticated(env, options = {})
  options[:action] ||= :unauthenticated
  proxy = env['warden']
  result = options[:result] || proxy.result

  ret = case result
  when :redirect
    body = proxy.message || "You are being redirected to #{proxy.headers['Location']}"
    [proxy.status, proxy.headers, [body]]
  when :custom
    proxy.custom_response
  else
    env["PATH_INFO"] = "/#{options[:action]}"
    env["warden.options"] = options
    Warden::Manager._run_callbacks(:before_failure, env, options)

    status, headers, response = Devise.warden_config[:failure_app].call(env).to_a
    @controller.response.headers.merge!(headers)
    r_opts = { status: status, content_type: headers["Content-Type"], location: headers["Location"] }
    r_opts[Rails.version.start_with?('5') ? :body : :text] = response.body
    @controller.send :render, r_opts
    nil # causes process return @response
  end

  # ensure that the controller response is set up. In production, this is
  # not necessary since warden returns the results to rack. However, at
  # testing time, we want the response to be available to the testing
  # framework to verify what would be returned to rack.
  if ret.is_a?(Array)
    # ensure the controller response is set to our response.
    @controller.response ||= @response
    @response.status = ret.first
    @response.headers.clear
    ret.second.each { |k,v| @response[k] = v }
    @response.body = ret.third
  end

  ret
end

#process ⇒ Object

Override process to consider warden.



17
18
19
20
21
22
23
 
# File 'lib/devise/test_helpers.rb', line 17

def process(*)
  # Make sure we always return @response, a la ActionController::TestCase::Behaviour#process, even if warden interrupts
  _catch_warden { super } # || @response  # _catch_warden will setup the @response object

  # process needs to return the ActionDispath::TestResponse object
  @response
end

#setup_controller_for_warden ⇒ Object

We need to set up the environment variables and the response in the controller.



26
27
28
 
# File 'lib/devise/test_helpers.rb', line 26

def setup_controller_for_warden #:nodoc:
  @request.env['action_controller.instance'] = @controller
end

#sign_in(resource_or_scope, resource = nil) ⇒ Object

sign_in a given resource by storing its keys in the session. This method bypass any warden authentication callback.

Examples:

 :user, @user   # sign_in(scope, resource)
 @user          # sign_in(resource)



48
49
50
51
52
53
 
# File 'lib/devise/test_helpers.rb', line 48

def (resource_or_scope, resource=nil)
  scope    ||= Devise::Mapping.find_scope!(resource_or_scope)
  resource ||= resource_or_scope
  warden.instance_variable_get(:@users).delete(scope)
  warden.session_serializer.store(resource, scope)
end

#sign_out(resource_or_scope) ⇒ Object

Sign out a given resource or scope by calling logout on Warden. This method bypass any warden logout callback.

Examples:

sign_out :user     # sign_out(scope)
sign_out @user     # sign_out(resource)



63
64
65
66
67
68
 
# File 'lib/devise/test_helpers.rb', line 63

def sign_out(resource_or_scope)
  scope = Devise::Mapping.find_scope!(resource_or_scope)
  @controller.instance_variable_set(:"@current_#{scope}", nil)
  user = warden.instance_variable_get(:@users).delete(scope)
  warden.session_serializer.delete(scope, user)
end

#warden ⇒ Object

Quick access to Warden::Proxy.



31
32
33
34
35
36
37
38
 
# File 'lib/devise/test_helpers.rb', line 31

def warden #:nodoc:
  @request.env['warden'] ||= begin
    manager = Warden::Manager.new(nil) do |config|
      config.merge! Devise.warden_config
    end
    Warden::Proxy.new(@request.env, manager)
  end
end