Include Decoratable in Mongoid::Document

Closes #375
This commit is contained in:
Andrew Haines 2012-12-11 20:59:49 +00:00
parent 7efeaa492b
commit 051dca6f30
2 changed files with 6 additions and 4 deletions

View File

@ -37,7 +37,7 @@ module Draper
end
end
def self.setup_active_record(base)
def self.setup_orm(base)
base.class_eval do
include Draper::Decoratable
end

View File

@ -30,9 +30,11 @@ module Draper
end
end
initializer "draper.setup_active_record" do |app|
ActiveSupport.on_load :active_record do
Draper.setup_active_record self
initializer "draper.setup_orm" do |app|
[:active_record, :mongoid].each do |orm|
ActiveSupport.on_load orm do
Draper.setup_orm self
end
end
end