cover methode getMovieById in controller

neuefische · Aug 19, 2024 · 08d06cf · 08d06cf
1 parent 46c0777
commit 08d06cf
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/backend/src/main/java/de/webdev/backend/controller/MovieController.java b/backend/src/main/java/de/webdev/backend/controller/MovieController.java
@@ -6,10 +6,12 @@
 import lombok.RequiredArgsConstructor;
 
 
+import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
 
 
 import java.util.List;
+import java.util.Optional;
 
 @RestController
 @RequestMapping("/api/movies")
@@ -25,12 +27,15 @@ public List<Movie> getMovies() {
 
 
     @GetMapping("{id}")
-    public Movie getMovieById(@PathVariable String id) {
-        return movieService.getMovieById(id);
+    public ResponseEntity<Movie> getMovieById(@PathVariable String id) {
+        Optional<Movie> movieOptional = Optional.ofNullable(movieService.getMovieById(id));
+        return movieOptional.map(ResponseEntity::ok)
+                .orElseGet(() -> ResponseEntity.notFound().build());
 
     }
+
     @PostMapping()
-    public Movie postMovie (@RequestBody MovieDto userEntries) {
+    public Movie postMovie(@RequestBody MovieDto userEntries) {
         return movieService.addMovie(userEntries);
 
     }