Merge pull request drapergem#71 from ayamomiji/namespaced-model

Namespaced model support

Author: jcasimir

lib/draper/base.rb

@@ -47,8 +47,8 @@ def self.find(input, context = {})
     # to query.
     #
     # @param [Symbol] class_name snakecase name of the decorated class, like `:product`
-    def self.decorates(input)
-      self.model_class = input.to_s.camelize.constantize
+    def self.decorates(input, options = {})
+      self.model_class = options[:class] || input.to_s.camelize.constantize
       model_class.send :include, Draper::ModelSupport
       define_method(input){ @model }
     end

spec/draper/base_spec.rb

@@ -58,6 +58,20 @@ class BusinessDecorator < Draper::Base
       pd = ProductDecorator.new(source)
       pd.send(:product).should == source
     end
+
+    context("namespaced model supporting") do
+      let(:source){ Namespace::Product.new }
+
+      it "sets the model class for the decorator" do
+        decorator = Namespace::ProductDecorator.new(source)
+        decorator.model_class.should == Namespace::Product
+      end
+
+      it "creates a named accessor for the wrapped model" do
+        pd = Namespace::ProductDecorator.new(source)
+        pd.send(:product).should == source
+      end
+    end
   end
 
   context(".model / .to_model") do

spec/draper/model_support_spec.rb

@@ -26,4 +26,14 @@
       subject.decorate.to_ary[0].model.should be_a(Product)
     end
   end
+
+  describe '#decorate - decorate collections of namespaced AR objects' do
+    subject { Namespace::Product.limit }
+    its(:decorate) { should be_kind_of(Draper::DecoratedEnumerableProxy) }
+
+    it "should decorate the collection" do
+      subject.decorate.size.should == 1
+      subject.decorate.to_ary[0].model.should be_a(Namespace::Product)
+    end
+  end
 end

spec/support/samples/namespaced_product.rb

@@ -0,0 +1,51 @@
+require './spec/support/samples/product'
+
+module Namespace
+  class Product < ActiveRecord::Base
+    include Draper::ModelSupport
+
+    def self.first
+      @@first ||= Namespace::Product.new
+    end
+
+    def self.last
+      @@last ||= Namespace::Product.new
+    end
+
+    def self.all
+      [Namespace::Product.new, Namespace::Product.new]
+    end
+
+    def self.scoped
+      [Namespace::Product.new]
+    end
+
+    def self.model_name
+      "Namespace::Product"
+    end
+
+    def self.find(id)
+      return Namespace::Product.new
+    end
+
+    def self.sample_class_method
+      "sample class method"
+    end
+
+    def hello_world
+      "Hello, World"
+    end
+
+    def goodnight_moon
+      "Goodnight, Moon"
+    end
+
+    def title
+      "Sample Title"
+    end
+
+    def block
+      yield
+    end
+  end
+end

spec/support/samples/namespaced_product_decorator.rb

@@ -0,0 +1,7 @@
+require './spec/support/samples/product'
+
+module Namespace
+  class ProductDecorator < Draper::Base
+    decorates :product, :class => Namespace::Product
+  end
+end