Merge 1f486a6 into f32359e

sul-dlss · Sep 6, 2019 · c16f3c8 · c16f3c8
2 parents f32359e + 1f486a6
commit c16f3c8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/app/controllers/objects_controller.rb b/app/controllers/objects_controller.rb
@@ -3,6 +3,10 @@
 class ObjectsController < ApplicationController
   before_action :load_item, except: [:create]
 
+  rescue_from(ActiveFedora::ObjectNotFoundError) do |e|
+    render status: :not_found, plain: e.message
+  end
+
   rescue_from(Dor::ParameterError) do |e|
     render status: :bad_request, plain: e.message
   end

diff --git a/spec/controllers/objects_controller_spec.rb b/spec/controllers/objects_controller_spec.rb
@@ -48,6 +48,13 @@
         expect(response.status).to eq(500)
         expect(response.body).to eq(errmsg)
       end
+
+      it 'returns a 404 error for ActiveFedora::ObjectNotFoundError' do
+        allow(RegistrationService).to receive(:register_object).and_raise(ActiveFedora::ObjectNotFoundError.new(errmsg))
+        post :create
+        expect(response.status).to eq(404)
+        expect(response.body).to eq(errmsg)
+      end
     end
 
     it 'registers the object with the registration service' do