Adding context to find, closes #28

lib/draper/base.rb

@@ -32,8 +32,8 @@ module Draper
     #
     # @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

spec/base_spec.rb

@@ -89,6 +89,11 @@ describe Draper::Base do
       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