fix recent test failures (#352)

Author: devhawk

transact/src/test/java/dev/dbos/transact/client/PgSqlClientTest.java

@@ -114,7 +114,7 @@ public void clientEnqueueTimeout() throws Exception {
 
   @Test
   public void clientEnqueueDeadline() throws Exception {
-    var wfid1 = enqueueWorkflow("sleep", null, null, Instant.now().plusMillis(1000), 10000);
+    var wfid1 = enqueueWorkflow("sleep", null, null, Instant.now().plusMillis(5000), 60000);
     var handle1 = dbos.retrieveWorkflow(wfid1);
     assertThrows(
         DBOSAwaitedWorkflowCancelledException.class,

transact/src/test/java/dev/dbos/transact/execution/DBOSExecutorTest.java

@@ -50,6 +50,16 @@ private ExecutingService register(DBOS dbos) {
     return service;
   }
 
+  private static void awaitStepCount(DBOS dbos, String wfid, int expected, int timeoutMs)
+      throws Exception {
+    long deadline = System.currentTimeMillis() + timeoutMs;
+    while (System.currentTimeMillis() < deadline) {
+      if (dbos.listWorkflowSteps(wfid).size() == expected) return;
+      Thread.sleep(50);
+    }
+    assertEquals(expected, dbos.listWorkflowSteps(wfid).size());
+  }
+
   @Test
   @EnabledForJreRange(min = JRE.JAVA_21)
   public void virtualThreadPoolJava21() throws Exception {
@@ -201,8 +211,7 @@ public void executeWithStep() throws Exception {
 
       DBUtils.setWorkflowState(dataSource, wfid, WorkflowState.PENDING.name());
       DBUtils.deleteAllStepOutputs(dataSource, wfid);
-      steps = dbos.listWorkflowSteps(wfid);
-      assertEquals(0, steps.size());
+      awaitStepCount(dbos, wfid, 0, 2000);
 
       WorkflowHandle<String, ?> handle = dbosExecutor.executeWorkflowById(wfid, true, false);
 
@@ -212,8 +221,7 @@ public void executeWithStep() throws Exception {
 
       wfs = dbos.listWorkflows(null);
       assertEquals(WorkflowState.SUCCESS, wfs.get(0).status());
-      steps = dbos.listWorkflowSteps(wfid);
-      assertEquals(2, steps.size());
+      awaitStepCount(dbos, wfid, 2, 2000);
     }
   }
 
@@ -246,8 +254,7 @@ public void ReExecuteWithStepTwoOnly() throws Exception {
 
       DBUtils.setWorkflowState(dataSource, wfid, WorkflowState.PENDING.name());
       DBUtils.deleteStepOutput(dataSource, wfid, 1);
-      steps = dbos.listWorkflowSteps(wfid);
-      assertEquals(1, steps.size());
+      awaitStepCount(dbos, wfid, 1, 2000);
 
       WorkflowHandle<String, ?> handle = dbosExecutor.executeWorkflowById(wfid, true, false);
 
@@ -260,8 +267,7 @@ public void ReExecuteWithStepTwoOnly() throws Exception {
 
       wfs = dbos.listWorkflows(null);
       assertEquals(WorkflowState.SUCCESS, wfs.get(0).status());
-      steps = dbos.listWorkflowSteps(wfid);
-      assertEquals(2, steps.size());
+      awaitStepCount(dbos, wfid, 2, 2000);
     }
   }
 

transact/src/test/java/dev/dbos/transact/scheduled/WorkflowScheduleTest.java

@@ -498,7 +498,7 @@ public void scheduleRunsAfterPolling() throws Exception {
     // Wait for all 3 counts to be released, which indicates the workflow ran at least 3 times
     // (scheduler should run it every second).
     assertTrue(
-        impl.latch.await(5, TimeUnit.SECONDS),
-        "Expected latch to count down to zero within 5 seconds");
+        impl.latch.await(15, TimeUnit.SECONDS),
+        "Expected latch to count down to zero within 15 seconds");
   }
 }