Skip to content

Commit

Permalink
Ensure only delegations to methods can have an automatic prefix. [#1235
Browse files Browse the repository at this point in the history
… state:resolved]

Signed-off-by: Pratik Naik <pratiknaik@gmail.com>
  • Loading branch information
dasch authored and lifo committed Nov 20, 2008
1 parent 206c564 commit a5156d9
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 0 deletions.
Expand Up @@ -78,6 +78,10 @@ def delegate(*methods)
raise ArgumentError, "Delegation needs a target. Supply an options hash with a :to key as the last argument (e.g. delegate :hello, :to => :greeter)."
end

if options[:prefix] == true && options[:to].to_s =~ /^[^a-z_]/
raise ArgumentError, "Can only automatically set the delegation prefix when delegating to a method."
end

prefix = options[:prefix] && "#{options[:prefix] == true ? to : options[:prefix]}_"

methods.each do |method|
Expand Down
11 changes: 11 additions & 0 deletions activesupport/test/core_ext/module_test.rb
Expand Up @@ -106,6 +106,17 @@ def test_delegation_custom_prefix
assert_equal invoice.customer_city, "Chicago"
end

def test_delegation_prefix_with_instance_variable
assert_raise ArgumentError do
Class.new do
def initialize(client)
@client = client
end
delegate :name, :address, :to => :@client, :prefix => true
end
end
end

def test_parent
assert_equal Yz::Zy, Yz::Zy::Cd.parent
assert_equal Yz, Yz::Zy.parent
Expand Down

0 comments on commit a5156d9

Please sign in to comment.