Skip to content

Commit

Permalink
Clear out AS callback method pollution in AC::Base.action_methods
Browse files Browse the repository at this point in the history
  • Loading branch information
josh committed Jan 18, 2010
1 parent 58fe329 commit c29bb88
Show file tree
Hide file tree
Showing 2 changed files with 39 additions and 9 deletions.
20 changes: 12 additions & 8 deletions actionpack/lib/abstract_controller/base.rb
Expand Up @@ -62,15 +62,19 @@ def hidden_actions
# Array[String]:: A list of all methods that should be considered # Array[String]:: A list of all methods that should be considered
# actions. # actions.
def action_methods def action_methods
@action_methods ||= @action_methods ||= begin
# All public instance methods of this class, including ancestors # All public instance methods of this class, including ancestors
public_instance_methods(true).map { |m| m.to_s }.to_set - methods = public_instance_methods(true).map { |m| m.to_s }.to_set -
# Except for public instance methods of Base and its ancestors # Except for public instance methods of Base and its ancestors
internal_methods.map { |m| m.to_s } + internal_methods.map { |m| m.to_s } +
# Be sure to include shadowed public instance methods of this class # Be sure to include shadowed public instance methods of this class
public_instance_methods(false).map { |m| m.to_s } - public_instance_methods(false).map { |m| m.to_s } -
# And always exclude explicitly hidden actions # And always exclude explicitly hidden actions
hidden_actions hidden_actions

# Clear out AS callback method pollution
methods.reject { |method| method =~ /_one_time_conditions/ }
end
end end


# Returns the full controller name, underscored, without the ending Controller. # Returns the full controller name, underscored, without the ending Controller.
Expand Down
28 changes: 27 additions & 1 deletion actionpack/test/controller/new_base/base_test.rb
Expand Up @@ -3,6 +3,8 @@
# Tests the controller dispatching happy path # Tests the controller dispatching happy path
module Dispatching module Dispatching
class SimpleController < ActionController::Base class SimpleController < ActionController::Base
before_filter :authenticate

def index def index
render :text => "success" render :text => "success"
end end
Expand All @@ -12,12 +14,20 @@ def modify_response_body
end end


def modify_response_body_twice def modify_response_body_twice
ret = (self.response_body = "success") ret = (self.response_body = "success")
self.response_body = "#{ret}!" self.response_body = "#{ret}!"
end end


def modify_response_headers def modify_response_headers
end end

def show_actions
render :text => "actions: #{action_methods.to_a.join(', ')}"
end

protected
def authenticate
end
end end


class EmptyController < ActionController::Base ; end class EmptyController < ActionController::Base ; end
Expand Down Expand Up @@ -64,5 +74,21 @@ class BaseTest < Rack::TestCase
assert_equal 'empty', EmptyController.controller_name assert_equal 'empty', EmptyController.controller_name
assert_equal 'contained_empty', Submodule::ContainedEmptyController.controller_name assert_equal 'contained_empty', Submodule::ContainedEmptyController.controller_name
end end

test "action methods" do
assert_equal Set.new(%w(
modify_response_headers
modify_response_body_twice
index
modify_response_body
show_actions
)), SimpleController.action_methods

assert_equal Set.new, EmptyController.action_methods
assert_equal Set.new, Submodule::ContainedEmptyController.action_methods

get "/dispatching/simple/show_actions"
assert_body "actions: modify_response_headers, modify_response_body_twice, index, modify_response_body, show_actions"
end
end end
end end

0 comments on commit c29bb88

Please sign in to comment.