abstract_unit.rb 7.3 KB
Newer Older
1
begin
2
  require File.expand_path('../../../vendor/gems/environment', __FILE__)
3
rescue LoadError
4 5
end

6
lib = File.expand_path('../../lib', __FILE__)
7 8
$:.unshift(lib) unless $:.include?('lib') || $:.include?(lib)

9
$:.unshift(File.dirname(__FILE__) + '/lib')
10
$:.unshift(File.dirname(__FILE__) + '/fixtures/helpers')
11
$:.unshift(File.dirname(__FILE__) + '/fixtures/alternate_helpers')
D
Initial  
David Heinemeier Hansson 已提交
12

J
Joshua Peek 已提交
13 14
ENV['TMPDIR'] = File.join(File.dirname(__FILE__), 'tmp')

D
Initial  
David Heinemeier Hansson 已提交
15
require 'test/unit'
J
Joshua Peek 已提交
16
require 'abstract_controller'
17
require 'action_controller'
J
Joshua Peek 已提交
18 19 20
require 'action_view'
require 'action_view/base'
require 'action_dispatch'
21 22
require 'fixture_template'
require 'active_support/dependencies'
D
Initial  
David Heinemeier Hansson 已提交
23

24 25 26 27
activemodel_path = File.expand_path('../../../activemodel/lib', __FILE__)
$:.unshift(activemodel_path) if File.directory?(activemodel_path) && !$:.include?(activemodel_path)
require 'active_model'

28 29
begin
  require 'ruby-debug'
J
Jeremy Kemper 已提交
30
  Debugger.settings[:autoeval] = true
31
  Debugger.start
32 33 34
rescue LoadError
  # Debugging disabled. `gem install ruby-debug` to enable.
end
35

J
Joshua Peek 已提交
36 37
require 'pp' # require 'pp' early to prevent hidden_methods from not picking up the pretty-print methods until too late

38
ActiveSupport::Dependencies.hook!
39

40 41 42
# Show backtraces for deprecated behavior for quicker cleanup.
ActiveSupport::Deprecation.debug = true

J
Joshua Peek 已提交
43 44
# Register danish language for testing
I18n.backend.store_translations 'da', {}
45
I18n.backend.store_translations 'pt-BR', {}
46
ORIGINAL_LOCALES = I18n.available_locales.map {|locale| locale.to_s }.sort
J
Joshua Peek 已提交
47

Y
Yehuda Katz + Carl Lerche 已提交
48
FIXTURE_LOAD_PATH = File.join(File.dirname(__FILE__), 'fixtures')
49
FIXTURES = Pathname.new(FIXTURE_LOAD_PATH)
50

51 52 53 54 55 56 57 58
# Turn on notifications
require 'active_support/notifications'
Thread.abort_on_exception = true

ActiveSupport::Notifications.subscribe do |*args|
  ActionController::Base.log_event(*args) if ActionController::Base.logger
end

J
Joshua Peek 已提交
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
module SetupOnce
  extend ActiveSupport::Concern

  included do
    cattr_accessor :setup_once_block
    self.setup_once_block = nil

    setup :run_setup_once
  end

  module ClassMethods
    def setup_once(&block)
      self.setup_once_block = block
    end
  end

  private
    def run_setup_once
      if self.setup_once_block
        self.setup_once_block.call
        self.setup_once_block = nil
      end
    end
end

class ActiveSupport::TestCase
  include SetupOnce

  # Hold off drawing routes until all the possible controller classes
  # have been loaded.
  setup_once do
    ActionController::Routing::Routes.draw do |map|
J
Joshua Peek 已提交
91
      match ':controller(/:action(/:id))'
J
Joshua Peek 已提交
92 93 94 95
    end
  end
end

96 97 98 99 100 101 102 103 104 105 106 107
class MockLogger
  attr_accessor :level

  def initialize
    @level = Logger::DEBUG
    @logged = []
  end

  def method_missing(method, *args, &blk)
    @logged << args.first
    @logged << blk.call if block_given?
  end
J
José Valim 已提交
108 109 110 111

  def logged
    @logged.compact.map { |l| l.to_s.strip }
  end
112 113
end

114
class ActionController::IntegrationTest < ActiveSupport::TestCase
115 116 117 118 119 120 121 122 123 124 125
  def self.build_app(routes = nil)
    ActionDispatch::MiddlewareStack.new { |middleware|
      middleware.use "ActionDispatch::ShowExceptions"
      middleware.use "ActionDispatch::Callbacks"
      middleware.use "ActionDispatch::ParamsParser"
      middleware.use "Rack::Head"
    }.build(routes || ActionController::Routing::Routes)
  end

  self.app = build_app

126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145
  class StubDispatcher
    def self.new(*args)
      lambda { |env|
        params = env['action_dispatch.request.path_parameters']
        controller, action = params[:controller], params[:action]
        [200, {'Content-Type' => 'text/html'}, ["#{controller}##{action}"]]
      }
    end
  end

  def self.stub_controllers
    old_dispatcher = ActionDispatch::Routing::RouteSet::Dispatcher
    ActionDispatch::Routing::RouteSet.module_eval { remove_const :Dispatcher }
    ActionDispatch::Routing::RouteSet.module_eval { const_set :Dispatcher, StubDispatcher }
    yield ActionDispatch::Routing::RouteSet.new
  ensure
    ActionDispatch::Routing::RouteSet.module_eval { remove_const :Dispatcher }
    ActionDispatch::Routing::RouteSet.module_eval { const_set :Dispatcher, old_dispatcher }
  end

146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161
  def with_routing(&block)
    real_routes = ActionController::Routing::Routes
    ActionController::Routing.module_eval { remove_const :Routes }

    temporary_routes = ActionController::Routing::RouteSet.new
    self.class.app = self.class.build_app(temporary_routes)
    ActionController::Routing.module_eval { const_set :Routes, temporary_routes }

    yield temporary_routes
  ensure
    if ActionController::Routing.const_defined? :Routes
      ActionController::Routing.module_eval { remove_const :Routes }
    end
    ActionController::Routing.const_set(:Routes, real_routes) if real_routes
    self.class.app = self.class.build_app
  end
162 163
end

J
Joshua Peek 已提交
164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214
# Temporary base class
class Rack::TestCase < ActionController::IntegrationTest
  setup do
    ActionController::Base.session_options[:key] = "abc"
    ActionController::Base.session_options[:secret] = ("*" * 30)
  end

  def self.testing(klass = nil)
    if klass
      @testing = "/#{klass.name.underscore}".sub!(/_controller$/, '')
    else
      @testing
    end
  end

  def get(thing, *args)
    if thing.is_a?(Symbol)
      super("#{self.class.testing}/#{thing}", *args)
    else
      super
    end
  end

  def assert_body(body)
    assert_equal body, Array.wrap(response.body).join
  end

  def assert_status(code)
    assert_equal code, response.status
  end

  def assert_response(body, status = 200, headers = {})
    assert_body   body
    assert_status status
    headers.each do |header, value|
      assert_header header, value
    end
  end

  def assert_content_type(type)
    assert_equal type, response.headers["Content-Type"]
  end

  def assert_header(name, value)
    assert_equal value, response.headers[name]
  end
end

class ::ApplicationController < ActionController::Base
end

215 216 217 218
module ActionController
  class Base
    include ActionController::Testing
  end
219

220
  Base.view_paths = FIXTURE_LOAD_PATH
221

222
  class TestCase
J
Joshua Peek 已提交
223
    include ActionDispatch::TestProcess
224

225 226 227
    def assert_template(options = {}, message = nil)
      validate_request!

228
      hax = @controller.view_context.instance_variable_get(:@_rendered)
229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247

      case options
      when NilClass, String
        rendered = (hax[:template] || []).map { |t| t.identifier }
        msg = build_message(message,
                "expecting <?> but rendering with <?>",
                options, rendered.join(', '))
        assert_block(msg) do
          if options.nil?
            hax[:template].blank?
          else
            rendered.any? { |t| t.match(options) }
          end
        end
      when Hash
        if expected_partial = options[:partial]
          partials = hax[:partials]
          if expected_count = options[:count]
            found = partials.detect { |p, _| p.identifier.match(expected_partial) }
248
            actual_count = found.nil? ? 0 : found[1]
249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266
            msg = build_message(message,
                    "expecting ? to be rendered ? time(s) but rendered ? time(s)",
                     expected_partial, expected_count, actual_count)
            assert(actual_count == expected_count.to_i, msg)
          else
            msg = build_message(message,
                    "expecting partial <?> but action rendered <?>",
                    options[:partial], partials.keys)
            assert(partials.keys.any? { |p| p.identifier.match(expected_partial) }, msg)
          end
        else
          assert hax[:partials].empty?,
            "Expected no partials to be rendered"
        end
      end
    end
  end
end