Merge pull request drapergem#186 from nisargshah100/enumerable_proxy_…

…context added context to enumerable
nashby · May 8, 2012 · b60d195 · b60d195
2 parents 15992b8 + f9f0982
commit b60d195
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/lib/draper/decorated_enumerable_proxy.rb b/lib/draper/decorated_enumerable_proxy.rb
@@ -38,6 +38,10 @@ def [](index)
     def to_s
       "#<DecoratedEnumerableProxy of #{@klass} for #{@wrapped_collection.inspect}>"
     end
+
+    def context=(input)
+      self.map { |member| member.context = input }
+    end
 
     def source
       @wrapped_collection

diff --git a/spec/draper/base_spec.rb b/spec/draper/base_spec.rb
@@ -313,6 +313,11 @@ class CustomDecorator < Draper::Base
         it "returns a collection of wrapped objects" do
           subject.each{ |decorated| decorated.should be_instance_of(Draper::Base) }
         end
+
+        it 'should accepted and store a context for a collection' do
+          subject.context = :admin
+          subject.each { |decorated| decorated.context.should == :admin }
+        end
       end
 
       context "when given a single source object" do