Added delegation support to Module that allows multiple delegations at once (unlike Forwardable in the stdlib) [DHH]

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@3535 5ecf4fe2-1ee6-0310-87b1-e25e094e27de

1 files changed, 48 insertions, 0 deletions

diff --git a/activesupport/test/core_ext/module_test.rb b/activesupport/test/core_ext/module_test.rb
index d6ea595ae2..464196a0ec 100644
--- a/activesupport/test/core_ext/module_test.rb
+++ b/activesupport/test/core_ext/module_test.rb
@@ -26,6 +26,34 @@ end
 class De
 end
 
+Somewhere = Struct.new(:street, :city)
+
+Someone   = Struct.new(:name, :place) do
+  delegate :street, :city, :to => :place
+  delegate :state, :to => :@place
+  delegate :upcase, :to => "place.city"
+end
+
+class Name
+  delegate :upcase, :to => :@full_name
+
+  def initialize(first, last)
+    @full_name = "#{first} #{last}"
+  end
+end
+
+$nowhere = <<-EOF
+class Name
+  delegate :nowhere
+end
+EOF
+
+$noplace = <<-EOF
+class Name
+  delegate :noplace, :tos => :hollywood
+end
+EOF
+
 class ModuleTest < Test::Unit::TestCase
   def test_included_in_classes
     assert One.included_in_classes.include?(Ab)
@@ -34,4 +62,24 @@ class ModuleTest < Test::Unit::TestCase
     assert !One.included_in_classes.include?(De)
   end
 
+  def test_delegation_to_methods
+    david = Someone.new("David", Somewhere.new("Paulina", "Chicago"))
+    assert_equal "Paulina", david.street
+    assert_equal "Chicago", david.city
+  end
+  
+  def test_delegation_down_hierarchy
+    david = Someone.new("David", Somewhere.new("Paulina", "Chicago"))
+    assert_equal "CHICAGO", david.upcase
+  end
+  
+  def test_delegation_to_instance_variable
+    david = Name.new("David", "Hansson")
+    assert_equal "DAVID HANSSON", david.upcase
+  end
+  
+  def test_missing_delegation_target
+    assert_raises(ArgumentError) { eval($nowhere) }
+    assert_raises(ArgumentError) { eval($noplace) }
+  end
 end
\ No newline at end of file