test: add unit test for RestExceptionHandler.goalsNotFound

Author: felipemelozx

src/main/java/fun/trackmoney/config/exception/RestExceptionHandler.java

@@ -3,6 +3,7 @@
 import fun.trackmoney.account.exception.AccountNotFoundException;
 import fun.trackmoney.auth.exception.LoginException;
 import fun.trackmoney.category.exception.CategoryNotFoundException;
+import fun.trackmoney.goal.exception.GoalsNotFoundException;
 import fun.trackmoney.transaction.exception.TransactionNotFoundException;
 import fun.trackmoney.user.exception.EmailAlreadyExistsException;
 import fun.trackmoney.user.exception.UserNotFoundException;
@@ -92,4 +93,12 @@ public ResponseEntity<ApiResponse<List<CustomFieldError>>> transactionNotFound(T
         .body(new ApiResponse<>(false, ex.getMessage(), null, ex.getErrors())
         );
   }
+
+  @ExceptionHandler(GoalsNotFoundException.class)
+  public ResponseEntity<ApiResponse<List<CustomFieldError>>> goalsNotFound(GoalsNotFoundException ex) {
+    return ResponseEntity
+        .status(HttpStatus.NOT_FOUND)
+        .body(new ApiResponse<>(false, ex.getMessage(), null, ex.getErrors())
+        );
+  }
 }

src/test/java/fun/trackmoney/config/exception/RestExceptionHandlerTest.java

@@ -3,6 +3,7 @@
 import fun.trackmoney.account.exception.AccountNotFoundException;
 import fun.trackmoney.auth.exception.LoginException;
 import fun.trackmoney.category.exception.CategoryNotFoundException;
+import fun.trackmoney.goal.exception.GoalsNotFoundException;
 import fun.trackmoney.transaction.exception.TransactionNotFoundException;
 import fun.trackmoney.user.exception.EmailAlreadyExistsException;
 import fun.trackmoney.user.exception.UserNotFoundException;
@@ -193,4 +194,22 @@ void TransactionNotFound_shouldReturnNotFoundWithError() {
     assertEquals("Transaction not found.", error.getMessage());
   }
 
+  @Test
+  void goalsNotFound_shouldReturnNotFoundWithError() {
+    GoalsNotFoundException exception = new GoalsNotFoundException("Goals not found.");
+    ResponseEntity<ApiResponse<List<CustomFieldError>>> response = restExceptionHandler.goalsNotFound(exception);
+
+    assertEquals(HttpStatus.NOT_FOUND, response.getStatusCode());
+    ApiResponse<List<CustomFieldError>> apiResponse = response.getBody();
+    assertNotNull(apiResponse);
+    assertFalse(apiResponse.isSuccess());
+    assertEquals("Goals not found.", apiResponse.getMessage());
+    assertNull(apiResponse.getData());
+    assertNotNull(apiResponse.getErrors());
+    assertEquals(1, (apiResponse.getErrors()).size());
+    CustomFieldError error = apiResponse.getErrors().get(0);
+    assertEquals("Goal", error.getField());
+    assertEquals("Goals not found.", error.getMessage());
+  }
+
 }