fix e2e tests

Author: YunchuWang

jest.config.js

@@ -0,0 +1,15 @@
+/** @type {import('ts-jest').JestConfigWithTsJest} */
+module.exports = {
+  preset: "ts-jest",
+  testEnvironment: "node",
+  testMatch: ["**/tests/**/*.spec.ts"],
+  moduleFileExtensions: ["ts", "tsx", "js", "jsx", "json", "node"],
+  transform: {
+    "^.+\\.tsx?$": [
+      "ts-jest",
+      {
+        tsconfig: "tsconfig.base.json",
+      },
+    ],
+  },
+};

package.json

@@ -13,7 +13,8 @@
     "test": "npm run test --workspaces",
     "test:unit": "npm run test:unit --workspaces --if-present",
     "test:e2e": "./scripts/test-e2e.sh",
-    "test:e2e:one": "jest test/e2e --runInBand --detectOpenHandles --testNamePattern",
+    "test:e2e:internal": "jest tests/e2e --runInBand --detectOpenHandles",
+    "test:e2e:one": "jest tests/e2e --runInBand --detectOpenHandles --testNamePattern",
     "lint": "eslint . --ext .js,.jsx,.ts,.tsx",
     "pretty": "prettier --list-different \"**/*.{ts,tsx,js,jsx,json,md}\"",
     "pretty-fix": "prettier --write \"**/*.{ts,tsx,js,jsx,json,md}\"",

packages/durabletask-js/package.json

@@ -15,7 +15,7 @@
     "prebuild": "node -p \"'// Auto-generated by prebuild\\nexport const SDK_VERSION = ' + JSON.stringify(require('./package.json').version) + ';\\nexport const SDK_PACKAGE_NAME = ' + JSON.stringify(require('./package.json').name) + ';'\" > src/version.ts",
     "build": "npm run prebuild && npm run clean && tsc -p tsconfig.build.json && npm run copy-proto",
     "test": "jest --runInBand --detectOpenHandles",
-    "test:unit": "jest test/unit --runInBand --detectOpenHandles"
+    "test:unit": "jest test --runInBand --detectOpenHandles"
   },
   "engines": {
     "node": ">=22.0.0"

packages/durabletask-js/src/index.ts

@@ -1,9 +1,27 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
-import { TaskHubGrpcClient } from "./client/client";
-import { TaskHubGrpcWorker } from "./worker/task-hub-grpc-worker";
-import { OrchestrationContext } from "./task/context/orchestration-context";
-import { ActivityContext } from "./task/context/activity-context";
+// Client and Worker
+export { TaskHubGrpcClient } from "./client/client";
+export { TaskHubGrpcWorker } from "./worker/task-hub-grpc-worker";
 
-export { TaskHubGrpcClient, TaskHubGrpcWorker, OrchestrationContext, ActivityContext };
+// Contexts
+export { OrchestrationContext } from "./task/context/orchestration-context";
+export { ActivityContext } from "./task/context/activity-context";
+
+// Orchestration types and utilities
+export { PurgeInstanceCriteria } from "./orchestration/orchestration-purge-criteria";
+export { OrchestrationStatus } from "./orchestration/enum/orchestration-status.enum";
+
+// Proto types (for advanced usage)
+export { OrchestrationStatus as ProtoOrchestrationStatus } from "./proto/orchestrator_service_pb";
+
+// Task utilities
+export { getName, whenAll, whenAny } from "./task";
+export { Task } from "./task/task";
+
+// Types
+export { TOrchestrator } from "./types/orchestrator.type";
+export { TActivity } from "./types/activity.type";
+export { TInput } from "./types/input.type";
+export { TOutput } from "./types/output.type";

tests/e2e/orchestration.spec.ts

@@ -1,16 +1,20 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
-import { TaskHubGrpcClient } from "../../src/client/client";
-import { PurgeInstanceCriteria } from "../../src/orchestration/orchestration-purge-criteria";
-import { OrchestrationStatus } from "../../src/proto/orchestrator_service_pb";
-import { OrchestrationStatus as RuntimeStatus } from "../../src/orchestration/enum/orchestration-status.enum";
-import { getName, whenAll, whenAny } from "../../src/task";
-import { ActivityContext } from "../../src/task/context/activity-context";
-import { OrchestrationContext } from "../../src/task/context/orchestration-context";
-import { Task } from "../../src/task/task";
-import { TOrchestrator } from "../../src/types/orchestrator.type";
-import { TaskHubGrpcWorker } from "../../src/worker/task-hub-grpc-worker";
+import {
+  TaskHubGrpcClient,
+  TaskHubGrpcWorker,
+  PurgeInstanceCriteria,
+  ProtoOrchestrationStatus as OrchestrationStatus,
+  OrchestrationStatus as RuntimeStatus,
+  getName,
+  whenAll,
+  whenAny,
+  ActivityContext,
+  OrchestrationContext,
+  Task,
+  TOrchestrator,
+} from "@microsoft/durabletask-js";
 
 describe("Durable Functions", () => {
   let taskHubClient: TaskHubGrpcClient;
@@ -421,7 +425,8 @@ describe("Durable Functions", () => {
     taskHubWorker.addActivity(plusOne);
     await taskHubWorker.start();
 
-    const startTime = new Date(Date.now());
+    // Set startTime slightly in the past to account for clock drift
+    const startTime = new Date(Date.now() - 1000);
     const id = await taskHubClient.scheduleNewOrchestration(orchestrator, 1);
     const state = await taskHubClient.waitForOrchestrationCompletion(id, undefined, 30);
 
@@ -478,6 +483,9 @@ describe("Durable Functions", () => {
     runtimeStatuses.push(RuntimeStatus.TERMINATED);
     runtimeStatuses.push(RuntimeStatus.COMPLETED);
 
+    // Add a small delay to ensure the orchestrations are fully persisted
+    await new Promise((resolve) => setTimeout(resolve, 1000));
+
     criteria.setCreatedTimeTo(new Date(Date.now()));
     criteria.setRuntimeStatusList(runtimeStatuses);
     purgeResult = await taskHubClient.purgeOrchestration(criteria);