From 7ca39df056544099e669d952d49b9e780149edab Mon Sep 17 00:00:00 2001
From: Joze RIHTARSIC <joze.rihtarsic@ext.ec.europa.eu>
Date: Wed, 18 Jul 2018 13:41:01 +0200
Subject: [PATCH] update exception handler

---
 .../smp/error/ErrorMappingControllerAdvice.java        | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/smp-webapp/src/main/java/eu/europa/ec/edelivery/smp/error/ErrorMappingControllerAdvice.java b/smp-webapp/src/main/java/eu/europa/ec/edelivery/smp/error/ErrorMappingControllerAdvice.java
index 3fdc1bcaa..b3dfc1b03 100644
--- a/smp-webapp/src/main/java/eu/europa/ec/edelivery/smp/error/ErrorMappingControllerAdvice.java
+++ b/smp-webapp/src/main/java/eu/europa/ec/edelivery/smp/error/ErrorMappingControllerAdvice.java
@@ -14,10 +14,7 @@
 package eu.europa.ec.edelivery.smp.error;
 
 import ec.services.smp._1.ErrorResponse;
-import eu.europa.ec.edelivery.smp.exceptions.NotFoundException;
-import eu.europa.ec.edelivery.smp.exceptions.UnknownUserException;
-import eu.europa.ec.edelivery.smp.exceptions.WrongInputFieldException;
-import eu.europa.ec.edelivery.smp.exceptions.XmlParsingException;
+import eu.europa.ec.edelivery.smp.exceptions.*;
 import eu.europa.ec.edelivery.smp.error.exceptions.BadRequestException;
 import eu.europa.ec.smp.api.exceptions.MalformedIdentifierException;
 import eu.europa.ec.smp.api.exceptions.XmlInvalidAgainstSchemaException;
@@ -87,6 +84,11 @@ public class ErrorMappingControllerAdvice {
         return buildAndWarn(BAD_REQUEST, USER_NOT_FOUND, ex.getMessage(), ex);
     }
 
+    @ExceptionHandler(InvalidOwnerException.class)
+    public ResponseEntity handleUnknownUserException(InvalidOwnerException ex) {
+        return buildAndWarn(BAD_REQUEST, ErrorBusinessCode.UNAUTHORIZED, ex.getMessage(), ex);
+    }
+
     @ExceptionHandler(XmlParsingException.class)
     public ResponseEntity handleXmlParsingException(XmlParsingException ex) {
         return buildAndWarn(BAD_REQUEST, XSD_INVALID, ex.getMessage(), ex);
-- 
GitLab