@temporalio/testing 0.23.3 → 1.0.0

package/LICENSE.md

@@ -2,7 +2,7 @@ Temporal TypeScript SDK
 
 MIT License
 
-Copyright (c) 2021 Temporal Technologies, Inc. All Rights Reserved
+Copyright (c) 2021 Temporal Technologies Inc. All Rights Reserved
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

package/README.md

@@ -0,0 +1,9 @@
+# `@temporalio/testing`
+
+[![NPM](https://img.shields.io/npm/v/@temporalio/testing?style=for-the-badge)](https://www.npmjs.com/package/@temporalio/testing)
+
+Part of [Temporal](https://temporal.io)'s [TypeScript SDK](https://docs.temporal.io/typescript/introduction/).
+
+- [Testing docs](https://docs.temporal.io/typescript/testing)
+- [API reference](https://typescript.temporal.io/api/namespaces/testing)
+- [Sample projects](https://github.com/temporalio/samples-typescript)

package/lib/child-process.d.ts

@@ -1,4 +1,5 @@
 /// <reference types="node" />
+/// <reference types="node" />
 import { ChildProcess } from 'child_process';
 export declare class ChildProcessError extends Error {
     readonly code: number | null;
@@ -11,4 +12,5 @@ export interface WaitOptions {
 }
 export declare function waitOnChild(child: ChildProcess, opts?: WaitOptions): Promise<void>;
 export declare function kill(child: ChildProcess, signal?: NodeJS.Signals, opts?: WaitOptions): Promise<void>;
+export declare function killIfExists(child: ChildProcess, signal?: NodeJS.Signals, opts?: WaitOptions): Promise<void>;
 export declare const shell: boolean;

package/lib/child-process.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.shell = exports.kill = exports.waitOnChild = exports.ChildProcessError = void 0;
+exports.shell = exports.killIfExists = exports.kill = exports.waitOnChild = exports.ChildProcessError = void 0;
 class ChildProcessError extends Error {
     constructor(message, code, signal) {
         super(message);
@@ -42,5 +42,16 @@ async function kill(child, signal = 'SIGINT', opts) {
     }
 }
 exports.kill = kill;
+async function killIfExists(child, signal = 'SIGINT', opts) {
+    try {
+        await kill(child, signal, opts);
+    }
+    catch (err) {
+        if (err.code !== 'ESRCH') {
+            throw err;
+        }
+    }
+}
+exports.killIfExists = killIfExists;
 exports.shell = process.platform === 'win32';
 //# sourceMappingURL=child-process.js.map

package/lib/child-process.js.map

@@ -1 +1 @@
-{"version":3,"file":"child-process.js","sourceRoot":"","sources":["../src/child-process.ts"],"names":[],"mappings":";;;AAGA,MAAa,iBAAkB,SAAQ,KAAK;IAG1C,YAAY,OAAe,EAAkB,IAAmB,EAAkB,MAA6B;QAC7G,KAAK,CAAC,OAAO,CAAC,CAAC;QAD4B,SAAI,GAAJ,IAAI,CAAe;QAAkB,WAAM,GAAN,MAAM,CAAuB;QAF/F,SAAI,GAAG,mBAAmB,CAAC;IAI3C,CAAC;CACF;AAND,8CAMC;AAMM,KAAK,UAAU,WAAW,CAAC,KAAmB,EAAE,IAAkB;IACvE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YAChC,IAAI,IAAI,KAAK,IAAI,IAAI,CAAC,IAAI,EAAE,gBAAgB,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;gBACnE,OAAO,EAAE,CAAC;aACX;iBAAM;gBACL,MAAM,CAAC,IAAI,iBAAiB,CAAC,gBAAgB,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;aAC/D;QACH,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC;AAXD,kCAWC;AAEM,KAAK,UAAU,IAAI,CAAC,KAAmB,EAAE,SAAyB,QAAQ,EAAE,IAAkB;IACnG,IAAI,KAAK,CAAC,GAAG,KAAK,SAAS,EAAE;QAC3B,MAAM,IAAI,SAAS,CAAC,yBAAyB,CAAC,CAAC;KAChD;IACD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAChC,IAAI;QACF,MAAM,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;KAChC;IAAC,OAAO,GAAG,EAAE;QACZ,0EAA0E;QAC1E,oEAAoE;QACpE,MAAM,WAAW,GAAG,CAAC,CAAC,GAAG,YAAY,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,KAAK,OAAO,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;QACnH,IAAI,WAAW,EAAE;YACf,MAAM,GAAG,CAAC;SACX;KACF;AACH,CAAC;AAfD,oBAeC;AAEY,QAAA,KAAK,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC"}
+{"version":3,"file":"child-process.js","sourceRoot":"","sources":["../src/child-process.ts"],"names":[],"mappings":";;;AAGA,MAAa,iBAAkB,SAAQ,KAAK;IAG1C,YAAY,OAAe,EAAkB,IAAmB,EAAkB,MAA6B;QAC7G,KAAK,CAAC,OAAO,CAAC,CAAC;QAD4B,SAAI,GAAJ,IAAI,CAAe;QAAkB,WAAM,GAAN,MAAM,CAAuB;QAF/F,SAAI,GAAG,mBAAmB,CAAC;IAI3C,CAAC;CACF;AAND,8CAMC;AAMM,KAAK,UAAU,WAAW,CAAC,KAAmB,EAAE,IAAkB;IACvE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YAChC,IAAI,IAAI,KAAK,IAAI,IAAI,CAAC,IAAI,EAAE,gBAAgB,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;gBACnE,OAAO,EAAE,CAAC;aACX;iBAAM;gBACL,MAAM,CAAC,IAAI,iBAAiB,CAAC,gBAAgB,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;aAC/D;QACH,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC;AAXD,kCAWC;AAEM,KAAK,UAAU,IAAI,CAAC,KAAmB,EAAE,SAAyB,QAAQ,EAAE,IAAkB;IACnG,IAAI,KAAK,CAAC,GAAG,KAAK,SAAS,EAAE;QAC3B,MAAM,IAAI,SAAS,CAAC,yBAAyB,CAAC,CAAC;KAChD;IACD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAChC,IAAI;QACF,MAAM,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;KAChC;IAAC,OAAO,GAAG,EAAE;QACZ,0EAA0E;QAC1E,oEAAoE;QACpE,MAAM,WAAW,GAAG,CAAC,CAAC,GAAG,YAAY,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,KAAK,OAAO,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;QACnH,IAAI,WAAW,EAAE;YACf,MAAM,GAAG,CAAC;SACX;KACF;AACH,CAAC;AAfD,oBAeC;AAEM,KAAK,UAAU,YAAY,CAChC,KAAmB,EACnB,SAAyB,QAAQ,EACjC,IAAkB;IAElB,IAAI;QACF,MAAM,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;KACjC;IAAC,OAAO,GAAQ,EAAE;QACjB,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO,EAAE;YACxB,MAAM,GAAG,CAAC;SACX;KACF;AACH,CAAC;AAZD,oCAYC;AAEY,QAAA,KAAK,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC"}

package/lib/index-for-docs.d.ts

@@ -0,0 +1 @@
+export * as testing from './index';

package/lib/index-for-docs.js

@@ -0,0 +1,29 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.testing = void 0;
+// Reexports the contents of this package under the "testing" namespace for better docs generation
+exports.testing = __importStar(require("./index"));
+//# sourceMappingURL=index-for-docs.js.map

package/lib/index-for-docs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index-for-docs.js","sourceRoot":"","sources":["../src/index-for-docs.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,kGAAkG;AAClG,mDAAmC"}

package/lib/index.d.ts

@@ -1,12 +1,21 @@
+/**
+ * `npm i @temporalio/testing`
+ *
+ * Testing library for the SDK.
+ *
+ * [Documentation](https://docs.temporal.io/typescript/testing)
+ *
+ * @module
+ */
+/// <reference types="node" />
 /// <reference types="node" />
 import * as activity from '@temporalio/activity';
-import { AsyncCompletionClient, WorkflowClient as BaseWorkflowClient, WorkflowClientOptions, WorkflowResultOptions as BaseWorkflowResultOptions, WorkflowStartOptions as BaseWorkflowStartOptions } from '@temporalio/client';
+import { AsyncCompletionClient, WorkflowClient as BaseWorkflowClient, WorkflowClientOptions as BaseWorkflowClientOptions, WorkflowResultOptions as BaseWorkflowResultOptions, WorkflowStartOptions as BaseWorkflowStartOptions } from '@temporalio/client';
 import { ActivityFunction, Workflow, WorkflowResultType } from '@temporalio/common';
 import { NativeConnection, Logger } from '@temporalio/worker';
 import { ChildProcess, StdioOptions } from 'child_process';
 import events from 'events';
 import { Connection, TestService } from './test-service-client';
-import { temporal } from '@temporalio/proto';
 export declare const DEFAULT_TEST_SERVER_PATH: string;
 /**
  * Options passed to {@link WorkflowClient.result}, these are the same as the
@@ -32,6 +41,9 @@ export declare type WorkflowStartOptions<T extends Workflow> = BaseWorkflowStart
      */
     runInNormalTime?: boolean;
 };
+export interface WorkflowClientOptions extends BaseWorkflowClientOptions {
+    connection: Connection;
+}
 /**
  * A client with the exact same API as the "normal" client with 1 exception,
  * When this client waits on a Workflow's result, it will enable time skipping
@@ -39,7 +51,7 @@ export declare type WorkflowStartOptions<T extends Workflow> = BaseWorkflowStart
  */
 export declare class WorkflowClient extends BaseWorkflowClient {
     protected readonly testService: TestService;
-    constructor(testService: TestService, workflowService?: temporal.api.workflowservice.v1.WorkflowService, options?: WorkflowClientOptions | undefined);
+    constructor(options: WorkflowClientOptions);
     /**
      * Execute a Workflow and wait for completion.
      *

package/lib/index.js

@@ -1,4 +1,13 @@
 "use strict";
+/**
+ * `npm i @temporalio/testing`
+ *
+ * Testing library for the SDK.
+ *
+ * [Documentation](https://docs.temporal.io/typescript/testing)
+ *
+ * @module
+ */
 var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
     if (k2 === undefined) k2 = k;
     var desc = Object.getOwnPropertyDescriptor(m, k);
@@ -46,9 +55,9 @@ exports.DEFAULT_TEST_SERVER_PATH = path_1.default.join(__dirname, `../${TEST_SER
  * in the test server.
  */
 class WorkflowClient extends client_1.WorkflowClient {
-    constructor(testService, workflowService, options) {
-        super(workflowService, options);
-        this.testService = testService;
+    constructor(options) {
+        super(options);
+        this.testService = options.connection.testService;
     }
     /**
      * Execute a Workflow and wait for completion.
@@ -151,8 +160,8 @@ class TestWorkflowEnvironment {
         };
         this.connection = connection;
         this.nativeConnection = nativeConnection;
-        this.workflowClient = new WorkflowClient(this.connection.testService, this.connection.service);
-        this.asyncCompletionClient = new client_1.AsyncCompletionClient(this.connection.service);
+        this.workflowClient = new WorkflowClient({ connection });
+        this.asyncCompletionClient = new client_1.AsyncCompletionClient({ connection });
     }
     /**
      * Create a new test environment
@@ -164,10 +173,10 @@ class TestWorkflowEnvironment {
         const { testServerSpawner, logger } = addDefaults(opts ?? {});
         const child = testServerSpawner(port);
         const address = `127.0.0.1:${port}`;
-        const conn = new test_service_client_1.Connection({ address });
+        const connPromise = test_service_client_1.Connection.connect({ address });
         try {
             await Promise.race([
-                conn.untilReady(),
+                connPromise,
                 (0, child_process_2.waitOnChild)(child).then(() => {
                     throw new Error('Test server child process exited prematurely');
                 }),
@@ -182,14 +191,16 @@ class TestWorkflowEnvironment {
             }
             throw err;
         }
-        const nativeConnection = await worker_1.NativeConnection.create({ address });
+        const conn = await connPromise;
+        const nativeConnection = await worker_1.NativeConnection.connect({ address });
         return new this(child, conn, nativeConnection);
     }
     /**
      * Kill the test server process and close the connection to it
      */
     async teardown() {
-        this.connection.client.close();
+        await this.connection.close();
+        await this.nativeConnection.close();
         // TODO: the server should return exit code 0
         await (0, child_process_2.kill)(this.serverProc, 'SIGINT', { validReturnCodes: [0, 130] });
     }
@@ -200,6 +211,7 @@ exports.TestWorkflowEnvironment = TestWorkflowEnvironment;
  */
 exports.defaultActivityInfo = {
     attempt: 1,
+    taskQueue: 'test',
     isLocal: false,
     taskToken: Buffer.from('test'),
     activityId: 'test',
@@ -233,6 +245,9 @@ class MockActivityEnvironment extends events_1.default.EventEmitter {
         });
         const heartbeatCallback = (details) => this.emit('heartbeat', details);
         this.context = new activity.Context({ ...exports.defaultActivityInfo, ...info }, promise, abortController.signal, heartbeatCallback);
+        promise.catch(() => {
+            /* avoid unhandled rejection */
+        });
     }
     /**
      * Run a function in Activity Context

package/lib/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,+DAAiD;AACjD,+CAM4B;AAC5B,+CAA8G;AAC9G,+CAA6E;AAC7E,gDAAwB;AACxB,4CAAoB;AACpB,uDAAmD;AACnD,iDAAkE;AAClE,oDAA4B;AAC5B,mDAAoD;AAEpD,+DAAgE;AAGhE,MAAM,2BAA2B,GAAG,YAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,aAAa,CAAC;AAErF,QAAA,wBAAwB,GAAG,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,2BAA2B,EAAE,CAAC,CAAC;AA4BlG;;;;GAIG;AACH,MAAa,cAAe,SAAQ,uBAAkB;IACpD,YACqB,WAAwB,EAC3C,eAAiE,EACjE,OAA2C;QAE3C,KAAK,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;QAJb,gBAAW,GAAX,WAAW,CAAa;IAK7C,CAAC;IAED;;;;OAIG;IACI,KAAK,CAAC,OAAO,CAClB,kBAA8B,EAC9B,OAAgC;QAEhC,OAAO,KAAK,CAAC,OAAO,CAAC,kBAAkB,EAAE,OAAO,CAAC,CAAC;IACpD,CAAC;IAED;;;;OAIG;IACM,KAAK,CAAC,MAAM,CACnB,UAAkB,EAClB,KAA0B,EAC1B,IAAwC;QAExC,IAAI,IAAI,EAAE,eAAe,EAAE;YACzB,OAAO,MAAM,KAAK,CAAC,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;SACpD;QACD,MAAM,IAAI,CAAC,WAAW,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;QAC9C,IAAI;YACF,OAAO,MAAM,KAAK,CAAC,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;SACpD;gBAAS;YACR,MAAM,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;SAC7C;IACH,CAAC;CACF;AAzCD,wCAyCC;AAED;;;;;GAKG;AACU,QAAA,0BAA0B,GAAG,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,+BAA+B,CAAC,CAAC,CAAC;AA+BlG,SAAS,WAAW,CAAC,EACnB,UAAU,EACV,MAAM,GACyB;IAC/B,OAAO;QACL,iBAAiB,EACf,OAAO,UAAU,KAAK,UAAU;YAC9B,CAAC,CAAC,UAAU;YACZ,CAAC,CAAC,CAAC,IAAY,EAAE,EAAE,CACf,IAAA,qBAAK,EAAC,UAAU,EAAE,IAAI,IAAI,gCAAwB,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE;gBAC/D,KAAK,EAAE,UAAU,EAAE,KAAK,IAAI,QAAQ;aACrC,CAAC;QACV,MAAM,EAAE,MAAM,IAAI,IAAI,sBAAa,CAAC,MAAM,CAAC;KAC5C,CAAC;AACJ,CAAC;AAED,gFAAgF;AAChF,sCAAsC;AACtC,MAAM,cAAc,GAAG,IAAI,QAAQ,CAAC,YAAY,EAAE,2BAA2B,CAAC,CAAC;AAE/E;;;;;GAKG;AACH,MAAa,uBAAuB;IAuBlC,YACqB,UAAwB,EAC3C,UAAsB,EACtB,gBAAkC;QAFf,eAAU,GAAV,UAAU,CAAc;QAuD7C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WAuCG;QACH,UAAK,GAAG,KAAK,EAAE,UAA2B,EAAiB,EAAE;YAC3D,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,2BAA2B,CAAC,EAAE,QAAQ,EAAE,IAAA,eAAM,EAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAClG,CAAC,CAAC;QA7FA,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC/F,IAAI,CAAC,qBAAqB,GAAG,IAAI,8BAAqB,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAClF,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAqC;QACvD,gEAAgE;QAChE,MAAM,OAAO,GAAG,CAAC,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,OAA6B,CAAC;QACjF,MAAM,IAAI,GAAG,MAAM,OAAO,EAAE,CAAC;QAE7B,MAAM,EAAE,iBAAiB,EAAE,MAAM,EAAE,GAAG,WAAW,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;QAE9D,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAEtC,MAAM,OAAO,GAAG,aAAa,IAAI,EAAE,CAAC;QACpC,MAAM,IAAI,GAAG,IAAI,gCAAU,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QAEzC,IAAI;YACF,MAAM,OAAO,CAAC,IAAI,CAAC;gBACjB,IAAI,CAAC,UAAU,EAAE;gBACjB,IAAA,2BAAW,EAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE;oBAC3B,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;gBAClE,CAAC,CAAC;aACH,CAAC,CAAC;SACJ;QAAC,OAAO,GAAG,EAAE;YACZ,IAAI;gBACF,MAAM,IAAA,oBAAI,EAAC,KAAK,CAAC,CAAC;aACnB;YAAC,OAAO,KAAK,EAAE;gBACd,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;aACrE;YACD,MAAM,GAAG,CAAC;SACX;QAED,MAAM,gBAAgB,GAAG,MAAM,yBAAgB,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QAEpE,OAAO,IAAI,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE,gBAAgB,CAAC,CAAC;IACjD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAC/B,6CAA6C;QAC7C,MAAM,IAAA,oBAAI,EAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;IACxE,CAAC;CA6CF;AA1HD,0DA0HC;AAED;;GAEG;AACU,QAAA,mBAAmB,GAAkB;IAChD,OAAO,EAAE,CAAC;IACV,OAAO,EAAE,KAAK;IACd,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;IAC9B,UAAU,EAAE,MAAM;IAClB,YAAY,EAAE,SAAS;IACvB,YAAY,EAAE,MAAM;IACpB,eAAe,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC;IACvD,gBAAgB,EAAE,SAAS;IAC3B,iBAAiB,EAAE,SAAS;IAC5B,iBAAiB,EAAE,SAAS;IAC5B,iBAAiB,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE;IAC7D,oBAAoB,EAAE,CAAC;IACvB,qBAAqB,EAAE,IAAI;IAC3B,wBAAwB,EAAE,IAAI;CAC/B,CAAC;AAEF;;;;;GAKG;AACH,MAAa,uBAAwB,SAAQ,gBAAM,CAAC,YAAY;IAI9D,YAAY,IAA6B;QACvC,KAAK,EAAE,CAAC;QAJH,WAAM,GAA2B,GAAG,EAAE,CAAC,SAAS,CAAC;QAKtD,MAAM,eAAe,GAAG,IAAI,kCAAe,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;YAC/C,IAAI,CAAC,MAAM,GAAG,CAAC,MAAY,EAAE,EAAE;gBAC7B,eAAe,CAAC,KAAK,EAAE,CAAC;gBACxB,MAAM,CAAC,IAAI,yBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC;YACvC,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,MAAM,iBAAiB,GAAG,CAAC,OAAiB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QACjF,IAAI,CAAC,OAAO,GAAG,IAAI,QAAQ,CAAC,OAAO,CACjC,EAAE,GAAG,2BAAmB,EAAE,GAAG,IAAI,EAAE,EACnC,OAAO,EACP,eAAe,CAAC,MAAM,EACtB,iBAAiB,CAClB,CAAC;IACJ,CAAC;IAED;;OAEG;IACI,GAAG,CAAuD,EAAK,EAAE,GAAG,IAAO;QAChF,OAAO,QAAQ,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IACnE,CAAC;CACF;AA5BD,0DA4BC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,+DAAiD;AACjD,+CAM4B;AAC5B,+CAA8G;AAC9G,+CAA6E;AAC7E,gDAAwB;AACxB,4CAAoB;AACpB,uDAAmD;AACnD,iDAAkE;AAClE,oDAA4B;AAC5B,mDAAoD;AAEpD,+DAAgE;AAEhE,MAAM,2BAA2B,GAAG,YAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,aAAa,CAAC;AAErF,QAAA,wBAAwB,GAAG,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,2BAA2B,EAAE,CAAC,CAAC;AAgClG;;;;GAIG;AACH,MAAa,cAAe,SAAQ,uBAAkB;IAGpD,YAAY,OAA8B;QACxC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC;IACpD,CAAC;IAED;;;;OAIG;IACI,KAAK,CAAC,OAAO,CAClB,kBAA8B,EAC9B,OAAgC;QAEhC,OAAO,KAAK,CAAC,OAAO,CAAC,kBAAkB,EAAE,OAAO,CAAC,CAAC;IACpD,CAAC;IAED;;;;OAIG;IACM,KAAK,CAAC,MAAM,CACnB,UAAkB,EAClB,KAA0B,EAC1B,IAAwC;QAExC,IAAI,IAAI,EAAE,eAAe,EAAE;YACzB,OAAO,MAAM,KAAK,CAAC,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;SACpD;QACD,MAAM,IAAI,CAAC,WAAW,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;QAC9C,IAAI;YACF,OAAO,MAAM,KAAK,CAAC,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;SACpD;gBAAS;YACR,MAAM,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;SAC7C;IACH,CAAC;CACF;AAxCD,wCAwCC;AAED;;;;;GAKG;AACU,QAAA,0BAA0B,GAAG,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,+BAA+B,CAAC,CAAC,CAAC;AA+BlG,SAAS,WAAW,CAAC,EACnB,UAAU,EACV,MAAM,GACyB;IAC/B,OAAO;QACL,iBAAiB,EACf,OAAO,UAAU,KAAK,UAAU;YAC9B,CAAC,CAAC,UAAU;YACZ,CAAC,CAAC,CAAC,IAAY,EAAE,EAAE,CACf,IAAA,qBAAK,EAAC,UAAU,EAAE,IAAI,IAAI,gCAAwB,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE;gBAC/D,KAAK,EAAE,UAAU,EAAE,KAAK,IAAI,QAAQ;aACrC,CAAC;QACV,MAAM,EAAE,MAAM,IAAI,IAAI,sBAAa,CAAC,MAAM,CAAC;KAC5C,CAAC;AACJ,CAAC;AAED,gFAAgF;AAChF,sCAAsC;AACtC,MAAM,cAAc,GAAG,IAAI,QAAQ,CAAC,YAAY,EAAE,2BAA2B,CAAC,CAAC;AAE/E;;;;;GAKG;AACH,MAAa,uBAAuB;IAuBlC,YACqB,UAAwB,EAC3C,UAAsB,EACtB,gBAAkC;QAFf,eAAU,GAAV,UAAU,CAAc;QAyD7C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WAuCG;QACH,UAAK,GAAG,KAAK,EAAE,UAA2B,EAAiB,EAAE;YAC3D,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,2BAA2B,CAAC,EAAE,QAAQ,EAAE,IAAA,eAAM,EAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAClG,CAAC,CAAC;QA/FA,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC;QACzD,IAAI,CAAC,qBAAqB,GAAG,IAAI,8BAAqB,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC;IACzE,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAqC;QACvD,gEAAgE;QAChE,MAAM,OAAO,GAAG,CAAC,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,OAA6B,CAAC;QACjF,MAAM,IAAI,GAAG,MAAM,OAAO,EAAE,CAAC;QAE7B,MAAM,EAAE,iBAAiB,EAAE,MAAM,EAAE,GAAG,WAAW,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;QAE9D,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAEtC,MAAM,OAAO,GAAG,aAAa,IAAI,EAAE,CAAC;QACpC,MAAM,WAAW,GAAG,gCAAU,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QAEpD,IAAI;YACF,MAAM,OAAO,CAAC,IAAI,CAAC;gBACjB,WAAW;gBACX,IAAA,2BAAW,EAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE;oBAC3B,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;gBAClE,CAAC,CAAC;aACH,CAAC,CAAC;SACJ;QAAC,OAAO,GAAG,EAAE;YACZ,IAAI;gBACF,MAAM,IAAA,oBAAI,EAAC,KAAK,CAAC,CAAC;aACnB;YAAC,OAAO,KAAK,EAAE;gBACd,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;aACrE;YACD,MAAM,GAAG,CAAC;SACX;QAED,MAAM,IAAI,GAAG,MAAM,WAAW,CAAC;QAC/B,MAAM,gBAAgB,GAAG,MAAM,yBAAgB,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QAErE,OAAO,IAAI,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE,gBAAgB,CAAC,CAAC;IACjD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QAC9B,MAAM,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QACpC,6CAA6C;QAC7C,MAAM,IAAA,oBAAI,EAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;IACxE,CAAC;CA6CF;AA5HD,0DA4HC;AAED;;GAEG;AACU,QAAA,mBAAmB,GAAkB;IAChD,OAAO,EAAE,CAAC;IACV,SAAS,EAAE,MAAM;IACjB,OAAO,EAAE,KAAK;IACd,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;IAC9B,UAAU,EAAE,MAAM;IAClB,YAAY,EAAE,SAAS;IACvB,YAAY,EAAE,MAAM;IACpB,eAAe,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC;IACvD,gBAAgB,EAAE,SAAS;IAC3B,iBAAiB,EAAE,SAAS;IAC5B,iBAAiB,EAAE,SAAS;IAC5B,iBAAiB,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE;IAC7D,oBAAoB,EAAE,CAAC;IACvB,qBAAqB,EAAE,IAAI;IAC3B,wBAAwB,EAAE,IAAI;CAC/B,CAAC;AAEF;;;;;GAKG;AACH,MAAa,uBAAwB,SAAQ,gBAAM,CAAC,YAAY;IAI9D,YAAY,IAA6B;QACvC,KAAK,EAAE,CAAC;QAJH,WAAM,GAA2B,GAAG,EAAE,CAAC,SAAS,CAAC;QAKtD,MAAM,eAAe,GAAG,IAAI,kCAAe,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;YAC/C,IAAI,CAAC,MAAM,GAAG,CAAC,MAAY,EAAE,EAAE;gBAC7B,eAAe,CAAC,KAAK,EAAE,CAAC;gBACxB,MAAM,CAAC,IAAI,yBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC;YACvC,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,MAAM,iBAAiB,GAAG,CAAC,OAAiB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QACjF,IAAI,CAAC,OAAO,GAAG,IAAI,QAAQ,CAAC,OAAO,CACjC,EAAE,GAAG,2BAAmB,EAAE,GAAG,IAAI,EAAE,EACnC,OAAO,EACP,eAAe,CAAC,MAAM,EACtB,iBAAiB,CAClB,CAAC;QACF,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE;YACjB,+BAA+B;QACjC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACI,GAAG,CAAuD,EAAK,EAAE,GAAG,IAAO;QAChF,OAAO,QAAQ,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IACnE,CAAC;CACF;AA/BD,0DA+BC"}

package/lib/test-service-client.d.ts

@@ -1,13 +1,21 @@
 import * as grpc from '@grpc/grpc-js';
 import { Connection as BaseConnection, ConnectionOptions } from '@temporalio/client';
+import { ConnectionCtorOptions as BaseConnectionCtorOptions } from '@temporalio/client/lib/connection';
 import { temporal } from '../generated-protos';
 export declare type TestService = temporal.api.testservice.v1.TestService;
 export declare const TestService: typeof temporal.api.testservice.v1.TestService;
+interface ConnectionCtorOptions extends BaseConnectionCtorOptions {
+    testService: TestService;
+}
 /**
  * A Connection class that can be used to interact with both the test server's TestService and WorkflowService
  */
 export declare class Connection extends BaseConnection {
     static readonly TestServiceClient: grpc.ServiceClientConstructor;
     readonly testService: TestService;
-    constructor(options?: ConnectionOptions);
+    protected static createCtorOptions(options?: ConnectionOptions): ConnectionCtorOptions;
+    static lazy(options?: ConnectionOptions): Connection;
+    static connect(options?: ConnectionOptions): Promise<Connection>;
+    protected constructor(options: ConnectionCtorOptions);
 }
+export {};

package/lib/test-service-client.js

@@ -34,8 +34,27 @@ exports.TestService = generated_protos_1.temporal.api.testservice.v1.TestService
 class Connection extends client_1.Connection {
     constructor(options) {
         super(options);
-        const rpcImpl = this.generateRPCImplementation('temporal.api.testservice.v1.TestService');
-        this.testService = new exports.TestService(rpcImpl, false, false);
+        this.testService = options.testService;
+    }
+    static createCtorOptions(options) {
+        const ctorOptions = client_1.Connection.createCtorOptions(options);
+        const rpcImpl = this.generateRPCImplementation({
+            serviceName: 'temporal.api.testservice.v1.TestService',
+            client: ctorOptions.client,
+            callContextStorage: ctorOptions.callContextStorage,
+            interceptors: ctorOptions.options.interceptors,
+        });
+        const testService = exports.TestService.create(rpcImpl, false, false);
+        return { ...ctorOptions, testService };
+    }
+    static lazy(options) {
+        const ctorOptions = this.createCtorOptions(options);
+        return new this(ctorOptions);
+    }
+    static async connect(options) {
+        const ret = this.lazy(options);
+        await ret.ensureConnected();
+        return ret;
     }
 }
 exports.Connection = Connection;

package/lib/test-service-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"test-service-client.js","sourceRoot":"","sources":["../src/test-service-client.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,oDAAsC;AACtC,+CAAqF;AACrF,0DAA+C;AAGhC,mBAAW,GAAK,2BAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,aAAC;AAE3D;;GAEG;AACH,MAAa,UAAW,SAAQ,mBAAc;IAI5C,YAAY,OAA2B;QACrC,KAAK,CAAC,OAAO,CAAC,CAAC;QAEf,MAAM,OAAO,GAAG,IAAI,CAAC,yBAAyB,CAAC,yCAAyC,CAAC,CAAC;QAC1F,IAAI,CAAC,WAAW,GAAG,IAAI,mBAAW,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;IAC5D,CAAC;;AATH,gCAUC;AATwB,4BAAiB,GAAG,IAAI,CAAC,4BAA4B,CAAC,EAAE,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC"}
+{"version":3,"file":"test-service-client.js","sourceRoot":"","sources":["../src/test-service-client.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,oDAAsC;AACtC,+CAAqF;AAErF,0DAA+C;AAGhC,mBAAW,GAAK,2BAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,aAAC;AAM3D;;GAEG;AACH,MAAa,UAAW,SAAQ,mBAAc;IA2B5C,YAAsB,OAA8B;QAClD,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;IACzC,CAAC;IA1BS,MAAM,CAAC,iBAAiB,CAAC,OAA2B;QAC5D,MAAM,WAAW,GAAG,mBAAc,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAC9D,MAAM,OAAO,GAAG,IAAI,CAAC,yBAAyB,CAAC;YAC7C,WAAW,EAAE,yCAAyC;YACtD,MAAM,EAAE,WAAW,CAAC,MAAM;YAC1B,kBAAkB,EAAE,WAAW,CAAC,kBAAkB;YAClD,YAAY,EAAE,WAAW,CAAC,OAAO,CAAC,YAAY;SAC/C,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,mBAAW,CAAC,MAAM,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QAC9D,OAAO,EAAE,GAAG,WAAW,EAAE,WAAW,EAAE,CAAC;IACzC,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,OAA2B;QACrC,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QACpD,OAAO,IAAI,IAAI,CAAC,WAAW,CAAC,CAAC;IAC/B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAA2B;QAC9C,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/B,MAAM,GAAG,CAAC,eAAe,EAAE,CAAC;QAC5B,OAAO,GAAG,CAAC;IACb,CAAC;;AAzBH,gCA+BC;AA9BwB,4BAAiB,GAAG,IAAI,CAAC,4BAA4B,CAAC,EAAE,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC"}

package/lib/utils.js

@@ -6,7 +6,7 @@ async function waitOnNamespace(connection, namespace, maxAttempts = 100, retryIn
     const runId = '12345678-dead-beef-1234-1234567890ab';
     for (let attempt = 1; attempt <= maxAttempts; ++attempt) {
         try {
-            await connection.service.getWorkflowExecutionHistory({
+            await connection.workflowService.getWorkflowExecutionHistory({
                 namespace,
                 execution: { workflowId: 'fake', runId },
             });
@@ -26,7 +26,10 @@ exports.waitOnNamespace = waitOnNamespace;
 async function createNamespace(connection, namespace, maxAttempts = 100, retryIntervalSecs = 1) {
     for (let attempt = 1; attempt <= maxAttempts; ++attempt) {
         try {
-            await connection.service.registerNamespace({ namespace, workflowExecutionRetentionPeriod: (0, internal_workflow_common_1.msToTs)('1 day') });
+            await connection.workflowService.registerNamespace({
+                namespace,
+                workflowExecutionRetentionPeriod: (0, internal_workflow_common_1.msToTs)('1 day'),
+            });
             break;
         }
         catch (err) {

package/lib/utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":";;;AACA,mFAA8D;AAEvD,KAAK,UAAU,eAAe,CACnC,UAAsB,EACtB,SAAiB,EACjB,WAAW,GAAG,GAAG,EACjB,iBAAiB,GAAG,CAAC;IAErB,MAAM,KAAK,GAAG,sCAAsC,CAAC;IACrD,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,WAAW,EAAE,EAAE,OAAO,EAAE;QACvD,IAAI;YACF,MAAM,UAAU,CAAC,OAAO,CAAC,2BAA2B,CAAC;gBACnD,SAAS;gBACT,SAAS,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE;aACzC,CAAC,CAAC;SACJ;QAAC,OAAO,GAAQ,EAAE;YACjB,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,4BAA4B,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE;gBACrF,MAAM;aACP;YACD,IAAI,OAAO,KAAK,WAAW,EAAE;gBAC3B,MAAM,GAAG,CAAC;aACX;YACD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,IAAI,CAAC,CAAC,CAAC;SAC/E;KACF;AACH,CAAC;AAvBD,0CAuBC;AAEM,KAAK,UAAU,eAAe,CACnC,UAAsB,EACtB,SAAiB,EACjB,WAAW,GAAG,GAAG,EACjB,iBAAiB,GAAG,CAAC;IAErB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,WAAW,EAAE,EAAE,OAAO,EAAE;QACvD,IAAI;YACF,MAAM,UAAU,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,SAAS,EAAE,gCAAgC,EAAE,IAAA,iCAAM,EAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC7G,MAAM;SACP;QAAC,OAAO,GAAQ,EAAE;YACjB,IAAI,GAAG,CAAC,OAAO,KAAK,2BAA2B,EAAE;gBAC/C,MAAM;aACP;YACD,IAAI,OAAO,KAAK,WAAW,EAAE;gBAC3B,MAAM,GAAG,CAAC;aACX;YACD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,IAAI,CAAC,CAAC,CAAC;SAC/E;KACF;AACH,CAAC;AApBD,0CAoBC"}
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":";;;AACA,mFAA8D;AAEvD,KAAK,UAAU,eAAe,CACnC,UAAsB,EACtB,SAAiB,EACjB,WAAW,GAAG,GAAG,EACjB,iBAAiB,GAAG,CAAC;IAErB,MAAM,KAAK,GAAG,sCAAsC,CAAC;IACrD,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,WAAW,EAAE,EAAE,OAAO,EAAE;QACvD,IAAI;YACF,MAAM,UAAU,CAAC,eAAe,CAAC,2BAA2B,CAAC;gBAC3D,SAAS;gBACT,SAAS,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE;aACzC,CAAC,CAAC;SACJ;QAAC,OAAO,GAAQ,EAAE;YACjB,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,4BAA4B,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE;gBACrF,MAAM;aACP;YACD,IAAI,OAAO,KAAK,WAAW,EAAE;gBAC3B,MAAM,GAAG,CAAC;aACX;YACD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,IAAI,CAAC,CAAC,CAAC;SAC/E;KACF;AACH,CAAC;AAvBD,0CAuBC;AAEM,KAAK,UAAU,eAAe,CACnC,UAAsB,EACtB,SAAiB,EACjB,WAAW,GAAG,GAAG,EACjB,iBAAiB,GAAG,CAAC;IAErB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,WAAW,EAAE,EAAE,OAAO,EAAE;QACvD,IAAI;YACF,MAAM,UAAU,CAAC,eAAe,CAAC,iBAAiB,CAAC;gBACjD,SAAS;gBACT,gCAAgC,EAAE,IAAA,iCAAM,EAAC,OAAO,CAAC;aAClD,CAAC,CAAC;YACH,MAAM;SACP;QAAC,OAAO,GAAQ,EAAE;YACjB,IAAI,GAAG,CAAC,OAAO,KAAK,2BAA2B,EAAE;gBAC/C,MAAM;aACP;YACD,IAAI,OAAO,KAAK,WAAW,EAAE;gBAC3B,MAAM,GAAG,CAAC;aACX;YACD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,IAAI,CAAC,CAAC,CAAC;SAC/E;KACF;AACH,CAAC;AAvBD,0CAuBC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@temporalio/testing",
-  "version": "0.23.3",
+  "version": "1.0.0",
   "description": "Temporal.io SDK Testing sub-package",
   "main": "lib/index.js",
   "types": "./lib/index.d.ts",
@@ -17,17 +17,17 @@
   "author": "Temporal Technologies Inc. <sdk@temporal.io>",
   "license": "MIT",
   "dependencies": {
-    "@grpc/grpc-js": "^1.5.7",
-    "@temporalio/activity": "^0.23.0",
-    "@temporalio/client": "^0.23.0",
-    "@temporalio/common": "^0.23.0",
-    "@temporalio/worker": "^0.23.2",
-    "@types/long": "^4.0.1",
+    "@grpc/grpc-js": "^1.6.7",
+    "@temporalio/activity": "^1.0.0",
+    "@temporalio/client": "^1.0.0",
+    "@temporalio/common": "^1.0.0",
+    "@temporalio/worker": "^1.0.0",
+    "@types/long": "^4.0.2",
     "abort-controller": "^3.0.0",
     "get-port": "^6.1.2",
-    "got": "^12.0.1",
-    "long": "^4.0.0",
-    "protobufjs": "^6.11.2",
+    "got": "^12.1.0",
+    "long": "^5.2.0",
+    "protobufjs": "6.11.2",
     "tar-stream": "^2.2.0",
     "unzipper": "^0.10.11"
   },
@@ -37,11 +37,12 @@
   "homepage": "https://github.com/temporalio/sdk-typescript/tree/main/packages/testing",
   "files": [
     "lib",
+    "src",
     "generated-protos",
     "scripts"
   ],
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "2c7ed196ac2379c06419a3c98a7e7eddf1a53b01"
+  "gitHead": "c4fc4dc608bf58701c11b6ae02d1d63b4457718d"
 }

package/scripts/compile-proto.mjs

@@ -58,24 +58,17 @@ async function compileProtos(protoPath, jsOutputFile, dtsOutputFile, ...args) {
   );
 }
 
-async function main() {
-  mkdirSync(outputDir, { recursive: true });
+mkdirSync(outputDir, { recursive: true });
 
-  const protoFiles = glob.sync(resolve(protoBaseDir, '**/*.proto'));
-  const protosMTime = Math.max(...protoFiles.map(mtime));
-  const genMTime = mtime(outputFile);
+const protoFiles = glob.sync(resolve(protoBaseDir, '**/*.proto'));
+const protosMTime = Math.max(...protoFiles.map(mtime));
+const genMTime = mtime(outputFile);
 
-  if (protosMTime < genMTime) {
-    console.log('Assuming protos are up to date');
-    return;
-  }
-
-  await compileProtos(serviceProtoPath, outputFile, resolve(outputDir, 'index.d.ts'), '--path', resolve(protoBaseDir));
-
-  console.log('Done');
+if (protosMTime < genMTime) {
+  console.log('Assuming protos are up to date');
+  process.exit(0);
 }
 
-main().catch((err) => {
-  console.error(err);
-  process.exit(1);
-});
+await compileProtos(serviceProtoPath, outputFile, resolve(outputDir, 'index.d.ts'), '--path', resolve(protoBaseDir));
+
+console.log('Done');

package/src/assert-to-failure-interceptor.ts

@@ -0,0 +1,38 @@
+import assert from 'assert';
+import { ApplicationFailure, WorkflowInterceptors } from '@temporalio/workflow';
+
+/**
+ * Simple interceptor that transforms {@link assert.AssertionError} into non retryable failures.
+ *
+ * This allows conveniently using `assert` directly from Workflows.
+ */
+export function interceptors(): WorkflowInterceptors {
+  return {
+    inbound: [
+      {
+        async handleSignal(input, next) {
+          try {
+            return await next(input);
+          } catch (err) {
+            if (err instanceof assert.AssertionError) {
+              const appErr = ApplicationFailure.nonRetryable(err.message);
+              appErr.stack = err.stack;
+              throw appErr;
+            }
+          }
+        },
+        async execute(input, next) {
+          try {
+            return await next(input);
+          } catch (err) {
+            if (err instanceof assert.AssertionError) {
+              const appErr = ApplicationFailure.nonRetryable(err.message);
+              appErr.stack = err.stack;
+              throw appErr;
+            }
+          }
+        },
+      },
+    ],
+  };
+}

package/src/child-process.ts

@@ -0,0 +1,60 @@
+// TODO: this code is duplicated in the scripts directory, consider moving to external dependency
+import { ChildProcess } from 'child_process';
+
+export class ChildProcessError extends Error {
+  public readonly name = 'ChildProcessError';
+
+  constructor(message: string, public readonly code: number | null, public readonly signal: NodeJS.Signals | null) {
+    super(message);
+  }
+}
+
+export interface WaitOptions {
+  validReturnCodes: number[];
+}
+
+export async function waitOnChild(child: ChildProcess, opts?: WaitOptions): Promise<void> {
+  return new Promise((resolve, reject) => {
+    child.on('exit', (code, signal) => {
+      if (code !== null && (opts?.validReturnCodes ?? [0]).includes(code)) {
+        resolve();
+      } else {
+        reject(new ChildProcessError('Process failed', code, signal));
+      }
+    });
+    child.on('error', reject);
+  });
+}
+
+export async function kill(child: ChildProcess, signal: NodeJS.Signals = 'SIGINT', opts?: WaitOptions): Promise<void> {
+  if (child.pid === undefined) {
+    throw new TypeError('Expected child with pid');
+  }
+  process.kill(child.pid, signal);
+  try {
+    await waitOnChild(child, opts);
+  } catch (err) {
+    // Should error if the error is not a child process error or it is a child
+    // process and either the platform is Windows or the signal matches.
+    const shouldError = !(err instanceof ChildProcessError) || (process.platform !== 'win32' && err.signal !== signal);
+    if (shouldError) {
+      throw err;
+    }
+  }
+}
+
+export async function killIfExists(
+  child: ChildProcess,
+  signal: NodeJS.Signals = 'SIGINT',
+  opts?: WaitOptions
+): Promise<void> {
+  try {
+    await kill(child, signal, opts);
+  } catch (err: any) {
+    if (err.code !== 'ESRCH') {
+      throw err;
+    }
+  }
+}
+
+export const shell = process.platform === 'win32';

package/src/index-for-docs.ts

@@ -0,0 +1,2 @@
+// Reexports the contents of this package under the "testing" namespace for better docs generation
+export * as testing from './index';

package/src/index.ts

@@ -0,0 +1,358 @@
+/**
+ * `npm i @temporalio/testing`
+ *
+ * Testing library for the SDK.
+ *
+ * [Documentation](https://docs.temporal.io/typescript/testing)
+ *
+ * @module
+ */
+
+import * as activity from '@temporalio/activity';
+import {
+  AsyncCompletionClient,
+  WorkflowClient as BaseWorkflowClient,
+  WorkflowClientOptions as BaseWorkflowClientOptions,
+  WorkflowResultOptions as BaseWorkflowResultOptions,
+  WorkflowStartOptions as BaseWorkflowStartOptions,
+} from '@temporalio/client';
+import { ActivityFunction, CancelledFailure, msToTs, Workflow, WorkflowResultType } from '@temporalio/common';
+import { NativeConnection, Logger, DefaultLogger } from '@temporalio/worker';
+import path from 'path';
+import os from 'os';
+import { AbortController } from 'abort-controller';
+import { ChildProcess, spawn, StdioOptions } from 'child_process';
+import events from 'events';
+import { kill, waitOnChild } from './child-process';
+import type getPortType from 'get-port';
+import { Connection, TestService } from './test-service-client';
+
+const TEST_SERVER_EXECUTABLE_NAME = os.platform() === 'win32' ? 'test-server.exe' : 'test-server';
+
+export const DEFAULT_TEST_SERVER_PATH = path.join(__dirname, `../${TEST_SERVER_EXECUTABLE_NAME}`);
+
+/**
+ * Options passed to {@link WorkflowClient.result}, these are the same as the
+ * {@link BaseWorkflowResultOptions} with an additional option that controls
+ * whether to toggle time skipping in the Test server while waiting on a
+ * Workflow's result.
+ */
+export interface WorkflowResultOptions extends BaseWorkflowResultOptions {
+  /**
+   * If set to `true`, waiting for the result does not enable time skipping
+   */
+  runInNormalTime?: boolean;
+}
+
+/**
+ * Options passed to {@link WorkflowClient.execute}, these are the same as the
+ * {@link BaseWorkflowStartOptions} with an additional option that controls
+ * whether to toggle time skipping in the Test server while waiting on a
+ * Workflow's result.
+ */
+export type WorkflowStartOptions<T extends Workflow> = BaseWorkflowStartOptions<T> & {
+  /**
+   * If set to `true`, waiting for the result does not enable time skipping
+   */
+  runInNormalTime?: boolean;
+};
+
+export interface WorkflowClientOptions extends BaseWorkflowClientOptions {
+  connection: Connection;
+}
+
+/**
+ * A client with the exact same API as the "normal" client with 1 exception,
+ * When this client waits on a Workflow's result, it will enable time skipping
+ * in the test server.
+ */
+export class WorkflowClient extends BaseWorkflowClient {
+  protected readonly testService: TestService;
+
+  constructor(options: WorkflowClientOptions) {
+    super(options);
+    this.testService = options.connection.testService;
+  }
+
+  /**
+   * Execute a Workflow and wait for completion.
+   *
+   * @see {@link BaseWorkflowClient.execute}
+   */
+  public async execute<T extends Workflow>(
+    workflowTypeOrFunc: string | T,
+    options: WorkflowStartOptions<T>
+  ): Promise<WorkflowResultType<T>> {
+    return super.execute(workflowTypeOrFunc, options);
+  }
+
+  /**
+   * Gets the result of a Workflow execution.
+   *
+   * @see {@link BaseWorkflowClient.result}
+   */
+  override async result<T>(
+    workflowId: string,
+    runId?: string | undefined,
+    opts?: WorkflowResultOptions | undefined
+  ): Promise<T> {
+    if (opts?.runInNormalTime) {
+      return await super.result(workflowId, runId, opts);
+    }
+    await this.testService.unlockTimeSkipping({});
+    try {
+      return await super.result(workflowId, runId, opts);
+    } finally {
+      await this.testService.lockTimeSkipping({});
+    }
+  }
+}
+
+/**
+ * Convenience workflow interceptors
+ *
+ * Contains a single interceptor for transforming `AssertionError`s into non
+ * retryable `ApplicationFailure`s.
+ */
+export const workflowInterceptorModules = [path.join(__dirname, 'assert-to-failure-interceptor')];
+
+export interface TestServerSpawnerOptions {
+  /**
+   * @default {@link DEFAULT_TEST_SERVER_PATH}
+   */
+  path?: string;
+  /**
+   * @default ignore
+   */
+  stdio?: StdioOptions;
+}
+
+/**
+ * A generic callback that returns a child process
+ */
+export type TestServerSpawner = (port: number) => ChildProcess;
+
+/**
+ * Options for {@link TestWorkflowEnvironment.create}
+ */
+export interface TestWorkflowEnvironmentOptions {
+  testServer?: TestServerSpawner | TestServerSpawnerOptions;
+  logger?: Logger;
+}
+
+interface TestWorkflowEnvironmentOptionsWithDefaults {
+  testServerSpawner: TestServerSpawner;
+  logger: Logger;
+}
+
+function addDefaults({
+  testServer,
+  logger,
+}: TestWorkflowEnvironmentOptions): TestWorkflowEnvironmentOptionsWithDefaults {
+  return {
+    testServerSpawner:
+      typeof testServer === 'function'
+        ? testServer
+        : (port: number) =>
+            spawn(testServer?.path || DEFAULT_TEST_SERVER_PATH, [`${port}`], {
+              stdio: testServer?.stdio || 'ignore',
+            }),
+    logger: logger ?? new DefaultLogger('INFO'),
+  };
+}
+
+// TS transforms `import` statements into `require`s, this is a workaround until
+// tsconfig module nodenext is stable.
+const _importDynamic = new Function('modulePath', 'return import(modulePath)');
+
+/**
+ * An execution environment for running Workflow integration tests.
+ *
+ * Runs an external server.
+ * By default, the Java test server is used which supports time skipping.
+ */
+export class TestWorkflowEnvironment {
+  /**
+   * Get an extablished {@link Connection} to the test server
+   */
+  public readonly connection: Connection;
+
+  /**
+   * An {@link AsyncCompletionClient} for interacting with the test server
+   */
+  public readonly asyncCompletionClient: AsyncCompletionClient;
+
+  /**
+   * A {@link WorkflowClient} for interacting with the test server
+   */
+  public readonly workflowClient: WorkflowClient;
+
+  /**
+   * A {@link NativeConnection} for interacting with the test server.
+   *
+   * Use this connection when creating Workers for testing.
+   */
+  public readonly nativeConnection: NativeConnection;
+
+  protected constructor(
+    protected readonly serverProc: ChildProcess,
+    connection: Connection,
+    nativeConnection: NativeConnection
+  ) {
+    this.connection = connection;
+    this.nativeConnection = nativeConnection;
+    this.workflowClient = new WorkflowClient({ connection });
+    this.asyncCompletionClient = new AsyncCompletionClient({ connection });
+  }
+
+  /**
+   * Create a new test environment
+   */
+  static async create(opts?: TestWorkflowEnvironmentOptions): Promise<TestWorkflowEnvironment> {
+    // No, we're not going to compile this to ESM for one dependency
+    const getPort = (await _importDynamic('get-port')).default as typeof getPortType;
+    const port = await getPort();
+
+    const { testServerSpawner, logger } = addDefaults(opts ?? {});
+
+    const child = testServerSpawner(port);
+
+    const address = `127.0.0.1:${port}`;
+    const connPromise = Connection.connect({ address });
+
+    try {
+      await Promise.race([
+        connPromise,
+        waitOnChild(child).then(() => {
+          throw new Error('Test server child process exited prematurely');
+        }),
+      ]);
+    } catch (err) {
+      try {
+        await kill(child);
+      } catch (error) {
+        logger.error('Failed to kill test server child process', { error });
+      }
+      throw err;
+    }
+
+    const conn = await connPromise;
+    const nativeConnection = await NativeConnection.connect({ address });
+
+    return new this(child, conn, nativeConnection);
+  }
+
+  /**
+   * Kill the test server process and close the connection to it
+   */
+  async teardown(): Promise<void> {
+    await this.connection.close();
+    await this.nativeConnection.close();
+    // TODO: the server should return exit code 0
+    await kill(this.serverProc, 'SIGINT', { validReturnCodes: [0, 130] });
+  }
+
+  /**
+   * Wait for `durationMs` in "test server time".
+   *
+   * The test server toggles between skipped time and normal time depending on what it needs to execute.
+   *
+   * This method is likely to resolve in less than `durationMs` of "real time".
+   *
+   * Useful for simulating events far into the future like completion of long running activities.
+   *
+   * @param durationMs {@link https://www.npmjs.com/package/ms | ms} formatted string or number of milliseconds
+   *
+   * @example
+   *
+   * `workflow.ts`
+   *
+   * ```ts
+   * const activities = proxyActivities({ startToCloseTimeout: 2_000_000 });
+   *
+   * export async function raceActivityAndTimer(): Promise<string> {
+   *   return await Promise.race([
+   *     wf.sleep(500_000).then(() => 'timer'),
+   *     activities.longRunning().then(() => 'activity'),
+   *   ]);
+   * }
+   * ```
+   *
+   * `test.ts`
+   *
+   * ```ts
+   * const worker = await Worker.create({
+   *   connection: testEnv.nativeConnection,
+   *   activities: {
+   *     async longRunning() {
+   *       await testEnv.sleep(1_000_000); // <-- sleep called here
+   *     },
+   *   },
+   *   // ...
+   * });
+   * ```
+   */
+  sleep = async (durationMs: number | string): Promise<void> => {
+    await this.connection.testService.unlockTimeSkippingWithSleep({ duration: msToTs(durationMs) });
+  };
+}
+
+/**
+ * Used as the default activity info for Activities executed in the {@link MockActivityEnvironment}
+ */
+export const defaultActivityInfo: activity.Info = {
+  attempt: 1,
+  taskQueue: 'test',
+  isLocal: false,
+  taskToken: Buffer.from('test'),
+  activityId: 'test',
+  activityType: 'unknown',
+  workflowType: 'test',
+  base64TaskToken: Buffer.from('test').toString('base64'),
+  heartbeatDetails: undefined,
+  activityNamespace: 'default',
+  workflowNamespace: 'default',
+  workflowExecution: { workflowId: 'test', runId: 'dead-beef' },
+  scheduledTimestampMs: 1,
+  startToCloseTimeoutMs: 1000,
+  scheduleToCloseTimeoutMs: 1000,
+};
+
+/**
+ * An execution environment for testing Activities.
+ *
+ * Mocks Activity {@link Context | activity.Context} and exposes hooks for
+ * cancellation and heartbeats.
+ */
+export class MockActivityEnvironment extends events.EventEmitter {
+  public cancel: (reason?: any) => void = () => undefined;
+  public readonly context: activity.Context;
+
+  constructor(info?: Partial<activity.Info>) {
+    super();
+    const abortController = new AbortController();
+    const promise = new Promise<never>((_, reject) => {
+      this.cancel = (reason?: any) => {
+        abortController.abort();
+        reject(new CancelledFailure(reason));
+      };
+    });
+    const heartbeatCallback = (details?: unknown) => this.emit('heartbeat', details);
+    this.context = new activity.Context(
+      { ...defaultActivityInfo, ...info },
+      promise,
+      abortController.signal,
+      heartbeatCallback
+    );
+    promise.catch(() => {
+      /* avoid unhandled rejection */
+    });
+  }
+
+  /**
+   * Run a function in Activity Context
+   */
+  public run<P extends any[], R, F extends ActivityFunction<P, R>>(fn: F, ...args: P): Promise<R> {
+    return activity.asyncLocalStorage.run(this.context, fn, ...args);
+  }
+}

package/src/test-service-client.ts

@@ -0,0 +1,47 @@
+import * as grpc from '@grpc/grpc-js';
+import { Connection as BaseConnection, ConnectionOptions } from '@temporalio/client';
+import { ConnectionCtorOptions as BaseConnectionCtorOptions } from '@temporalio/client/lib/connection';
+import { temporal } from '../generated-protos';
+
+export type TestService = temporal.api.testservice.v1.TestService;
+export const { TestService } = temporal.api.testservice.v1;
+
+interface ConnectionCtorOptions extends BaseConnectionCtorOptions {
+  testService: TestService;
+}
+
+/**
+ * A Connection class that can be used to interact with both the test server's TestService and WorkflowService
+ */
+export class Connection extends BaseConnection {
+  public static readonly TestServiceClient = grpc.makeGenericClientConstructor({}, 'TestService', {});
+  public readonly testService: TestService;
+
+  protected static createCtorOptions(options?: ConnectionOptions): ConnectionCtorOptions {
+    const ctorOptions = BaseConnection.createCtorOptions(options);
+    const rpcImpl = this.generateRPCImplementation({
+      serviceName: 'temporal.api.testservice.v1.TestService',
+      client: ctorOptions.client,
+      callContextStorage: ctorOptions.callContextStorage,
+      interceptors: ctorOptions.options.interceptors,
+    });
+    const testService = TestService.create(rpcImpl, false, false);
+    return { ...ctorOptions, testService };
+  }
+
+  static lazy(options?: ConnectionOptions): Connection {
+    const ctorOptions = this.createCtorOptions(options);
+    return new this(ctorOptions);
+  }
+
+  static async connect(options?: ConnectionOptions): Promise<Connection> {
+    const ret = this.lazy(options);
+    await ret.ensureConnected();
+    return ret;
+  }
+
+  protected constructor(options: ConnectionCtorOptions) {
+    super(options);
+    this.testService = options.testService;
+  }
+}

package/src/utils.ts

@@ -0,0 +1,52 @@
+import { Connection } from '@temporalio/client';
+import { msToTs } from '@temporalio/internal-workflow-common';
+
+export async function waitOnNamespace(
+  connection: Connection,
+  namespace: string,
+  maxAttempts = 100,
+  retryIntervalSecs = 1
+): Promise<void> {
+  const runId = '12345678-dead-beef-1234-1234567890ab';
+  for (let attempt = 1; attempt <= maxAttempts; ++attempt) {
+    try {
+      await connection.workflowService.getWorkflowExecutionHistory({
+        namespace,
+        execution: { workflowId: 'fake', runId },
+      });
+    } catch (err: any) {
+      if (err.details.includes('workflow history not found') || err.details.includes(runId)) {
+        break;
+      }
+      if (attempt === maxAttempts) {
+        throw err;
+      }
+      await new Promise((resolve) => setTimeout(resolve, retryIntervalSecs * 1000));
+    }
+  }
+}
+
+export async function createNamespace(
+  connection: Connection,
+  namespace: string,
+  maxAttempts = 100,
+  retryIntervalSecs = 1
+): Promise<void> {
+  for (let attempt = 1; attempt <= maxAttempts; ++attempt) {
+    try {
+      await connection.workflowService.registerNamespace({
+        namespace,
+        workflowExecutionRetentionPeriod: msToTs('1 day'),
+      });
+      break;
+    } catch (err: any) {
+      if (err.details === 'Namespace already exists.') {
+        break;
+      }
+      if (attempt === maxAttempts) {
+        throw err;
+      }
+      await new Promise((resolve) => setTimeout(resolve, retryIntervalSecs * 1000));
+    }
+  }
+}