lib/mongo_mapper/railtie.rb in mongo_mapper-0.13.0.beta2 vs lib/mongo_mapper/railtie.rb in mongo_mapper-0.13.0

@@ -1,13 +1,33 @@
 require "mongo_mapper"
 require "rails"
 require "active_model/railtie"
 
+# Need the action_dispatch railtie to have action_dispatch.rescu_responses initialized correctly
+require "action_dispatch/railtie"
+
 module MongoMapper
   # = MongoMapper Railtie
   class Railtie < Rails::Railtie
 
     config.mongo_mapper = ActiveSupport::OrderedOptions.new
+
+    # Rescue responses similar to ActiveRecord.
+    # For rails 3.0 and 3.1
+    if Rails.version < "3.2"
+      ActionDispatch::ShowExceptions.rescue_responses['MongoMapper::DocumentNotFound']  = :not_found,
+      ActionDispatch::ShowExceptions.rescue_responses['MongoMapper::InvalidKey']        = :unprocessable_entity,
+      ActionDispatch::ShowExceptions.rescue_responses['MongoMapper::InvalidScheme']     = :unprocessable_entity,
+      ActionDispatch::ShowExceptions.rescue_responses['MongoMapper::NotSupported']      = :unprocessable_entity
+    else
+      # For rails 3.2 and 4.0
+      config.action_dispatch.rescue_responses.merge!(
+          'MongoMapper::DocumentNotFound'  => :not_found,
+          'MongoMapper::InvalidKey'        => :unprocessable_entity,
+          'MongoMapper::InvalidScheme'     => :unprocessable_entity,
+          'MongoMapper::NotSupported'      => :unprocessable_entity
+        )
+    end
 
     rake_tasks do
       load "mongo_mapper/railtie/database.rake"
     end