Added test case for IllegalArgumentException

ita-social-projects · Oct 1, 2024 · 23ac181 · 23ac181
1 parent e8bb312
commit 23ac181
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/core/src/test/java/greencity/exception/handler/CustomExceptionHandlerTest.java b/core/src/test/java/greencity/exception/handler/CustomExceptionHandlerTest.java
@@ -139,6 +139,16 @@ void testHandleBadRefreshTokenException() {
             ResponseEntity.status(HttpStatus.BAD_REQUEST).body(exceptionResponse));
     }
 
+    @Test
+    void testHandleIllegalTokenException() {
+        IllegalArgumentException illegalArgumentException = new IllegalArgumentException("test");
+        ExceptionResponse exceptionResponse = new ExceptionResponse(objectMap);
+        when(errorAttributes.getErrorAttributes(eq(webRequest),
+            any(ErrorAttributeOptions.class))).thenReturn(objectMap);
+        assertEquals(customExceptionHandler.handleIllegalArgumentException(illegalArgumentException, webRequest),
+            ResponseEntity.status(HttpStatus.BAD_REQUEST).body(exceptionResponse));
+    }
+
     @Test
     void handleBadRequestException() {
         BadRequestException badRequestException = new BadRequestException("test");