[#225] 테스트 코드 수정

Kernel360 · Sep 5, 2024 · 850014d · 850014d
1 parent 508472d
commit 850014d
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/main/java/com/speech/up/common/exception/handler/ExceptionController.java b/src/main/java/com/speech/up/common/exception/handler/ExceptionController.java
@@ -1,5 +1,7 @@
 package com.speech.up.common.exception.handler;
 
+import java.io.IOException;
+
 import org.springframework.http.HttpStatus;
 import org.springframework.http.HttpStatusCode;
 import org.springframework.http.ResponseEntity;
@@ -15,6 +17,7 @@
 
 @ControllerAdvice
 public class ExceptionController {
+
 	@ExceptionHandler(MethodArgumentNotValidException.class)
 	public ResponseEntity<ApiExceptionResponse<Void>> handleMethodArgumentNotValidException(
 		MethodArgumentNotValidException exception) {
@@ -67,6 +70,7 @@ public ResponseEntity<ApiExceptionResponse<Void>> handleCustomException(CustomIl
 			HttpStatusCode.valueOf(exception.getErrorCode().getCode())
 		);
 	}
+
 	@ExceptionHandler(CustomIOException.class)
 	public ResponseEntity<ApiExceptionResponse<Void>> handleCustomException(CustomIOException exception) {
 		ApiExceptionResponse<Void> responseDto = ApiExceptionResponse.<Void>builder()
@@ -80,4 +84,14 @@ public ResponseEntity<ApiExceptionResponse<Void>> handleCustomException(CustomIO
 		);
 	}
 
+	@ExceptionHandler(NullPointerException.class)
+	public ResponseEntity<ApiExceptionResponse<Void>> handleNullPointerException(NullPointerException exception) {
+		return ResponseEntity.status(HttpStatus.BAD_REQUEST).build();
+	}
+
+	@ExceptionHandler(IOException.class)
+	public ResponseEntity<ApiExceptionResponse<Void>> handleIOException(IOException exception) {
+		return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).build();
+	}
+
 }