Update previous complete test endpoints

smcvb · smcvb · commit a5ad7b16e560 · 2021-03-19T11:54:20.000+01:00
The test endpoints used to create an order, add a product, confirm the
order and ship it are not embracing the asynchronous approach of the
CommandGateway. Make sure that all commands are composed with one
another and return the completable future for Spring to resolve

#BAEL-4767
diff --git a/axon/src/main/java/com/baeldung/axon/gui/OrderRestEndpoint.java b/axon/src/main/java/com/baeldung/axon/gui/OrderRestEndpoint.java
@@ -32,21 +32,21 @@ public OrderRestEndpoint(CommandGateway commandGateway, QueryGateway queryGatewa
     }
 
     @PostMapping("/ship-order")
-    public void shipOrder() {
+    public CompletableFuture<Void> shipOrder() {
         String orderId = UUID.randomUUID().toString();
-        commandGateway.send(new CreateOrderCommand(orderId));
-        commandGateway.send(new AddProductCommand(orderId, "Deluxe Chair"));
-        commandGateway.send(new ConfirmOrderCommand(orderId));
-        commandGateway.send(new ShipOrderCommand(orderId));
+        return commandGateway.send(new CreateOrderCommand(orderId))
+                             .thenCompose(result -> commandGateway.send(new AddProductCommand(orderId, "Deluxe Chair")))
+                             .thenCompose(result -> commandGateway.send(new ConfirmOrderCommand(orderId)))
+                             .thenCompose(result -> commandGateway.send(new ShipOrderCommand(orderId)));
     }
 
     @PostMapping("/ship-unconfirmed-order")
-    public void shipUnconfirmedOrder() {
+    public CompletableFuture<Void> shipUnconfirmedOrder() {
         String orderId = UUID.randomUUID().toString();
-        commandGateway.send(new CreateOrderCommand(orderId));
-        commandGateway.send(new AddProductCommand(orderId, "Deluxe Chair"));
-        // This throws an exception, as an Order cannot be shipped if it has not been confirmed yet.
-        commandGateway.send(new ShipOrderCommand(orderId));
+        return commandGateway.send(new CreateOrderCommand(orderId))
+                             .thenCompose(result -> commandGateway.send(new AddProductCommand(orderId, "Deluxe Chair")))
+                             // This throws an exception, as an Order cannot be shipped if it has not been confirmed yet.
+                             .thenCompose(result -> commandGateway.send(new ShipOrderCommand(orderId)));
     }
 
     @PostMapping("/order")
