Adding context to find, closes drapergem#28

ersatzryan · Oct 3, 2011 · 1e28686 · 1e28686
1 parent f1d61a6
commit 1e28686
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/lib/draper/base.rb b/lib/draper/base.rb
@@ -32,8 +32,8 @@ def initialize(input, context = {})
     #
     # @param [Symbol or String] id to lookup
     # @return [Object] instance of this decorator class
-    def self.find(input)
-      self.new(model_class.find(input))
+    def self.find(input, context = {})
+      self.new(model_class.find(input), context)
     end
 
     # Typically called within a decorator definition, this method

diff --git a/spec/base_spec.rb b/spec/base_spec.rb
@@ -89,6 +89,11 @@ class BusinessDecorator < Draper::Base
       pd.should be_instance_of(ProductDecorator)
       pd.model.should be_instance_of(Product)
     end
+
+    it "should accept and store a context" do
+      pd = ProductDecorator.find(1, :admin)
+      pd.context.should == :admin
+    end
   end
 
   context ".decorate" do