lib/mongoid/lock.rb in mongoid-lock-0.0.2 vs lib/mongoid/lock.rb in mongoid-lock-0.0.3

@@ -1,5 +1,6 @@
+require 'mongoid'
 require 'mongoid/lock/synch_methods'
 module Mongoid
   module Lock
 #------------------------------------------------------------------------------
 # => Embedded class for Unsynchronized use Exceptions
@@ -23,10 +24,14 @@
 # => Module inclusion hook
 #------------------------------------------------------------------------------
     def self.included(base)
       super if defined?(super)
       base.class_eval do
-        # => Add special fields for use
+        # => Extend Document, If not Already Extended
+        unless (base.included_modules.include?(Mongoid::Document))
+          include Mongoid::Document
+        end
+        # => Add special field
         field :lock_used_by, :type => String, :default => nil
       end
     end
   end
 end
\ No newline at end of file