0
@@ -1,5 +1,7 @@
0
 *2.3.0 [Edge]*
0
 
0
+* Add :allow_nil option to delegate. #1127 [Sergio Gil]
0
+
0
 * Add Benchmark.ms convenience method to benchmark realtime in milliseconds.  [Jeremy Kemper]
0
 
0
 * Updated included memcache-client to the 1.5.0.5 version which includes fixes from fiveruns and 37signals to deal with failover and timeouts #1535 [Joshua Sierles]

0
@@ -72,6 +72,30 @@ class Module
0
   #   invoice.customer_name    # => "John Doe"
0
   #   invoice.customer_address # => "Vimmersvej 13"
0
   #
0
+  # If the object to which you delegate can be nil, you may want to use the
0
+  # :allow_nil option. In that case, it returns nil instead of raising a
0
+  # NoMethodError exception:
0
+  #
0
+  #  class Foo
0
+  #    attr_accessor :bar
0
+  #    def initialize(bar = nil)
0
+  #      @bar = bar
0
+  #    end
0
+  #    delegate :zoo, :to => :bar
0
+  #  end
0
+  #
0
+  #  Foo.new.zoo   # raises NoMethodError exception (you called nil.zoo)
0
+  #
0
+  #  class Foo
0
+  #    attr_accessor :bar
0
+  #    def initialize(bar = nil)
0
+  #      @bar = bar
0
+  #    end
0
+  #    delegate :zoo, :to => :bar, :allow_nil => true
0
+  #  end
0
+  #
0
+  #  Foo.new.zoo   # returns nil
0
+  #
0
   def delegate(*methods)
0
     options = methods.pop
0
     unless options.is_a?(Hash) && to = options[:to]
0
@@ -84,10 +108,12 @@ class Module
0
 
0
     prefix = options[:prefix] && "#{options[:prefix] == true ? to : options[:prefix]}_"
0
 
0
+    allow_nil = options[:allow_nil] && "#{to} && "
0
+
0
     methods.each do |method|
0
       module_eval(<<-EOS, "(__DELEGATION__)", 1)
0
         def #{prefix}#{method}(*args, &block)
0
-          #{to}.__send__(#{method.inspect}, *args, &block)
0
+          #{allow_nil}#{to}.__send__(#{method.inspect}, *args, &block)
0
         end
0
       EOS
0
     end

0
@@ -41,6 +41,10 @@ Invoice   = Struct.new(:client) do
0
   delegate :street, :city, :name, :to => :client, :prefix => :customer
0
 end
0
 
0
+Project   = Struct.new(:description, :person) do
0
+  delegate :name, :to => :person, :allow_nil => true
0
+end
0
+
0
 class Name
0
   delegate :upcase, :to => :@full_name
0
 
0
@@ -117,6 +121,29 @@ class ModuleTest < Test::Unit::TestCase
0
     end
0
   end
0
 
0
+  def test_delegation_with_allow_nil
0
+    rails = Project.new("Rails", Someone.new("David"))
0
+    assert_equal rails.name, "David"
0
+  end
0
+
0
+  def test_delegation_with_allow_nil_and_nil_value
0
+    rails = Project.new("Rails")
0
+    assert_nil rails.name
0
+  end
0
+
0
+  def test_delegation_with_allow_nil_and_nil_value_and_prefix
0
+    Project.class_eval do
0
+      delegate :name, :to => :person, :allow_nil => true, :prefix => true
0
+    end
0
+    rails = Project.new("Rails")
0
+    assert_nil rails.person_name
0
+  end
0
+
0
+  def test_delegation_without_allow_nil_and_nil_value
0
+    david = Someone.new("David")
0
+    assert_raises(NoMethodError) { david.street }
0
+  end
0
+
0
   def test_parent
0
     assert_equal Yz::Zy, Yz::Zy::Cd.parent
0
     assert_equal Yz, Yz::Zy.parent