elseif presistance

rubyist · Dec 16, 2011 · 53ea701 · 53ea701
1 parent 1359605
commit 53ea701
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/aasm/persistence.rb b/lib/aasm/persistence.rb
@@ -9,6 +9,9 @@ def self.set_persistence(base)
       if hierarchy.include?("ActiveRecord::Base")
         require File.join(File.dirname(__FILE__), 'persistence', 'active_record_persistence')
         base.send(:include, AASM::Persistence::ActiveRecordPersistence)
+      elsif hierarchy.include?("Mongoid::Document")
+        require File.join(File.dirname(__FILE__), 'persistence', 'mongoid_persistence')
+        base.send(:include, AASM::Persistence::MongoidPersistence)
       end
     end
   end