Cover IllegalAttachmentFileNameException in GenieExceptionMapperTest (N…

…etflix#1213) Co-authored-by: bhou <bhou@netflix.com>
natedanner · May 20, 2024 · 3acceb7 · 3acceb7
1 parent 72b906b
commit 3acceb7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...rc/test/java/com/netflix/genie/web/apis/rest/v3/controllers/GenieExceptionMapperTest.java b/...rc/test/java/com/netflix/genie/web/apis/rest/v3/controllers/GenieExceptionMapperTest.java
@@ -40,6 +40,7 @@
 import com.netflix.genie.common.internal.exceptions.unchecked.GenieRuntimeException;
 import com.netflix.genie.web.exceptions.checked.AttachmentTooLargeException;
 import com.netflix.genie.web.exceptions.checked.IdAlreadyExistsException;
+import com.netflix.genie.web.exceptions.checked.IllegalAttachmentFileNameException;
 import com.netflix.genie.web.exceptions.checked.JobNotFoundException;
 import com.netflix.genie.web.exceptions.checked.NotFoundException;
 import com.netflix.genie.web.exceptions.checked.PreconditionFailedException;
@@ -232,6 +233,7 @@ void canHandleGenieCheckedExceptions() {
         exceptions.put(new JobNotFoundException(), HttpStatus.NOT_FOUND);
         exceptions.put(new NotFoundException(), HttpStatus.NOT_FOUND);
         exceptions.put(new PreconditionFailedException(), HttpStatus.BAD_REQUEST);
+        exceptions.put(new IllegalAttachmentFileNameException(), HttpStatus.BAD_REQUEST);
         exceptions.put(new AttachmentTooLargeException(), HttpStatus.PAYLOAD_TOO_LARGE);
 
         for (final Map.Entry<GenieCheckedException, HttpStatus> exception : exceptions.entrySet()) {