test update

Author: YunchuWang

test/e2e-azuremanaged/orchestration.spec.ts

@@ -586,8 +586,9 @@ describe("Durable Task Scheduler (DTS) E2E Tests", () => {
       ctx.setCustomStatus("Step 1: Starting");
       yield ctx.createTimer(1);
 
+      // Use longer timer to give CI environments more time to observe this status
       ctx.setCustomStatus({ step: 2, message: "Processing" });
-      yield ctx.createTimer(1);
+      yield ctx.createTimer(5);
 
       ctx.setCustomStatus("Step 3: Completed");
       return "done";
@@ -598,28 +599,38 @@ describe("Durable Task Scheduler (DTS) E2E Tests", () => {
 
     const id = await taskHubClient.scheduleNewOrchestration(orchestrator);
 
-    // Poll for custom status changes
+    // Poll for custom status changes - collect all observed statuses
     let foundObjectStatus = false;
+    const observedStatuses: string[] = [];
     const startTime = Date.now();
-    while (Date.now() - startTime < 15000) {
+    while (Date.now() - startTime < 20000) {
       const state = await taskHubClient.getOrchestrationState(id);
       if (state?.serializedCustomStatus) {
         const status = state.serializedCustomStatus;
+        // Track unique statuses for debugging
+        if (!observedStatuses.includes(status)) {
+          observedStatuses.push(status);
+        }
         if (status.includes("step") && status.includes("Processing")) {
           foundObjectStatus = true;
           break;
         }
       }
-      await new Promise((resolve) => setTimeout(resolve, 500));
+      // Poll more frequently
+      await new Promise((resolve) => setTimeout(resolve, 200));
     }
 
     const finalState = await taskHubClient.waitForOrchestrationCompletion(id, undefined, 30);
 
     expect(finalState).toBeDefined();
     expect(finalState?.runtimeStatus).toEqual(OrchestrationStatus.ORCHESTRATION_STATUS_COMPLETED);
     expect(finalState?.serializedCustomStatus).toEqual(JSON.stringify("Step 3: Completed"));
+    // Provide better error message on failure - log observed statuses if assertion fails
+    if (!foundObjectStatus) {
+      console.log("Observed statuses during polling:", observedStatuses);
+    }
     expect(foundObjectStatus).toBe(true);
-  }, 31000);
+  }, 60000);
 
   it("should update custom status to empty string when explicitly set", async () => {
     const orchestrator: TOrchestrator = async function* (ctx: OrchestrationContext): any {