lib/locomotive/mongoid/patches.rb in locomotivecms-3.0.1 vs lib/locomotive/mongoid/patches.rb in locomotivecms-3.1.0.rc1

@@ -97,9 +97,17 @@
       attrs
     end
   end
 
   class Criteria
+    def first!
+      self.first.tap do |model|
+        if model.nil?
+          raise Mongoid::Errors::DocumentNotFound.new(self.klass, self.selector)
+        end
+      end
+    end
+
     def without_sorting
       clone.tap { |crit| crit.options.delete(:sort) }
     end
 
     # http://code.dblock.org/paging-and-iterating-over-large-mongo-collections