abstract_unit.rb 7.5 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
require 'fixture_template'
22
require 'active_support/test_case'
J
Joshua Peek 已提交
23
require 'action_view/test_case'
24
require 'active_support/dependencies'
D
Initial  
David Heinemeier Hansson 已提交
25

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

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

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

40
ActiveSupport::Dependencies.hook!
41

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

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

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

53 54 55 56 57 58 59 60
# 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 已提交
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 91 92
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 已提交
93
      match ':controller(/:action(/:id))'
J
Joshua Peek 已提交
94 95 96 97
    end
  end
end

98 99 100 101 102 103 104 105 106 107 108 109
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 已提交
110 111 112 113

  def logged
    @logged.compact.map { |l| l.to_s.strip }
  end
114 115
end

116
class ActionController::IntegrationTest < ActiveSupport::TestCase
117 118
  def self.build_app(routes = nil)
    ActionDispatch::MiddlewareStack.new { |middleware|
119
      middleware.use "ActionDispatch::StringCoercion"
120 121 122 123 124 125 126 127 128
      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

129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148
  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

149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164
  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
165 166
end

J
Joshua Peek 已提交
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 215 216 217
# 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

218 219 220 221
module ActionController
  class Base
    include ActionController::Testing
  end
222

223
  Base.view_paths = FIXTURE_LOAD_PATH
224

225
  class TestCase
J
Joshua Peek 已提交
226
    include ActionDispatch::TestProcess
227

228 229 230
    def assert_template(options = {}, message = nil)
      validate_request!

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

      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) }
251
            actual_count = found.nil? ? 0 : found[1]
252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269
            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