fixing up bug in update endpoint

src/main/java/com/ascendfinalproject/warehouse/controllers/WarehouseController.java

@@ -42,8 +42,11 @@ public class WarehouseController {
 
   @CrossOrigin
   @PutMapping(value = "/orders/{id}")
-  public Mono<WarehouseOrderResponse> updateOrder(@RequestBody WarehouseOrderResponse order, @PathVariable(value = "id") String id) {
-    return orderService.updateOrder(order, id);
+  public Mono<ResponseEntity> updateOrder(@RequestBody WarehouseOrderResponse order, @PathVariable(value = "id") String id) {
+    return orderService.updateOrder(order, id)
+            .map(updatedOrder -> (ResponseEntity.status(HttpStatus.OK).body(order)))
+            .cast(ResponseEntity.class)
+            .defaultIfEmpty(ResponseEntity.status(HttpStatus.NOT_FOUND).body(null));
   }
 
   @CrossOrigin

src/main/java/com/ascendfinalproject/warehouse/services/WarehouseOrderService.java

@@ -45,9 +45,6 @@ public class WarehouseOrderService {
 
   public Mono<WarehouseOrderResponse> updateOrder(WarehouseOrderResponse order, String id) {
     return orderRepository.findById(id)
-        .doOnError(error -> {
-            throw new NotFoundException("Warehouse order of ID: " + id + " not found");
-        })
         .flatMap(existingOrder -> {
           existingOrder.setStatus(order.getStatus());
           existingOrder.setModifiedAt(new Date(System.currentTimeMillis()));