Fix flaky test in examples

Author: adwsingh

examples/end-to-end/src/it/java/software/amazon/smithy/java/server/example/RoundTripTests.java

@@ -72,9 +72,8 @@ void executesCorrectly() throws InterruptedException {
         var getResponse1 = client.getOrder(getRequest);
         assertEquals(getResponse1.status(), OrderStatus.IN_PROGRESS);
 
-        // Give order some time to complete
-        System.out.println("Waiting for order to complete....");
-        TimeUnit.SECONDS.sleep(5);
+        // Complete the order
+        OrderTracker.completeOrder(getResponse1.id());
 
         var getResponse2 = client.getOrder(getRequest);
         assertEquals(getResponse2.status(), OrderStatus.COMPLETED);

examples/end-to-end/src/main/java/software/amazon/smithy/java/server/example/Order.java

@@ -16,4 +16,9 @@
  * @param type Type of drink for the order
  * @param status status of the order.
  */
-public record Order(UUID id, CoffeeType type, OrderStatus status) {}
+public record Order(UUID id, CoffeeType type, OrderStatus status) {
+
+    Order complete() {
+        return new Order(id, type, OrderStatus.COMPLETED);
+    }
+}

examples/end-to-end/src/main/java/software/amazon/smithy/java/server/example/OrderTracker.java

@@ -16,17 +16,16 @@
  * This class is a stand-in for a database.
  */
 final class OrderTracker {
-    private static final ScheduledThreadPoolExecutor executor = new ScheduledThreadPoolExecutor(1);
     private static final Map<UUID, Order> ORDERS = new ConcurrentHashMap<>();
 
+    private OrderTracker() {}
+
     public static void putOrder(Order order) {
         ORDERS.put(order.id(), order);
+    }
 
-        // Start a process to complete the order in the background.
-        executor.schedule(
-                () -> ORDERS.put(order.id(), new Order(order.id(), order.type(), OrderStatus.COMPLETED)),
-                5,
-                TimeUnit.SECONDS);
+    public static void completeOrder(String id) {
+        ORDERS.computeIfPresent(UUID.fromString(id), (k,v) -> v.complete());
     }
 
     public static Order getOrderById(UUID id) {