Merge branch 'master' of github.com:jcasimir/draper

drapergem · Nov 30, 2011 · 58a733e · 58a733e
2 parents 5e58b82 + 524fc97
commit 58a733e
Show file tree

Hide file tree

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

diff --git a/spec/draper/base_spec.rb b/spec/draper/base_spec.rb
@@ -310,6 +310,12 @@ module Paginator; def page_number; "magic_value"; end; end
       subject_one.should_not == subject_two
     end
 
+    it "should allow decorated access by index" do
+      subject = ProductDecorator.decorate(paged_array)
+
+      subject[0].should be_instance_of ProductDecorator
+    end
+
     context '#all' do
       it "should return a decorated collection" do
         ProductDecorator.all.first.should be_instance_of ProductDecorator