Merge pull request #182 from brainspec/issue-165

Uses after_initialize callback to set default value in mongoid documents as well.

lib/enumerize.rb

@@ -13,6 +13,7 @@ module Enumerize
   autoload :ModuleAttributes, 'enumerize/module_attributes'
 
   autoload :ActiveRecordSupport, 'enumerize/activerecord'
+  autoload :MongoidSupport,      'enumerize/mongoid'
 
   module Scope
     autoload :ActiveRecord, 'enumerize/scope/activerecord'
@@ -27,9 +28,16 @@ def self.included(base)
   def self.extended(base)
     base.send :include, Enumerize::Base
     base.extend Enumerize::Predicates
-    base.extend Enumerize::ActiveRecordSupport
-    base.extend Enumerize::Scope::ActiveRecord if defined?(::ActiveRecord::Base)
-    base.extend Enumerize::Scope::Mongoid      if defined?(::Mongoid::Document)
+
+    if defined?(::ActiveRecord::Base)
+      base.extend Enumerize::ActiveRecordSupport
+      base.extend Enumerize::Scope::ActiveRecord
+    end
+
+    if defined?(::Mongoid::Document)
+      base.extend Enumerize::MongoidSupport
+      base.extend Enumerize::Scope::Mongoid
+    end
 
     if defined?(::RailsAdmin)
       require 'enumerize/integrations/rails_admin'

lib/enumerize/activerecord.rb

@@ -4,7 +4,7 @@ def enumerize(name, options={})
       super
 
       _enumerize_module.dependent_eval do
-        if defined?(::ActiveRecord::Base) && self < ::ActiveRecord::Base
+        if self < ::ActiveRecord::Base
           include InstanceMethods
 
           # Since Rails use `allocate` method on models and initializes them with `init_with` method.

lib/enumerize/mongoid.rb

@@ -0,0 +1,13 @@
+module Enumerize
+  module MongoidSupport
+    def enumerize(name, options={})
+      super
+
+      _enumerize_module.dependent_eval do
+        if self < ::Mongoid::Document
+          after_initialize :_set_default_value_for_enumerized_attributes
+        end
+      end
+    end
+  end
+end

test/mongoid_test.rb

@@ -55,6 +55,14 @@ class MongoidUser
     model.new.role.must_equal 'user'
   end
 
+  it 'uses after_initialize callback to set default value' do
+    model.delete_all
+    model.create!(sex: 'male', role: nil)
+
+    user = model.where(sex: 'male').first
+    user.role.must_equal 'user'
+  end
+
   it 'validates inclusion' do
     user = model.new
     user.role = 'wrong'