Update packages/durabletask-js/test/orchestration_executor.spec.ts

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: YunchuWang

packages/durabletask-js/test/orchestration_executor.spec.ts

@@ -1549,43 +1549,30 @@ describe("Orchestration Executor", () => {
     expect(completeAction?.getResult()?.getValue()).toEqual(JSON.stringify(expectedResult));
   });
 
-  it("should fail when orchestrator yields null as its first value", async () => {
-    // An orchestrator that yields null instead of a Task should fail with a clear error
-    const badOrchestrator: TOrchestrator = async function* (_ctx: OrchestrationContext): any {
-      yield null;
-    };
-
-    const registry = new Registry();
-    const name = registry.addOrchestrator(badOrchestrator);
-    const newEvents = [
-      newOrchestratorStartedEvent(),
-      newExecutionStartedEvent(name, TEST_INSTANCE_ID, undefined),
-    ];
-    const executor = new OrchestrationExecutor(registry, testLogger);
-    const result = await executor.execute(TEST_INSTANCE_ID, [], newEvents);
-    const completeAction = getAndValidateSingleCompleteOrchestrationAction(result);
-    expect(completeAction?.getOrchestrationstatus()).toEqual(pb.OrchestrationStatus.ORCHESTRATION_STATUS_FAILED);
-    expect(completeAction?.getFailuredetails()?.getErrormessage()).toContain("non-Task");
-  });
-
-  it("should fail when orchestrator yields undefined as its first value", async () => {
-    // An orchestrator that yields undefined instead of a Task should fail with a clear error
-    const badOrchestrator: TOrchestrator = async function* (_ctx: OrchestrationContext): any {
-      yield undefined;
-    };
+  it.each([
+    { description: "null", yieldedValue: null as any },
+    { description: "undefined", yieldedValue: undefined as any },
+  ])(
+    "should fail when orchestrator yields $description as its first value",
+    async ({ description, yieldedValue }) => {
+      // An orchestrator that yields a non-Task value as its first yield should fail with a clear error
+      const badOrchestrator: TOrchestrator = async function* (_ctx: OrchestrationContext): any {
+        yield yieldedValue;
+      };
 
-    const registry = new Registry();
-    const name = registry.addOrchestrator(badOrchestrator);
-    const newEvents = [
-      newOrchestratorStartedEvent(),
-      newExecutionStartedEvent(name, TEST_INSTANCE_ID, undefined),
-    ];
-    const executor = new OrchestrationExecutor(registry, testLogger);
-    const result = await executor.execute(TEST_INSTANCE_ID, [], newEvents);
-    const completeAction = getAndValidateSingleCompleteOrchestrationAction(result);
-    expect(completeAction?.getOrchestrationstatus()).toEqual(pb.OrchestrationStatus.ORCHESTRATION_STATUS_FAILED);
-    expect(completeAction?.getFailuredetails()?.getErrormessage()).toContain("non-Task");
-  });
+      const registry = new Registry();
+      const name = registry.addOrchestrator(badOrchestrator);
+      const newEvents = [
+        newOrchestratorStartedEvent(),
+        newExecutionStartedEvent(name, TEST_INSTANCE_ID, undefined),
+      ];
+      const executor = new OrchestrationExecutor(registry, testLogger);
+      const result = await executor.execute(TEST_INSTANCE_ID, [], newEvents);
+      const completeAction = getAndValidateSingleCompleteOrchestrationAction(result);
+      expect(completeAction?.getOrchestrationstatus()).toEqual(pb.OrchestrationStatus.ORCHESTRATION_STATUS_FAILED);
+      expect(completeAction?.getFailuredetails()?.getErrormessage()).toContain("non-Task");
+    }
+  );
 
   it("should fail when orchestrator yields a plain object as its first value", async () => {
     // An orchestrator that yields a non-Task object should fail with a clear error