Fix a 404 test case

spring-guides · Oct 10, 2023 · 879c9e5 · 879c9e5
1 parent c128ccf
commit 879c9e5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/complete/src/main/java/com/example/uploadingfiles/FileUploadController.java b/complete/src/main/java/com/example/uploadingfiles/FileUploadController.java
@@ -48,6 +48,10 @@ public String listUploadedFiles(Model model) throws IOException {
 	public ResponseEntity<Resource> serveFile(@PathVariable String filename) {
 
 		Resource file = storageService.loadAsResource(filename);
+
+		if (file == null)
+			return ResponseEntity.notFound().build();
+
 		return ResponseEntity.ok().header(HttpHeaders.CONTENT_DISPOSITION,
 				"attachment; filename=\"" + file.getFilename() + "\"").body(file);
 	}