@azure-rest/load-testing 1.1.0-beta.1 → 1.2.0-beta.1

package/dist/esm/getTestProfileRunCompletionPoller.js

@@ -1,127 +1,119 @@
-// Copyright (c) Microsoft Corporation.
-// Licensed under the MIT License.
 import { AbortError } from "@azure/abort-controller";
 import { isUnexpected } from "./isUnexpected.js";
 import { sleep, isTestProfileRunInProgress } from "./util/LROUtil.js";
-/**
- * Creates a poller to poll for test run status.
- * @param client - The Load Testing client.
- * @param options - The operation options.
- * @returns A poller which can be called to poll until completion of the job.
- */
-export async function getTestProfileRunCompletionPoller(client, createTestProfileRunResponse, polledOperationOptions = {}) {
-    var _a;
-    const state = {
-        status: "notStarted",
-    };
-    const progressCallbacks = new Map();
-    const processProgressCallbacks = async () => progressCallbacks.forEach((h) => h(state));
-    let resultPromise;
-    let cancelJob;
-    const abortController = new AbortController();
-    const currentPollIntervalInMs = (_a = polledOperationOptions.updateIntervalInMs) !== null && _a !== void 0 ? _a : 10000;
-    const testProfileRunId = createTestProfileRunResponse.body.testProfileRunId;
-    const poller = {
-        async poll(options) {
-            var _a;
-            if ((_a = options === null || options === void 0 ? void 0 : options.abortSignal) === null || _a === void 0 ? void 0 : _a.aborted) {
-                throw new AbortError("The polling was aborted.");
+async function getTestProfileRunCompletionPoller(client, createTestProfileRunResponse, polledOperationOptions = {}) {
+  const state = {
+    status: "notStarted"
+  };
+  const progressCallbacks = /* @__PURE__ */ new Map();
+  const processProgressCallbacks = async () => progressCallbacks.forEach((h) => h(state));
+  let resultPromise;
+  let cancelJob;
+  const abortController = new AbortController();
+  const currentPollIntervalInMs = polledOperationOptions.updateIntervalInMs ?? 1e4;
+  const testProfileRunId = createTestProfileRunResponse.body.testProfileRunId;
+  const poller = {
+    async poll(options) {
+      if (options?.abortSignal?.aborted) {
+        throw new AbortError("The polling was aborted.");
+      }
+      if (testProfileRunId) {
+        const getTestProfileRunResult = await client.path("/test-profile-runs/{testProfileRunId}", testProfileRunId).get();
+        if (isUnexpected(getTestProfileRunResult)) {
+          state.status = "failed";
+          state.error = new Error(getTestProfileRunResult.body.error.message);
+          return state;
+        }
+        if (getTestProfileRunResult.body.status === "FAILED") {
+          state.status = "failed";
+          state.error = new Error(getTestProfileRunResult.body.status);
+        }
+        if (getTestProfileRunResult.body.status === "CANCELLED") {
+          state.status = "canceled";
+        }
+        if (getTestProfileRunResult.body.status === "DONE") {
+          state.status = "succeeded";
+        }
+        if (isTestProfileRunInProgress(getTestProfileRunResult.body)) {
+          state.status = "running";
+        }
+        state.result = getTestProfileRunResult;
+        await processProgressCallbacks();
+      }
+      return state;
+    },
+    pollUntilDone(pollOptions) {
+      return resultPromise ??= (async () => {
+        const { abortSignal: inputAbortSignal } = pollOptions || {};
+        function abortListener() {
+          abortController.abort();
+        }
+        const abortSignal = abortController.signal;
+        if (inputAbortSignal?.aborted) {
+          abortController.abort();
+        } else if (!abortSignal.aborted) {
+          inputAbortSignal?.addEventListener("abort", abortListener, { once: true });
+        }
+        try {
+          if (!poller.isDone()) {
+            await poller.poll({ abortSignal });
+            while (!poller.isDone()) {
+              const delay = sleep(currentPollIntervalInMs, abortSignal);
+              cancelJob = () => abortController.abort();
+              await delay;
+              await poller.poll({ abortSignal });
             }
-            if (testProfileRunId) {
-                const getTestProfileRunResult = await client
-                    .path("/test-profile-runs/{testProfileRunId}", testProfileRunId)
-                    .get();
-                if (isUnexpected(getTestProfileRunResult)) {
-                    state.status = "failed";
-                    state.error = new Error(getTestProfileRunResult.body.error.message);
-                    return;
-                }
-                if (getTestProfileRunResult.body.status === "FAILED") {
-                    state.status = "failed";
-                    state.error = new Error(getTestProfileRunResult.body.status);
-                }
-                if (getTestProfileRunResult.body.status === "CANCELLED") {
-                    state.status = "canceled";
-                }
-                if (getTestProfileRunResult.body.status === "DONE") {
-                    state.status = "succeeded";
-                }
-                if (isTestProfileRunInProgress(getTestProfileRunResult.body)) {
-                    state.status = "running";
-                }
-                state.result = getTestProfileRunResult;
-                await processProgressCallbacks();
-            }
-        },
-        pollUntilDone(pollOptions) {
-            return (resultPromise !== null && resultPromise !== void 0 ? resultPromise : (resultPromise = (async () => {
-                const { abortSignal: inputAbortSignal } = pollOptions || {};
-                // In the future we can use AbortSignal.any() instead
-                function abortListener() {
-                    abortController.abort();
-                }
-                const abortSignal = abortController.signal;
-                if (inputAbortSignal === null || inputAbortSignal === void 0 ? void 0 : inputAbortSignal.aborted) {
-                    abortController.abort();
-                }
-                else if (!abortSignal.aborted) {
-                    inputAbortSignal === null || inputAbortSignal === void 0 ? void 0 : inputAbortSignal.addEventListener("abort", abortListener, { once: true });
-                }
-                try {
-                    if (!poller.isDone()) {
-                        await poller.poll({ abortSignal });
-                        while (!poller.isDone()) {
-                            const delay = sleep(currentPollIntervalInMs, abortSignal);
-                            cancelJob = () => abortController.abort();
-                            await delay;
-                            await poller.poll({ abortSignal });
-                        }
-                    }
-                }
-                finally {
-                    inputAbortSignal === null || inputAbortSignal === void 0 ? void 0 : inputAbortSignal.removeEventListener("abort", abortListener);
-                }
-                switch (state.status) {
-                    case "succeeded":
-                    case "failed":
-                    case "canceled": {
-                        return poller.getResult();
-                    }
-                    case "notStarted":
-                    case "running": {
-                        // Unreachable
-                        throw new Error(`polling completed without succeeding or failing`);
-                    }
-                }
-            })().finally(() => {
-                resultPromise = undefined;
-            })));
-        },
-        onProgress(callback) {
-            const s = Symbol();
-            progressCallbacks.set(s, callback);
-            return () => progressCallbacks.delete(s);
-        },
-        isDone() {
-            return ["succeeded", "failed", "canceled"].includes(state.status);
-        },
-        stopPolling() {
-            abortController.abort();
-            cancelJob === null || cancelJob === void 0 ? void 0 : cancelJob();
-        },
-        isStopped() {
-            return resultPromise === undefined;
-        },
-        getOperationState() {
-            return state;
-        },
-        getResult() {
-            return state.result;
-        },
-        toString() {
-            return JSON.stringify({ state });
-        },
-    };
-    return poller;
+          }
+        } finally {
+          inputAbortSignal?.removeEventListener("abort", abortListener);
+        }
+        switch (state.status) {
+          case "succeeded":
+          case "failed":
+          case "canceled": {
+            return poller.getResult();
+          }
+          case "notStarted":
+          case "running": {
+            throw new Error(`polling completed without succeeding or failing`);
+          }
+        }
+      })().finally(() => {
+        resultPromise = void 0;
+      });
+    },
+    onProgress(callback) {
+      const s = /* @__PURE__ */ Symbol();
+      progressCallbacks.set(s, callback);
+      return () => progressCallbacks.delete(s);
+    },
+    isDone() {
+      return ["succeeded", "failed", "canceled"].includes(state.status);
+    },
+    stopPolling() {
+      abortController.abort();
+      cancelJob?.();
+    },
+    isStopped() {
+      return resultPromise === void 0;
+    },
+    getOperationState() {
+      return state;
+    },
+    getResult() {
+      return state.result;
+    },
+    toString() {
+      return JSON.stringify({ state });
+    },
+    async serialize() {
+      return JSON.stringify({ state });
+    },
+    async submitted() {
+    }
+  };
+  return poller;
 }
-//# sourceMappingURL=getTestProfileRunCompletionPoller.js.map
+export {
+  getTestProfileRunCompletionPoller
+};

package/dist/esm/getTestProfileRunCompletionPoller.js.map

@@ -1 +1,7 @@
-{"version":3,"file":"getTestProfileRunCompletionPoller.js","sourceRoot":"","sources":["../../src/getTestProfileRunCompletionPoller.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AASrD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,KAAK,EAAE,0BAA0B,EAAE,MAAM,mBAAmB,CAAC;AAEtE;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,iCAAiC,CACrD,MAA8B,EAC9B,4BAEuE,EACvE,yBAAiD,EAAE;;IAMnD,MAAM,KAAK,GAA6E;QACtF,MAAM,EAAE,YAAY;KACrB,CAAC;IAEF,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAmB,CAAC;IACrD,MAAM,wBAAwB,GAAG,KAAK,IAAmB,EAAE,CACzD,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IAC7C,IAAI,aAA4F,CAAC;IACjG,IAAI,SAAmC,CAAC;IACxC,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;IAC9C,MAAM,uBAAuB,GAAG,MAAA,sBAAsB,CAAC,kBAAkB,mCAAI,KAAK,CAAC;IACnF,MAAM,gBAAgB,GAAG,4BAA4B,CAAC,IAAI,CAAC,gBAAgB,CAAC;IAE5E,MAAM,MAAM,GAGR;QACF,KAAK,CAAC,IAAI,CAAC,OAA2C;;YACpD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW,0CAAE,OAAO,EAAE,CAAC;gBAClC,MAAM,IAAI,UAAU,CAAC,0BAA0B,CAAC,CAAC;YACnD,CAAC;YAED,IAAI,gBAAgB,EAAE,CAAC;gBACrB,MAAM,uBAAuB,GAAG,MAAM,MAAM;qBACzC,IAAI,CAAC,uCAAuC,EAAE,gBAAgB,CAAC;qBAC/D,GAAG,EAAE,CAAC;gBACT,IAAI,YAAY,CAAC,uBAAuB,CAAC,EAAE,CAAC;oBAC1C,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAC;oBACxB,KAAK,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC,uBAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACpE,OAAO;gBACT,CAAC;gBAED,IAAI,uBAAuB,CAAC,IAAI,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;oBACrD,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAC;oBACxB,KAAK,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC,uBAAuB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC/D,CAAC;gBAED,IAAI,uBAAuB,CAAC,IAAI,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;oBACxD,KAAK,CAAC,MAAM,GAAG,UAAU,CAAC;gBAC5B,CAAC;gBAED,IAAI,uBAAuB,CAAC,IAAI,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;oBACnD,KAAK,CAAC,MAAM,GAAG,WAAW,CAAC;gBAC7B,CAAC;gBAED,IAAI,0BAA0B,CAAC,uBAAuB,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC7D,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC;gBAC3B,CAAC;gBACD,KAAK,CAAC,MAAM,GAAG,uBAAuB,CAAC;gBACvC,MAAM,wBAAwB,EAAE,CAAC;YACnC,CAAC;QACH,CAAC;QAED,aAAa,CAAC,WAEb;YACC,OAAO,CAAC,aAAa,aAAb,aAAa,cAAb,aAAa,IAAb,aAAa,GAAK,CAAC,KAAK,IAAI,EAAE;gBACpC,MAAM,EAAE,WAAW,EAAE,gBAAgB,EAAE,GAAG,WAAW,IAAI,EAAE,CAAC;gBAC5D,qDAAqD;gBACrD,SAAS,aAAa;oBACpB,eAAe,CAAC,KAAK,EAAE,CAAC;gBAC1B,CAAC;gBACD,MAAM,WAAW,GAAG,eAAe,CAAC,MAAM,CAAC;gBAC3C,IAAI,gBAAgB,aAAhB,gBAAgB,uBAAhB,gBAAgB,CAAE,OAAO,EAAE,CAAC;oBAC9B,eAAe,CAAC,KAAK,EAAE,CAAC;gBAC1B,CAAC;qBAAM,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;oBAChC,gBAAgB,aAAhB,gBAAgB,uBAAhB,gBAAgB,CAAE,gBAAgB,CAAC,OAAO,EAAE,aAAa,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC7E,CAAC;gBAED,IAAI,CAAC;oBACH,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;wBACrB,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC;wBACnC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;4BACxB,MAAM,KAAK,GAAG,KAAK,CAAC,uBAAuB,EAAE,WAAW,CAAC,CAAC;4BAC1D,SAAS,GAAG,GAAG,EAAE,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;4BAC1C,MAAM,KAAK,CAAC;4BACZ,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC;wBACrC,CAAC;oBACH,CAAC;gBACH,CAAC;wBAAS,CAAC;oBACT,gBAAgB,aAAhB,gBAAgB,uBAAhB,gBAAgB,CAAE,mBAAmB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;gBAChE,CAAC;gBACD,QAAQ,KAAK,CAAC,MAAM,EAAE,CAAC;oBACrB,KAAK,WAAW,CAAC;oBACjB,KAAK,QAAQ,CAAC;oBACd,KAAK,UAAU,CAAC,CAAC,CAAC;wBAChB,OAAO,MAAM,CAAC,SAAS,EAA8D,CAAC;oBACxF,CAAC;oBACD,KAAK,YAAY,CAAC;oBAClB,KAAK,SAAS,CAAC,CAAC,CAAC;wBACf,cAAc;wBACd,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;oBACrE,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBAChB,aAAa,GAAG,SAAS,CAAC;YAC5B,CAAC,CAAC,EAAC,CAAC;QACN,CAAC;QAED,UAAU,CACR,QAES;YAET,MAAM,CAAC,GAAG,MAAM,EAAE,CAAC;YACnB,iBAAiB,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;YAEnC,OAAO,GAAG,EAAE,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM;YACJ,OAAO,CAAC,WAAW,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACpE,CAAC;QAED,WAAW;YACT,eAAe,CAAC,KAAK,EAAE,CAAC;YACxB,SAAS,aAAT,SAAS,uBAAT,SAAS,EAAI,CAAC;QAChB,CAAC;QAED,SAAS;YACP,OAAO,aAAa,KAAK,SAAS,CAAC;QACrC,CAAC;QAED,iBAAiB;YACf,OAAO,KAAK,CAAC;QACf,CAAC;QAED,SAAS;YACP,OAAO,KAAK,CAAC,MAAM,CAAC;QACtB,CAAC;QAED,QAAQ;YACN,OAAO,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QACnC,CAAC;KACF,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport { AbortError } from \"@azure/abort-controller\";\nimport type { CancelOnProgress, OperationState, SimplePollerLike } from \"@azure/core-lro\";\nimport type { PolledOperationOptions, TestProfileRunCompletionPoller } from \"./models.js\";\nimport type { AzureLoadTestingClient } from \"./clientDefinitions.js\";\nimport type {\n  TestProfileRunAdministrationCreateOrUpdateTestProfileRun200Response,\n  TestProfileRunAdministrationCreateOrUpdateTestProfileRun201Response,\n  TestProfileRunAdministrationGetTestProfileRun200Response,\n} from \"./responses.js\";\nimport { isUnexpected } from \"./isUnexpected.js\";\nimport { sleep, isTestProfileRunInProgress } from \"./util/LROUtil.js\";\n\n/**\n * Creates a poller to poll for test run status.\n * @param client - The Load Testing client.\n * @param options - The operation options.\n * @returns A poller which can be called to poll until completion of the job.\n */\nexport async function getTestProfileRunCompletionPoller(\n  client: AzureLoadTestingClient,\n  createTestProfileRunResponse:\n    | TestProfileRunAdministrationCreateOrUpdateTestProfileRun200Response\n    | TestProfileRunAdministrationCreateOrUpdateTestProfileRun201Response,\n  polledOperationOptions: PolledOperationOptions = {},\n): Promise<TestProfileRunCompletionPoller> {\n  type Handler = (\n    state: OperationState<TestProfileRunAdministrationGetTestProfileRun200Response>,\n  ) => void;\n\n  const state: OperationState<TestProfileRunAdministrationGetTestProfileRun200Response> = {\n    status: \"notStarted\",\n  };\n\n  const progressCallbacks = new Map<symbol, Handler>();\n  const processProgressCallbacks = async (): Promise<void> =>\n    progressCallbacks.forEach((h) => h(state));\n  let resultPromise: Promise<TestProfileRunAdministrationGetTestProfileRun200Response> | undefined;\n  let cancelJob: (() => void) | undefined;\n  const abortController = new AbortController();\n  const currentPollIntervalInMs = polledOperationOptions.updateIntervalInMs ?? 10000;\n  const testProfileRunId = createTestProfileRunResponse.body.testProfileRunId;\n\n  const poller: SimplePollerLike<\n    OperationState<TestProfileRunAdministrationGetTestProfileRun200Response>,\n    TestProfileRunAdministrationGetTestProfileRun200Response\n  > = {\n    async poll(options?: { abortSignal?: AbortSignalLike }): Promise<void> {\n      if (options?.abortSignal?.aborted) {\n        throw new AbortError(\"The polling was aborted.\");\n      }\n\n      if (testProfileRunId) {\n        const getTestProfileRunResult = await client\n          .path(\"/test-profile-runs/{testProfileRunId}\", testProfileRunId)\n          .get();\n        if (isUnexpected(getTestProfileRunResult)) {\n          state.status = \"failed\";\n          state.error = new Error(getTestProfileRunResult.body.error.message);\n          return;\n        }\n\n        if (getTestProfileRunResult.body.status === \"FAILED\") {\n          state.status = \"failed\";\n          state.error = new Error(getTestProfileRunResult.body.status);\n        }\n\n        if (getTestProfileRunResult.body.status === \"CANCELLED\") {\n          state.status = \"canceled\";\n        }\n\n        if (getTestProfileRunResult.body.status === \"DONE\") {\n          state.status = \"succeeded\";\n        }\n\n        if (isTestProfileRunInProgress(getTestProfileRunResult.body)) {\n          state.status = \"running\";\n        }\n        state.result = getTestProfileRunResult;\n        await processProgressCallbacks();\n      }\n    },\n\n    pollUntilDone(pollOptions?: {\n      abortSignal?: AbortSignalLike;\n    }): Promise<TestProfileRunAdministrationGetTestProfileRun200Response> {\n      return (resultPromise ??= (async () => {\n        const { abortSignal: inputAbortSignal } = pollOptions || {};\n        // In the future we can use AbortSignal.any() instead\n        function abortListener(): void {\n          abortController.abort();\n        }\n        const abortSignal = abortController.signal;\n        if (inputAbortSignal?.aborted) {\n          abortController.abort();\n        } else if (!abortSignal.aborted) {\n          inputAbortSignal?.addEventListener(\"abort\", abortListener, { once: true });\n        }\n\n        try {\n          if (!poller.isDone()) {\n            await poller.poll({ abortSignal });\n            while (!poller.isDone()) {\n              const delay = sleep(currentPollIntervalInMs, abortSignal);\n              cancelJob = () => abortController.abort();\n              await delay;\n              await poller.poll({ abortSignal });\n            }\n          }\n        } finally {\n          inputAbortSignal?.removeEventListener(\"abort\", abortListener);\n        }\n        switch (state.status) {\n          case \"succeeded\":\n          case \"failed\":\n          case \"canceled\": {\n            return poller.getResult() as TestProfileRunAdministrationGetTestProfileRun200Response;\n          }\n          case \"notStarted\":\n          case \"running\": {\n            // Unreachable\n            throw new Error(`polling completed without succeeding or failing`);\n          }\n        }\n      })().finally(() => {\n        resultPromise = undefined;\n      }));\n    },\n\n    onProgress(\n      callback: (\n        state: OperationState<TestProfileRunAdministrationGetTestProfileRun200Response>,\n      ) => void,\n    ): CancelOnProgress {\n      const s = Symbol();\n      progressCallbacks.set(s, callback);\n\n      return () => progressCallbacks.delete(s);\n    },\n\n    isDone(): boolean {\n      return [\"succeeded\", \"failed\", \"canceled\"].includes(state.status);\n    },\n\n    stopPolling(): void {\n      abortController.abort();\n      cancelJob?.();\n    },\n\n    isStopped(): boolean {\n      return resultPromise === undefined;\n    },\n\n    getOperationState(): OperationState<TestProfileRunAdministrationGetTestProfileRun200Response> {\n      return state;\n    },\n\n    getResult(): TestProfileRunAdministrationGetTestProfileRun200Response | undefined {\n      return state.result;\n    },\n\n    toString() {\n      return JSON.stringify({ state });\n    },\n  };\n\n  return poller;\n}\n"]}
+{
+  "version": 3,
+  "sources": ["/mnt/vss/_work/1/s/sdk/loadtesting/load-testing-rest/src/getTestProfileRunCompletionPoller.ts"],
+  "sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport { AbortError } from \"@azure/abort-controller\";\nimport type { CancelOnProgress, OperationState } from \"@azure/core-lro\";\nimport type { SimplePollerLike } from \"./pollingHelper.js\";\nimport type { PolledOperationOptions, TestProfileRunCompletionPoller } from \"./models.js\";\nimport type { AzureLoadTestingClient } from \"./clientDefinitions.js\";\nimport type {\n  TestProfileRunAdministrationCreateOrUpdateTestProfileRun200Response,\n  TestProfileRunAdministrationCreateOrUpdateTestProfileRun201Response,\n  TestProfileRunAdministrationGetTestProfileRun200Response,\n} from \"./responses.js\";\nimport { isUnexpected } from \"./isUnexpected.js\";\nimport { sleep, isTestProfileRunInProgress } from \"./util/LROUtil.js\";\n\n/**\n * Creates a poller to poll for test run status.\n * @param client - The Load Testing client.\n * @param options - The operation options.\n * @returns A poller which can be called to poll until completion of the job.\n */\nexport async function getTestProfileRunCompletionPoller(\n  client: AzureLoadTestingClient,\n  createTestProfileRunResponse:\n    | TestProfileRunAdministrationCreateOrUpdateTestProfileRun200Response\n    | TestProfileRunAdministrationCreateOrUpdateTestProfileRun201Response,\n  polledOperationOptions: PolledOperationOptions = {},\n): Promise<TestProfileRunCompletionPoller> {\n  type Handler = (\n    state: OperationState<TestProfileRunAdministrationGetTestProfileRun200Response>,\n  ) => void;\n\n  const state: OperationState<TestProfileRunAdministrationGetTestProfileRun200Response> = {\n    status: \"notStarted\",\n  };\n\n  const progressCallbacks = new Map<symbol, Handler>();\n  const processProgressCallbacks = async (): Promise<void> =>\n    progressCallbacks.forEach((h) => h(state));\n  let resultPromise: Promise<TestProfileRunAdministrationGetTestProfileRun200Response> | undefined;\n  let cancelJob: (() => void) | undefined;\n  const abortController = new AbortController();\n  const currentPollIntervalInMs = polledOperationOptions.updateIntervalInMs ?? 10000;\n  const testProfileRunId = createTestProfileRunResponse.body.testProfileRunId;\n\n  const poller: SimplePollerLike<\n    OperationState<TestProfileRunAdministrationGetTestProfileRun200Response>,\n    TestProfileRunAdministrationGetTestProfileRun200Response\n  > = {\n    async poll(options?: {\n      abortSignal?: AbortSignalLike;\n    }): Promise<OperationState<TestProfileRunAdministrationGetTestProfileRun200Response>> {\n      if (options?.abortSignal?.aborted) {\n        throw new AbortError(\"The polling was aborted.\");\n      }\n\n      if (testProfileRunId) {\n        const getTestProfileRunResult = await client\n          .path(\"/test-profile-runs/{testProfileRunId}\", testProfileRunId)\n          .get();\n        if (isUnexpected(getTestProfileRunResult)) {\n          state.status = \"failed\";\n          state.error = new Error(getTestProfileRunResult.body.error.message);\n          return state;\n        }\n\n        if (getTestProfileRunResult.body.status === \"FAILED\") {\n          state.status = \"failed\";\n          state.error = new Error(getTestProfileRunResult.body.status);\n        }\n\n        if (getTestProfileRunResult.body.status === \"CANCELLED\") {\n          state.status = \"canceled\";\n        }\n\n        if (getTestProfileRunResult.body.status === \"DONE\") {\n          state.status = \"succeeded\";\n        }\n\n        if (isTestProfileRunInProgress(getTestProfileRunResult.body)) {\n          state.status = \"running\";\n        }\n        state.result = getTestProfileRunResult;\n        await processProgressCallbacks();\n      }\n\n      return state;\n    },\n\n    pollUntilDone(pollOptions?: {\n      abortSignal?: AbortSignalLike;\n    }): Promise<TestProfileRunAdministrationGetTestProfileRun200Response> {\n      return (resultPromise ??= (async () => {\n        const { abortSignal: inputAbortSignal } = pollOptions || {};\n        // In the future we can use AbortSignal.any() instead\n        function abortListener(): void {\n          abortController.abort();\n        }\n        const abortSignal = abortController.signal;\n        if (inputAbortSignal?.aborted) {\n          abortController.abort();\n        } else if (!abortSignal.aborted) {\n          inputAbortSignal?.addEventListener(\"abort\", abortListener, { once: true });\n        }\n\n        try {\n          if (!poller.isDone()) {\n            await poller.poll({ abortSignal });\n            while (!poller.isDone()) {\n              const delay = sleep(currentPollIntervalInMs, abortSignal);\n              cancelJob = () => abortController.abort();\n              await delay;\n              await poller.poll({ abortSignal });\n            }\n          }\n        } finally {\n          inputAbortSignal?.removeEventListener(\"abort\", abortListener);\n        }\n        switch (state.status) {\n          case \"succeeded\":\n          case \"failed\":\n          case \"canceled\": {\n            return poller.getResult() as TestProfileRunAdministrationGetTestProfileRun200Response;\n          }\n          case \"notStarted\":\n          case \"running\": {\n            // Unreachable\n            throw new Error(`polling completed without succeeding or failing`);\n          }\n        }\n      })().finally(() => {\n        resultPromise = undefined;\n      }));\n    },\n\n    onProgress(\n      callback: (\n        state: OperationState<TestProfileRunAdministrationGetTestProfileRun200Response>,\n      ) => void,\n    ): CancelOnProgress {\n      const s = Symbol();\n      progressCallbacks.set(s, callback);\n\n      return () => progressCallbacks.delete(s);\n    },\n\n    isDone(): boolean {\n      return [\"succeeded\", \"failed\", \"canceled\"].includes(state.status);\n    },\n\n    stopPolling(): void {\n      abortController.abort();\n      cancelJob?.();\n    },\n\n    isStopped(): boolean {\n      return resultPromise === undefined;\n    },\n\n    getOperationState(): OperationState<TestProfileRunAdministrationGetTestProfileRun200Response> {\n      return state;\n    },\n\n    getResult(): TestProfileRunAdministrationGetTestProfileRun200Response | undefined {\n      return state.result;\n    },\n\n    toString() {\n      return JSON.stringify({ state });\n    },\n\n    async serialize(): Promise<string> {\n      return JSON.stringify({ state });\n    },\n\n    async submitted(): Promise<void> {\n      // No-op: the test profile run is a custom poller\n    },\n  };\n\n  return poller;\n}\n"],
+  "mappings": "AAIA,SAAS,kBAAkB;AAU3B,SAAS,oBAAoB;AAC7B,SAAS,OAAO,kCAAkC;AAQlD,eAAsB,kCACpB,QACA,8BAGA,yBAAiD,CAAC,GACT;AAKzC,QAAM,QAAkF;AAAA,IACtF,QAAQ;AAAA,EACV;AAEA,QAAM,oBAAoB,oBAAI,IAAqB;AACnD,QAAM,2BAA2B,YAC/B,kBAAkB,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC;AAC3C,MAAI;AACJ,MAAI;AACJ,QAAM,kBAAkB,IAAI,gBAAgB;AAC5C,QAAM,0BAA0B,uBAAuB,sBAAsB;AAC7E,QAAM,mBAAmB,6BAA6B,KAAK;AAE3D,QAAM,SAGF;AAAA,IACF,MAAM,KAAK,SAE2E;AACpF,UAAI,SAAS,aAAa,SAAS;AACjC,cAAM,IAAI,WAAW,0BAA0B;AAAA,MACjD;AAEA,UAAI,kBAAkB;AACpB,cAAM,0BAA0B,MAAM,OACnC,KAAK,yCAAyC,gBAAgB,EAC9D,IAAI;AACP,YAAI,aAAa,uBAAuB,GAAG;AACzC,gBAAM,SAAS;AACf,gBAAM,QAAQ,IAAI,MAAM,wBAAwB,KAAK,MAAM,OAAO;AAClE,iBAAO;AAAA,QACT;AAEA,YAAI,wBAAwB,KAAK,WAAW,UAAU;AACpD,gBAAM,SAAS;AACf,gBAAM,QAAQ,IAAI,MAAM,wBAAwB,KAAK,MAAM;AAAA,QAC7D;AAEA,YAAI,wBAAwB,KAAK,WAAW,aAAa;AACvD,gBAAM,SAAS;AAAA,QACjB;AAEA,YAAI,wBAAwB,KAAK,WAAW,QAAQ;AAClD,gBAAM,SAAS;AAAA,QACjB;AAEA,YAAI,2BAA2B,wBAAwB,IAAI,GAAG;AAC5D,gBAAM,SAAS;AAAA,QACjB;AACA,cAAM,SAAS;AACf,cAAM,yBAAyB;AAAA,MACjC;AAEA,aAAO;AAAA,IACT;AAAA,IAEA,cAAc,aAEwD;AACpE,aAAQ,mBAAmB,YAAY;AACrC,cAAM,EAAE,aAAa,iBAAiB,IAAI,eAAe,CAAC;AAE1D,iBAAS,gBAAsB;AAC7B,0BAAgB,MAAM;AAAA,QACxB;AACA,cAAM,cAAc,gBAAgB;AACpC,YAAI,kBAAkB,SAAS;AAC7B,0BAAgB,MAAM;AAAA,QACxB,WAAW,CAAC,YAAY,SAAS;AAC/B,4BAAkB,iBAAiB,SAAS,eAAe,EAAE,MAAM,KAAK,CAAC;AAAA,QAC3E;AAEA,YAAI;AACF,cAAI,CAAC,OAAO,OAAO,GAAG;AACpB,kBAAM,OAAO,KAAK,EAAE,YAAY,CAAC;AACjC,mBAAO,CAAC,OAAO,OAAO,GAAG;AACvB,oBAAM,QAAQ,MAAM,yBAAyB,WAAW;AACxD,0BAAY,MAAM,gBAAgB,MAAM;AACxC,oBAAM;AACN,oBAAM,OAAO,KAAK,EAAE,YAAY,CAAC;AAAA,YACnC;AAAA,UACF;AAAA,QACF,UAAE;AACA,4BAAkB,oBAAoB,SAAS,aAAa;AAAA,QAC9D;AACA,gBAAQ,MAAM,QAAQ;AAAA,UACpB,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK,YAAY;AACf,mBAAO,OAAO,UAAU;AAAA,UAC1B;AAAA,UACA,KAAK;AAAA,UACL,KAAK,WAAW;AAEd,kBAAM,IAAI,MAAM,iDAAiD;AAAA,UACnE;AAAA,QACF;AAAA,MACF,GAAG,EAAE,QAAQ,MAAM;AACjB,wBAAgB;AAAA,MAClB,CAAC;AAAA,IACH;AAAA,IAEA,WACE,UAGkB;AAClB,YAAM,IAAI,uBAAO;AACjB,wBAAkB,IAAI,GAAG,QAAQ;AAEjC,aAAO,MAAM,kBAAkB,OAAO,CAAC;AAAA,IACzC;AAAA,IAEA,SAAkB;AAChB,aAAO,CAAC,aAAa,UAAU,UAAU,EAAE,SAAS,MAAM,MAAM;AAAA,IAClE;AAAA,IAEA,cAAoB;AAClB,sBAAgB,MAAM;AACtB,kBAAY;AAAA,IACd;AAAA,IAEA,YAAqB;AACnB,aAAO,kBAAkB;AAAA,IAC3B;AAAA,IAEA,oBAA8F;AAC5F,aAAO;AAAA,IACT;AAAA,IAEA,YAAkF;AAChF,aAAO,MAAM;AAAA,IACf;AAAA,IAEA,WAAW;AACT,aAAO,KAAK,UAAU,EAAE,MAAM,CAAC;AAAA,IACjC;AAAA,IAEA,MAAM,YAA6B;AACjC,aAAO,KAAK,UAAU,EAAE,MAAM,CAAC;AAAA,IACjC;AAAA,IAEA,MAAM,YAA2B;AAAA,IAEjC;AAAA,EACF;AAEA,SAAO;AACT;",
+  "names": []
+}

package/dist/esm/getTestRunCompletionPoller.js

@@ -1,125 +1,119 @@
-// Copyright (c) Microsoft Corporation.
-// Licensed under the MIT License.
 import { AbortError } from "@azure/abort-controller";
 import { isUnexpected } from "./isUnexpected.js";
 import { sleep, isTestRunInProgress } from "./util/LROUtil.js";
-/**
- * Creates a poller to poll for test run status.
- * @param client - The Load Testing client.
- * @param options - The operation options.
- * @returns A poller which can be called to poll until completion of the job.
- */
-export async function getTestRunCompletionPoller(client, createTestRunResponse, polledOperationOptions = {}) {
-    var _a;
-    const state = {
-        status: "notStarted",
-    };
-    const progressCallbacks = new Map();
-    const processProgressCallbacks = async () => progressCallbacks.forEach((h) => h(state));
-    let resultPromise;
-    let cancelJob;
-    const abortController = new AbortController();
-    const currentPollIntervalInMs = (_a = polledOperationOptions.updateIntervalInMs) !== null && _a !== void 0 ? _a : 2000;
-    const testRunId = createTestRunResponse.body.testRunId;
-    const poller = {
-        async poll(options) {
-            var _a;
-            if ((_a = options === null || options === void 0 ? void 0 : options.abortSignal) === null || _a === void 0 ? void 0 : _a.aborted) {
-                throw new AbortError("The polling was aborted.");
+async function getTestRunCompletionPoller(client, createTestRunResponse, polledOperationOptions = {}) {
+  const state = {
+    status: "notStarted"
+  };
+  const progressCallbacks = /* @__PURE__ */ new Map();
+  const processProgressCallbacks = async () => progressCallbacks.forEach((h) => h(state));
+  let resultPromise;
+  let cancelJob;
+  const abortController = new AbortController();
+  const currentPollIntervalInMs = polledOperationOptions.updateIntervalInMs ?? 2e3;
+  const testRunId = createTestRunResponse.body.testRunId;
+  const poller = {
+    async poll(options) {
+      if (options?.abortSignal?.aborted) {
+        throw new AbortError("The polling was aborted.");
+      }
+      if (testRunId) {
+        const getTestRunResult = await client.path("/test-runs/{testRunId}", testRunId).get();
+        if (isUnexpected(getTestRunResult)) {
+          state.status = "failed";
+          state.error = new Error(getTestRunResult.body.error.message);
+          return state;
+        }
+        if (getTestRunResult.body.status === "FAILED") {
+          state.status = "failed";
+          state.error = new Error(getTestRunResult.body.status);
+        }
+        if (getTestRunResult.body.status === "CANCELLED") {
+          state.status = "canceled";
+        }
+        if (getTestRunResult.body.status === "DONE") {
+          state.status = "succeeded";
+        }
+        if (isTestRunInProgress(getTestRunResult.body)) {
+          state.status = "running";
+        }
+        state.result = getTestRunResult;
+        await processProgressCallbacks();
+      }
+      return state;
+    },
+    pollUntilDone(pollOptions) {
+      return resultPromise ??= (async () => {
+        const { abortSignal: inputAbortSignal } = pollOptions || {};
+        function abortListener() {
+          abortController.abort();
+        }
+        const abortSignal = abortController.signal;
+        if (inputAbortSignal?.aborted) {
+          abortController.abort();
+        } else if (!abortSignal.aborted) {
+          inputAbortSignal?.addEventListener("abort", abortListener, { once: true });
+        }
+        try {
+          if (!poller.isDone()) {
+            await poller.poll({ abortSignal });
+            while (!poller.isDone()) {
+              const delay = sleep(currentPollIntervalInMs, abortSignal);
+              cancelJob = () => abortController.abort();
+              await delay;
+              await poller.poll({ abortSignal });
             }
-            if (testRunId) {
-                const getTestRunResult = await client.path("/test-runs/{testRunId}", testRunId).get();
-                if (isUnexpected(getTestRunResult)) {
-                    state.status = "failed";
-                    state.error = new Error(getTestRunResult.body.error.message);
-                    return;
-                }
-                if (getTestRunResult.body.status === "FAILED") {
-                    state.status = "failed";
-                    state.error = new Error(getTestRunResult.body.status);
-                }
-                if (getTestRunResult.body.status === "CANCELLED") {
-                    state.status = "canceled";
-                }
-                if (getTestRunResult.body.status === "DONE") {
-                    state.status = "succeeded";
-                }
-                if (isTestRunInProgress(getTestRunResult.body)) {
-                    state.status = "running";
-                }
-                state.result = getTestRunResult;
-                await processProgressCallbacks();
-            }
-        },
-        pollUntilDone(pollOptions) {
-            return (resultPromise !== null && resultPromise !== void 0 ? resultPromise : (resultPromise = (async () => {
-                const { abortSignal: inputAbortSignal } = pollOptions || {};
-                // In the future we can use AbortSignal.any() instead
-                function abortListener() {
-                    abortController.abort();
-                }
-                const abortSignal = abortController.signal;
-                if (inputAbortSignal === null || inputAbortSignal === void 0 ? void 0 : inputAbortSignal.aborted) {
-                    abortController.abort();
-                }
-                else if (!abortSignal.aborted) {
-                    inputAbortSignal === null || inputAbortSignal === void 0 ? void 0 : inputAbortSignal.addEventListener("abort", abortListener, { once: true });
-                }
-                try {
-                    if (!poller.isDone()) {
-                        await poller.poll({ abortSignal });
-                        while (!poller.isDone()) {
-                            const delay = sleep(currentPollIntervalInMs, abortSignal);
-                            cancelJob = () => abortController.abort();
-                            await delay;
-                            await poller.poll({ abortSignal });
-                        }
-                    }
-                }
-                finally {
-                    inputAbortSignal === null || inputAbortSignal === void 0 ? void 0 : inputAbortSignal.removeEventListener("abort", abortListener);
-                }
-                switch (state.status) {
-                    case "succeeded":
-                    case "failed":
-                    case "canceled": {
-                        return poller.getResult();
-                    }
-                    case "notStarted":
-                    case "running": {
-                        // Unreachable
-                        throw new Error(`polling completed without succeeding or failing`);
-                    }
-                }
-            })().finally(() => {
-                resultPromise = undefined;
-            })));
-        },
-        onProgress(callback) {
-            const s = Symbol();
-            progressCallbacks.set(s, callback);
-            return () => progressCallbacks.delete(s);
-        },
-        isDone() {
-            return ["succeeded", "failed", "canceled"].includes(state.status);
-        },
-        stopPolling() {
-            abortController.abort();
-            cancelJob === null || cancelJob === void 0 ? void 0 : cancelJob();
-        },
-        isStopped() {
-            return resultPromise === undefined;
-        },
-        getOperationState() {
-            return state;
-        },
-        getResult() {
-            return state.result;
-        },
-        toString() {
-            return JSON.stringify({ state });
-        },
-    };
-    return poller;
+          }
+        } finally {
+          inputAbortSignal?.removeEventListener("abort", abortListener);
+        }
+        switch (state.status) {
+          case "succeeded":
+          case "failed":
+          case "canceled": {
+            return poller.getResult();
+          }
+          case "notStarted":
+          case "running": {
+            throw new Error(`polling completed without succeeding or failing`);
+          }
+        }
+      })().finally(() => {
+        resultPromise = void 0;
+      });
+    },
+    onProgress(callback) {
+      const s = /* @__PURE__ */ Symbol();
+      progressCallbacks.set(s, callback);
+      return () => progressCallbacks.delete(s);
+    },
+    isDone() {
+      return ["succeeded", "failed", "canceled"].includes(state.status);
+    },
+    stopPolling() {
+      abortController.abort();
+      cancelJob?.();
+    },
+    isStopped() {
+      return resultPromise === void 0;
+    },
+    getOperationState() {
+      return state;
+    },
+    getResult() {
+      return state.result;
+    },
+    toString() {
+      return JSON.stringify({ state });
+    },
+    async serialize() {
+      return JSON.stringify({ state });
+    },
+    async submitted() {
+    }
+  };
+  return poller;
 }
-//# sourceMappingURL=getTestRunCompletionPoller.js.map
+export {
+  getTestRunCompletionPoller
+};

package/dist/esm/getTestRunCompletionPoller.js.map

@@ -1 +1,7 @@
-{"version":3,"file":"getTestRunCompletionPoller.js","sourceRoot":"","sources":["../../src/getTestRunCompletionPoller.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AASrD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,KAAK,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAE/D;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,MAA8B,EAC9B,qBAE+C,EAC/C,yBAAiD,EAAE;;IAInD,MAAM,KAAK,GAAqD;QAC9D,MAAM,EAAE,YAAY;KACrB,CAAC;IAEF,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAmB,CAAC;IACrD,MAAM,wBAAwB,GAAG,KAAK,IAAmB,EAAE,CACzD,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IAC7C,IAAI,aAAoE,CAAC;IACzE,IAAI,SAAmC,CAAC;IACxC,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;IAC9C,MAAM,uBAAuB,GAAG,MAAA,sBAAsB,CAAC,kBAAkB,mCAAI,IAAI,CAAC;IAClF,MAAM,SAAS,GAAG,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC;IAEvD,MAAM,MAAM,GAGR;QACF,KAAK,CAAC,IAAI,CAAC,OAA2C;;YACpD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW,0CAAE,OAAO,EAAE,CAAC;gBAClC,MAAM,IAAI,UAAU,CAAC,0BAA0B,CAAC,CAAC;YACnD,CAAC;YAED,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,gBAAgB,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE,SAAS,CAAC,CAAC,GAAG,EAAE,CAAC;gBACtF,IAAI,YAAY,CAAC,gBAAgB,CAAC,EAAE,CAAC;oBACnC,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAC;oBACxB,KAAK,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBAC7D,OAAO;gBACT,CAAC;gBAED,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;oBAC9C,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAC;oBACxB,KAAK,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACxD,CAAC;gBAED,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;oBACjD,KAAK,CAAC,MAAM,GAAG,UAAU,CAAC;gBAC5B,CAAC;gBAED,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;oBAC5C,KAAK,CAAC,MAAM,GAAG,WAAW,CAAC;gBAC7B,CAAC;gBAED,IAAI,mBAAmB,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC/C,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC;gBAC3B,CAAC;gBACD,KAAK,CAAC,MAAM,GAAG,gBAAgB,CAAC;gBAChC,MAAM,wBAAwB,EAAE,CAAC;YACnC,CAAC;QACH,CAAC;QAED,aAAa,CAAC,WAEb;YACC,OAAO,CAAC,aAAa,aAAb,aAAa,cAAb,aAAa,IAAb,aAAa,GAAK,CAAC,KAAK,IAAI,EAAE;gBACpC,MAAM,EAAE,WAAW,EAAE,gBAAgB,EAAE,GAAG,WAAW,IAAI,EAAE,CAAC;gBAC5D,qDAAqD;gBACrD,SAAS,aAAa;oBACpB,eAAe,CAAC,KAAK,EAAE,CAAC;gBAC1B,CAAC;gBACD,MAAM,WAAW,GAAG,eAAe,CAAC,MAAM,CAAC;gBAC3C,IAAI,gBAAgB,aAAhB,gBAAgB,uBAAhB,gBAAgB,CAAE,OAAO,EAAE,CAAC;oBAC9B,eAAe,CAAC,KAAK,EAAE,CAAC;gBAC1B,CAAC;qBAAM,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;oBAChC,gBAAgB,aAAhB,gBAAgB,uBAAhB,gBAAgB,CAAE,gBAAgB,CAAC,OAAO,EAAE,aAAa,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC7E,CAAC;gBAED,IAAI,CAAC;oBACH,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;wBACrB,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC;wBACnC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;4BACxB,MAAM,KAAK,GAAG,KAAK,CAAC,uBAAuB,EAAE,WAAW,CAAC,CAAC;4BAC1D,SAAS,GAAG,GAAG,EAAE,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;4BAC1C,MAAM,KAAK,CAAC;4BACZ,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC;wBACrC,CAAC;oBACH,CAAC;gBACH,CAAC;wBAAS,CAAC;oBACT,gBAAgB,aAAhB,gBAAgB,uBAAhB,gBAAgB,CAAE,mBAAmB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;gBAChE,CAAC;gBACD,QAAQ,KAAK,CAAC,MAAM,EAAE,CAAC;oBACrB,KAAK,WAAW,CAAC;oBACjB,KAAK,QAAQ,CAAC;oBACd,KAAK,UAAU,CAAC,CAAC,CAAC;wBAChB,OAAO,MAAM,CAAC,SAAS,EAAsC,CAAC;oBAChE,CAAC;oBACD,KAAK,YAAY,CAAC;oBAClB,KAAK,SAAS,CAAC,CAAC,CAAC;wBACf,cAAc;wBACd,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;oBACrE,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBAChB,aAAa,GAAG,SAAS,CAAC;YAC5B,CAAC,CAAC,EAAC,CAAC;QACN,CAAC;QAED,UAAU,CACR,QAA2E;YAE3E,MAAM,CAAC,GAAG,MAAM,EAAE,CAAC;YACnB,iBAAiB,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;YAEnC,OAAO,GAAG,EAAE,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM;YACJ,OAAO,CAAC,WAAW,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACpE,CAAC;QAED,WAAW;YACT,eAAe,CAAC,KAAK,EAAE,CAAC;YACxB,SAAS,aAAT,SAAS,uBAAT,SAAS,EAAI,CAAC;QAChB,CAAC;QAED,SAAS;YACP,OAAO,aAAa,KAAK,SAAS,CAAC;QACrC,CAAC;QAED,iBAAiB;YACf,OAAO,KAAK,CAAC;QACf,CAAC;QAED,SAAS;YACP,OAAO,KAAK,CAAC,MAAM,CAAC;QACtB,CAAC;QAED,QAAQ;YACN,OAAO,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QACnC,CAAC;KACF,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport { AbortError } from \"@azure/abort-controller\";\nimport type { CancelOnProgress, OperationState, SimplePollerLike } from \"@azure/core-lro\";\nimport type { TestRunCompletionPoller, PolledOperationOptions } from \"./models.js\";\nimport type { AzureLoadTestingClient } from \"./clientDefinitions.js\";\nimport type {\n  LoadTestRunCreateOrUpdateTestRun200Response,\n  LoadTestRunCreateOrUpdateTestRun201Response,\n  LoadTestRunGetTestRun200Response,\n} from \"./responses.js\";\nimport { isUnexpected } from \"./isUnexpected.js\";\nimport { sleep, isTestRunInProgress } from \"./util/LROUtil.js\";\n\n/**\n * Creates a poller to poll for test run status.\n * @param client - The Load Testing client.\n * @param options - The operation options.\n * @returns A poller which can be called to poll until completion of the job.\n */\nexport async function getTestRunCompletionPoller(\n  client: AzureLoadTestingClient,\n  createTestRunResponse:\n    | LoadTestRunCreateOrUpdateTestRun200Response\n    | LoadTestRunCreateOrUpdateTestRun201Response,\n  polledOperationOptions: PolledOperationOptions = {},\n): Promise<TestRunCompletionPoller> {\n  type Handler = (state: OperationState<LoadTestRunGetTestRun200Response>) => void;\n\n  const state: OperationState<LoadTestRunGetTestRun200Response> = {\n    status: \"notStarted\",\n  };\n\n  const progressCallbacks = new Map<symbol, Handler>();\n  const processProgressCallbacks = async (): Promise<void> =>\n    progressCallbacks.forEach((h) => h(state));\n  let resultPromise: Promise<LoadTestRunGetTestRun200Response> | undefined;\n  let cancelJob: (() => void) | undefined;\n  const abortController = new AbortController();\n  const currentPollIntervalInMs = polledOperationOptions.updateIntervalInMs ?? 2000;\n  const testRunId = createTestRunResponse.body.testRunId;\n\n  const poller: SimplePollerLike<\n    OperationState<LoadTestRunGetTestRun200Response>,\n    LoadTestRunGetTestRun200Response\n  > = {\n    async poll(options?: { abortSignal?: AbortSignalLike }): Promise<void> {\n      if (options?.abortSignal?.aborted) {\n        throw new AbortError(\"The polling was aborted.\");\n      }\n\n      if (testRunId) {\n        const getTestRunResult = await client.path(\"/test-runs/{testRunId}\", testRunId).get();\n        if (isUnexpected(getTestRunResult)) {\n          state.status = \"failed\";\n          state.error = new Error(getTestRunResult.body.error.message);\n          return;\n        }\n\n        if (getTestRunResult.body.status === \"FAILED\") {\n          state.status = \"failed\";\n          state.error = new Error(getTestRunResult.body.status);\n        }\n\n        if (getTestRunResult.body.status === \"CANCELLED\") {\n          state.status = \"canceled\";\n        }\n\n        if (getTestRunResult.body.status === \"DONE\") {\n          state.status = \"succeeded\";\n        }\n\n        if (isTestRunInProgress(getTestRunResult.body)) {\n          state.status = \"running\";\n        }\n        state.result = getTestRunResult;\n        await processProgressCallbacks();\n      }\n    },\n\n    pollUntilDone(pollOptions?: {\n      abortSignal?: AbortSignalLike;\n    }): Promise<LoadTestRunGetTestRun200Response> {\n      return (resultPromise ??= (async () => {\n        const { abortSignal: inputAbortSignal } = pollOptions || {};\n        // In the future we can use AbortSignal.any() instead\n        function abortListener(): void {\n          abortController.abort();\n        }\n        const abortSignal = abortController.signal;\n        if (inputAbortSignal?.aborted) {\n          abortController.abort();\n        } else if (!abortSignal.aborted) {\n          inputAbortSignal?.addEventListener(\"abort\", abortListener, { once: true });\n        }\n\n        try {\n          if (!poller.isDone()) {\n            await poller.poll({ abortSignal });\n            while (!poller.isDone()) {\n              const delay = sleep(currentPollIntervalInMs, abortSignal);\n              cancelJob = () => abortController.abort();\n              await delay;\n              await poller.poll({ abortSignal });\n            }\n          }\n        } finally {\n          inputAbortSignal?.removeEventListener(\"abort\", abortListener);\n        }\n        switch (state.status) {\n          case \"succeeded\":\n          case \"failed\":\n          case \"canceled\": {\n            return poller.getResult() as LoadTestRunGetTestRun200Response;\n          }\n          case \"notStarted\":\n          case \"running\": {\n            // Unreachable\n            throw new Error(`polling completed without succeeding or failing`);\n          }\n        }\n      })().finally(() => {\n        resultPromise = undefined;\n      }));\n    },\n\n    onProgress(\n      callback: (state: OperationState<LoadTestRunGetTestRun200Response>) => void,\n    ): CancelOnProgress {\n      const s = Symbol();\n      progressCallbacks.set(s, callback);\n\n      return () => progressCallbacks.delete(s);\n    },\n\n    isDone(): boolean {\n      return [\"succeeded\", \"failed\", \"canceled\"].includes(state.status);\n    },\n\n    stopPolling(): void {\n      abortController.abort();\n      cancelJob?.();\n    },\n\n    isStopped(): boolean {\n      return resultPromise === undefined;\n    },\n\n    getOperationState(): OperationState<LoadTestRunGetTestRun200Response> {\n      return state;\n    },\n\n    getResult(): LoadTestRunGetTestRun200Response | undefined {\n      return state.result;\n    },\n\n    toString() {\n      return JSON.stringify({ state });\n    },\n  };\n\n  return poller;\n}\n"]}
+{
+  "version": 3,
+  "sources": ["/mnt/vss/_work/1/s/sdk/loadtesting/load-testing-rest/src/getTestRunCompletionPoller.ts"],
+  "sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport { AbortError } from \"@azure/abort-controller\";\nimport type { CancelOnProgress, OperationState } from \"@azure/core-lro\";\nimport type { SimplePollerLike } from \"./pollingHelper.js\";\nimport type { TestRunCompletionPoller, PolledOperationOptions } from \"./models.js\";\nimport type { AzureLoadTestingClient } from \"./clientDefinitions.js\";\nimport type {\n  LoadTestRunCreateOrUpdateTestRun200Response,\n  LoadTestRunCreateOrUpdateTestRun201Response,\n  LoadTestRunGetTestRun200Response,\n} from \"./responses.js\";\nimport { isUnexpected } from \"./isUnexpected.js\";\nimport { sleep, isTestRunInProgress } from \"./util/LROUtil.js\";\n\n/**\n * Creates a poller to poll for test run status.\n * @param client - The Load Testing client.\n * @param options - The operation options.\n * @returns A poller which can be called to poll until completion of the job.\n */\nexport async function getTestRunCompletionPoller(\n  client: AzureLoadTestingClient,\n  createTestRunResponse:\n    | LoadTestRunCreateOrUpdateTestRun200Response\n    | LoadTestRunCreateOrUpdateTestRun201Response,\n  polledOperationOptions: PolledOperationOptions = {},\n): Promise<TestRunCompletionPoller> {\n  type Handler = (state: OperationState<LoadTestRunGetTestRun200Response>) => void;\n\n  const state: OperationState<LoadTestRunGetTestRun200Response> = {\n    status: \"notStarted\",\n  };\n\n  const progressCallbacks = new Map<symbol, Handler>();\n  const processProgressCallbacks = async (): Promise<void> =>\n    progressCallbacks.forEach((h) => h(state));\n  let resultPromise: Promise<LoadTestRunGetTestRun200Response> | undefined;\n  let cancelJob: (() => void) | undefined;\n  const abortController = new AbortController();\n  const currentPollIntervalInMs = polledOperationOptions.updateIntervalInMs ?? 2000;\n  const testRunId = createTestRunResponse.body.testRunId;\n\n  const poller: SimplePollerLike<\n    OperationState<LoadTestRunGetTestRun200Response>,\n    LoadTestRunGetTestRun200Response\n  > = {\n    async poll(options?: {\n      abortSignal?: AbortSignalLike;\n    }): Promise<OperationState<LoadTestRunGetTestRun200Response>> {\n      if (options?.abortSignal?.aborted) {\n        throw new AbortError(\"The polling was aborted.\");\n      }\n\n      if (testRunId) {\n        const getTestRunResult = await client.path(\"/test-runs/{testRunId}\", testRunId).get();\n        if (isUnexpected(getTestRunResult)) {\n          state.status = \"failed\";\n          state.error = new Error(getTestRunResult.body.error.message);\n          return state;\n        }\n\n        if (getTestRunResult.body.status === \"FAILED\") {\n          state.status = \"failed\";\n          state.error = new Error(getTestRunResult.body.status);\n        }\n\n        if (getTestRunResult.body.status === \"CANCELLED\") {\n          state.status = \"canceled\";\n        }\n\n        if (getTestRunResult.body.status === \"DONE\") {\n          state.status = \"succeeded\";\n        }\n\n        if (isTestRunInProgress(getTestRunResult.body)) {\n          state.status = \"running\";\n        }\n        state.result = getTestRunResult;\n        await processProgressCallbacks();\n      }\n      return state;\n    },\n\n    pollUntilDone(pollOptions?: {\n      abortSignal?: AbortSignalLike;\n    }): Promise<LoadTestRunGetTestRun200Response> {\n      return (resultPromise ??= (async () => {\n        const { abortSignal: inputAbortSignal } = pollOptions || {};\n        // In the future we can use AbortSignal.any() instead\n        function abortListener(): void {\n          abortController.abort();\n        }\n        const abortSignal = abortController.signal;\n        if (inputAbortSignal?.aborted) {\n          abortController.abort();\n        } else if (!abortSignal.aborted) {\n          inputAbortSignal?.addEventListener(\"abort\", abortListener, { once: true });\n        }\n\n        try {\n          if (!poller.isDone()) {\n            await poller.poll({ abortSignal });\n            while (!poller.isDone()) {\n              const delay = sleep(currentPollIntervalInMs, abortSignal);\n              cancelJob = () => abortController.abort();\n              await delay;\n              await poller.poll({ abortSignal });\n            }\n          }\n        } finally {\n          inputAbortSignal?.removeEventListener(\"abort\", abortListener);\n        }\n        switch (state.status) {\n          case \"succeeded\":\n          case \"failed\":\n          case \"canceled\": {\n            return poller.getResult() as LoadTestRunGetTestRun200Response;\n          }\n          case \"notStarted\":\n          case \"running\": {\n            // Unreachable\n            throw new Error(`polling completed without succeeding or failing`);\n          }\n        }\n      })().finally(() => {\n        resultPromise = undefined;\n      }));\n    },\n\n    onProgress(\n      callback: (state: OperationState<LoadTestRunGetTestRun200Response>) => void,\n    ): CancelOnProgress {\n      const s = Symbol();\n      progressCallbacks.set(s, callback);\n\n      return () => progressCallbacks.delete(s);\n    },\n\n    isDone(): boolean {\n      return [\"succeeded\", \"failed\", \"canceled\"].includes(state.status);\n    },\n\n    stopPolling(): void {\n      abortController.abort();\n      cancelJob?.();\n    },\n\n    isStopped(): boolean {\n      return resultPromise === undefined;\n    },\n\n    getOperationState(): OperationState<LoadTestRunGetTestRun200Response> {\n      return state;\n    },\n\n    getResult(): LoadTestRunGetTestRun200Response | undefined {\n      return state.result;\n    },\n\n    toString() {\n      return JSON.stringify({ state });\n    },\n\n    async serialize(): Promise<string> {\n      return JSON.stringify({ state });\n    },\n\n    async submitted(): Promise<void> {\n      // No-op: the test run is a custom poller\n    },\n  };\n\n  return poller;\n}\n"],
+  "mappings": "AAIA,SAAS,kBAAkB;AAU3B,SAAS,oBAAoB;AAC7B,SAAS,OAAO,2BAA2B;AAQ3C,eAAsB,2BACpB,QACA,uBAGA,yBAAiD,CAAC,GAChB;AAGlC,QAAM,QAA0D;AAAA,IAC9D,QAAQ;AAAA,EACV;AAEA,QAAM,oBAAoB,oBAAI,IAAqB;AACnD,QAAM,2BAA2B,YAC/B,kBAAkB,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC;AAC3C,MAAI;AACJ,MAAI;AACJ,QAAM,kBAAkB,IAAI,gBAAgB;AAC5C,QAAM,0BAA0B,uBAAuB,sBAAsB;AAC7E,QAAM,YAAY,sBAAsB,KAAK;AAE7C,QAAM,SAGF;AAAA,IACF,MAAM,KAAK,SAEmD;AAC5D,UAAI,SAAS,aAAa,SAAS;AACjC,cAAM,IAAI,WAAW,0BAA0B;AAAA,MACjD;AAEA,UAAI,WAAW;AACb,cAAM,mBAAmB,MAAM,OAAO,KAAK,0BAA0B,SAAS,EAAE,IAAI;AACpF,YAAI,aAAa,gBAAgB,GAAG;AAClC,gBAAM,SAAS;AACf,gBAAM,QAAQ,IAAI,MAAM,iBAAiB,KAAK,MAAM,OAAO;AAC3D,iBAAO;AAAA,QACT;AAEA,YAAI,iBAAiB,KAAK,WAAW,UAAU;AAC7C,gBAAM,SAAS;AACf,gBAAM,QAAQ,IAAI,MAAM,iBAAiB,KAAK,MAAM;AAAA,QACtD;AAEA,YAAI,iBAAiB,KAAK,WAAW,aAAa;AAChD,gBAAM,SAAS;AAAA,QACjB;AAEA,YAAI,iBAAiB,KAAK,WAAW,QAAQ;AAC3C,gBAAM,SAAS;AAAA,QACjB;AAEA,YAAI,oBAAoB,iBAAiB,IAAI,GAAG;AAC9C,gBAAM,SAAS;AAAA,QACjB;AACA,cAAM,SAAS;AACf,cAAM,yBAAyB;AAAA,MACjC;AACA,aAAO;AAAA,IACT;AAAA,IAEA,cAAc,aAEgC;AAC5C,aAAQ,mBAAmB,YAAY;AACrC,cAAM,EAAE,aAAa,iBAAiB,IAAI,eAAe,CAAC;AAE1D,iBAAS,gBAAsB;AAC7B,0BAAgB,MAAM;AAAA,QACxB;AACA,cAAM,cAAc,gBAAgB;AACpC,YAAI,kBAAkB,SAAS;AAC7B,0BAAgB,MAAM;AAAA,QACxB,WAAW,CAAC,YAAY,SAAS;AAC/B,4BAAkB,iBAAiB,SAAS,eAAe,EAAE,MAAM,KAAK,CAAC;AAAA,QAC3E;AAEA,YAAI;AACF,cAAI,CAAC,OAAO,OAAO,GAAG;AACpB,kBAAM,OAAO,KAAK,EAAE,YAAY,CAAC;AACjC,mBAAO,CAAC,OAAO,OAAO,GAAG;AACvB,oBAAM,QAAQ,MAAM,yBAAyB,WAAW;AACxD,0BAAY,MAAM,gBAAgB,MAAM;AACxC,oBAAM;AACN,oBAAM,OAAO,KAAK,EAAE,YAAY,CAAC;AAAA,YACnC;AAAA,UACF;AAAA,QACF,UAAE;AACA,4BAAkB,oBAAoB,SAAS,aAAa;AAAA,QAC9D;AACA,gBAAQ,MAAM,QAAQ;AAAA,UACpB,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK,YAAY;AACf,mBAAO,OAAO,UAAU;AAAA,UAC1B;AAAA,UACA,KAAK;AAAA,UACL,KAAK,WAAW;AAEd,kBAAM,IAAI,MAAM,iDAAiD;AAAA,UACnE;AAAA,QACF;AAAA,MACF,GAAG,EAAE,QAAQ,MAAM;AACjB,wBAAgB;AAAA,MAClB,CAAC;AAAA,IACH;AAAA,IAEA,WACE,UACkB;AAClB,YAAM,IAAI,uBAAO;AACjB,wBAAkB,IAAI,GAAG,QAAQ;AAEjC,aAAO,MAAM,kBAAkB,OAAO,CAAC;AAAA,IACzC;AAAA,IAEA,SAAkB;AAChB,aAAO,CAAC,aAAa,UAAU,UAAU,EAAE,SAAS,MAAM,MAAM;AAAA,IAClE;AAAA,IAEA,cAAoB;AAClB,sBAAgB,MAAM;AACtB,kBAAY;AAAA,IACd;AAAA,IAEA,YAAqB;AACnB,aAAO,kBAAkB;AAAA,IAC3B;AAAA,IAEA,oBAAsE;AACpE,aAAO;AAAA,IACT;AAAA,IAEA,YAA0D;AACxD,aAAO,MAAM;AAAA,IACf;AAAA,IAEA,WAAW;AACT,aAAO,KAAK,UAAU,EAAE,MAAM,CAAC;AAAA,IACjC;AAAA,IAEA,MAAM,YAA6B;AACjC,aAAO,KAAK,UAAU,EAAE,MAAM,CAAC;AAAA,IACjC;AAAA,IAEA,MAAM,YAA2B;AAAA,IAEjC;AAAA,EACF;AAEA,SAAO;AACT;",
+  "names": []
+}

package/dist/esm/index.d.ts

@@ -1,12 +1,12 @@
 import AzureLoadTesting from "./azureLoadTesting.js";
 export * from "./azureLoadTesting.js";
-export * from "./parameters.js";
-export * from "./responses.js";
-export * from "./clientDefinitions.js";
+export type * from "./parameters.js";
+export type * from "./responses.js";
+export type * from "./clientDefinitions.js";
 export * from "./isUnexpected.js";
-export * from "./models.js";
-export * from "./outputModels.js";
+export type * from "./models.js";
+export type * from "./outputModels.js";
 export * from "./paginateHelper.js";
-export { getLongRunningPoller } from "./pollingHelper.js";
+export * from "./pollingHelper.js";
 export default AzureLoadTesting;
 //# sourceMappingURL=index.d.ts.map

package/dist/esm/index.js

@@ -1,15 +1,9 @@
-// Copyright (c) Microsoft Corporation.
-// Licensed under the MIT License.
 import AzureLoadTesting from "./azureLoadTesting.js";
 export * from "./azureLoadTesting.js";
-export * from "./parameters.js";
-export * from "./responses.js";
-export * from "./clientDefinitions.js";
 export * from "./isUnexpected.js";
-export * from "./models.js";
-export * from "./outputModels.js";
 export * from "./paginateHelper.js";
-export { getLongRunningPoller } from "./pollingHelper.js";
-// eslint-disable-next-line @azure/azure-sdk/ts-modules-only-named
-export default AzureLoadTesting;
-//# sourceMappingURL=index.js.map
+export * from "./pollingHelper.js";
+var src_default = AzureLoadTesting;
+export {
+  src_default as default
+};