@loadstrike/loadstrike-sdk 1.0.22801 → 1.0.23001

package/README.md

@@ -55,7 +55,7 @@ const result = await LoadStrikeRunner
 
 ## Trace-To-Test Autopilot
 
-Use `LoadStrikeAutopilot.generate(...)` to infer a starter plan from a captured artifact. Check `result.Readiness` and `result.ReadinessFailures` first; call `result.buildScenario()` only when it is `LoadStrikeAutopilotReadiness.Ready`, then execute the scenario through the normal runner with a valid `RunnerKey`.
+Use `await LoadStrikeAutopilot.generate(...)` to infer a starter plan from a captured artifact. Set `Options.RunnerKey` so generation can validate the Trace-To-Test Autopilot entitlement. Check `result.Readiness` and `result.ReadinessFailures` first; call `result.buildScenario()` only when it is `LoadStrikeAutopilotReadiness.Ready`, then execute the scenario through the normal runner with a valid `RunnerKey`.
 
 Use `SecretBindings` to map redaction locations such as `header:Authorization` or `body:$.client_secret` to environment variables, `TrackingSelector` when the selector cannot be inferred, and `EndpointBindings`, `AllowedReplayHosts`, or `BaseUrlRewrite` when a replay target must be bound. Secret values are resolved when the generated scenario runs; they are not written into the generated plan. Any gate satisfied by user setup is omitted from `ReadinessFailures`.
 

package/dist/cjs/autopilot.js

@@ -15,14 +15,16 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 var _LoadStrikeAutopilotResult_httpRequest;
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.LoadStrikeAutopilot = exports.LoadStrikeAutopilotResult = void 0;
+exports.__private = exports.LoadStrikeAutopilot = exports.LoadStrikeAutopilotResult = void 0;
 const node_fs_1 = __importDefault(require("node:fs"));
 const runtime_js_1 = require("./runtime.js");
+const local_js_1 = require("./local.js");
 const autopilot_contracts_js_1 = require("./autopilot-contracts.js");
 const REDACTED = "[REDACTED]";
 const ENV_MARKER_PREFIX = "${LOADSTRIKE_ENV:";
 const ENV_MARKER_SUFFIX = "}";
 const TRACE_TO_TEST_AUTOPILOT_FEATURE = "autopilot.trace_to_test";
+let autopilotLicenseValidationBypassForTests = false;
 const SECRET_KEYS = new Set([
     "authorization",
     "proxy_authorization",
@@ -95,9 +97,10 @@ class LoadStrikeAutopilotResult {
 exports.LoadStrikeAutopilotResult = LoadStrikeAutopilotResult;
 _LoadStrikeAutopilotResult_httpRequest = new WeakMap();
 class LoadStrikeAutopilot {
-    static generate(request) {
+    static async generate(request) {
         const artifact = loadAutopilotArtifact(request);
         const options = request.Options ?? {};
+        await validateGenerationEntitlement(options);
         const result = inferAutopilotResult(artifact, options);
         if (options.IncludePreviewReport) {
             result.PreviewReport = buildPreviewReport(result);
@@ -109,6 +112,57 @@ class LoadStrikeAutopilot {
     }
 }
 exports.LoadStrikeAutopilot = LoadStrikeAutopilot;
+exports.__private = {
+    setAutopilotLicenseValidationBypassForTests(value) {
+        autopilotLicenseValidationBypassForTests = value;
+    }
+};
+async function validateGenerationEntitlement(options) {
+    if (autopilotLicenseValidationBypassForTests) {
+        return;
+    }
+    const licensePayload = {
+        Context: {
+            RunnerKey: options.RunnerKey ?? "",
+            LicenseValidationTimeoutSeconds: options.LicenseValidationTimeoutSeconds,
+            TestSuite: "loadstrike-autopilot",
+            TestName: options.ScenarioName?.trim() || "autopilot-generation"
+        },
+        Scenarios: [
+            {
+                Name: "autopilot-generation",
+                InternalLicenseFeatures: [TRACE_TO_TEST_AUTOPILOT_FEATURE]
+            }
+        ],
+        RunArgs: []
+    };
+    const client = new local_js_1.LoadStrikeLocalClient({
+        licenseValidationTimeoutMs: normalizeLicenseValidationTimeoutMs(options.LicenseValidationTimeoutSeconds)
+    });
+    let session;
+    try {
+        session = await client.acquireLicenseLease(licensePayload);
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        if (message.toLowerCase().includes("runner key is required")) {
+            throw new Error("Runner key is required for Trace-To-Test Autopilot generation. " +
+                "Set Options.RunnerKey before calling LoadStrikeAutopilot.generate(...).");
+        }
+        throw error;
+    }
+    finally {
+        if (session) {
+            await client.releaseLicenseLease(session, licensePayload);
+        }
+    }
+}
+function normalizeLicenseValidationTimeoutMs(value) {
+    if (typeof value !== "number" || !Number.isFinite(value) || value <= 0) {
+        return undefined;
+    }
+    return Math.trunc(value * 1000);
+}
 function loadAutopilotArtifact(request) {
     if (!request || typeof request.Kind !== "string" || !request.Kind.trim()) {
         throw new Error("Autopilot kind must be provided.");

package/dist/cjs/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.InfluxDbReportingSink = exports.GrafanaLokiReportingSinkOptions = exports.GrafanaLokiReportingSink = exports.DatadogReportingSinkOptions = exports.DatadogReportingSink = exports.HttpAuthOptions = exports.HttpOAuth2ClientCredentialsOptions = exports.PushDiffusionEndpointDefinition = exports.DelegateStreamEndpointDefinition = exports.AzureEventHubsEndpointDefinition = exports.RedisStreamsEndpointDefinition = exports.NatsEndpointDefinition = exports.RabbitMqEndpointDefinition = exports.KafkaSaslOptions = exports.KafkaEndpointDefinition = exports.HttpEndpointDefinition = exports.TrafficEndpointDefinition = exports.LOADSTRIKE_TRACE_ID_TRACKING_FIELD = exports.LOADSTRIKE_TRACE_ID_HEADER = exports.EndpointAdapterFactory = exports.TrackingFieldSelector = exports.TrackingPayloadBuilder = exports.RedisCorrelationStore = exports.RedisCorrelationStoreOptions = exports.InMemoryCorrelationStore = exports.CrossPlatformTrackingRuntime = exports.CorrelationStoreConfiguration = exports.LoadStrikeThreshold = exports.LoadStrikeStep = exports.LoadStrikeGauge = exports.LoadStrikeCounter = exports.LoadStrikeMetric = exports.CrossPlatformTrackingConfiguration = exports.LoadStrikeSimulation = exports.LoadStrikeScenario = exports.LoadStrikeRunner = exports.LoadStrikeOperationType = exports.LoadStrikeScenarioOperation = exports.LoadStrikeLogLevel = exports.LoadStrikeResponse = exports.LoadStrikeReportFormat = exports.LoadStrikeNodeType = exports.LoadStrikePluginDataTable = exports.LoadStrikePluginData = exports.LoadStrikeContext = exports.ScenarioTrackingExtensions = exports.CrossPlatformScenarioConfigurator = exports.LoadStrikeAutopilotReadiness = exports.LoadStrikeAutopilotResult = exports.LoadStrikeAutopilot = void 0;
+exports.InfluxDbReportingSink = exports.GrafanaLokiReportingSinkOptions = exports.GrafanaLokiReportingSink = exports.DatadogReportingSinkOptions = exports.DatadogReportingSink = exports.HttpAuthOptions = exports.HttpOAuth2ClientCredentialsOptions = exports.PushDiffusionEndpointDefinition = exports.DelegateStreamEndpointDefinition = exports.SqsEndpointDefinition = exports.AzureEventHubsEndpointDefinition = exports.RedisStreamsEndpointDefinition = exports.NatsEndpointDefinition = exports.RabbitMqEndpointDefinition = exports.KafkaSaslOptions = exports.KafkaEndpointDefinition = exports.HttpEndpointDefinition = exports.TrafficEndpointDefinition = exports.LOADSTRIKE_TRACE_ID_TRACKING_FIELD = exports.LOADSTRIKE_TRACE_ID_HEADER = exports.TrackingFieldSelector = exports.TrackingPayloadBuilder = exports.RedisCorrelationStore = exports.RedisCorrelationStoreOptions = exports.InMemoryCorrelationStore = exports.CrossPlatformTrackingRuntime = exports.CorrelationStoreConfiguration = exports.LoadStrikeThreshold = exports.LoadStrikeStep = exports.LoadStrikeGauge = exports.LoadStrikeCounter = exports.LoadStrikeMetric = exports.CrossPlatformTrackingConfiguration = exports.LoadStrikeSimulation = exports.LoadStrikeScenario = exports.LoadStrikeRunner = exports.LoadStrikeOperationType = exports.LoadStrikeScenarioOperation = exports.LoadStrikeLogLevel = exports.LoadStrikeResponse = exports.LoadStrikeReportFormat = exports.LoadStrikeNodeType = exports.LoadStrikePluginDataTable = exports.LoadStrikePluginData = exports.LoadStrikeContext = exports.ScenarioTrackingExtensions = exports.CrossPlatformScenarioConfigurator = exports.LoadStrikeAutopilotReadiness = exports.LoadStrikeAutopilotResult = exports.LoadStrikeAutopilot = void 0;
 exports.TimescaleDbReportingSinkOptions = exports.TimescaleDbReportingSink = exports.SplunkReportingSinkOptions = exports.SplunkReportingSink = exports.OtelCollectorReportingSinkOptions = exports.OtelCollectorReportingSink = exports.InfluxDbReportingSinkOptions = void 0;
 var autopilot_js_1 = require("./autopilot.js");
 Object.defineProperty(exports, "LoadStrikeAutopilot", { enumerable: true, get: function () { return autopilot_js_1.LoadStrikeAutopilot; } });
@@ -37,7 +37,6 @@ Object.defineProperty(exports, "RedisCorrelationStore", { enumerable: true, get:
 Object.defineProperty(exports, "TrackingPayloadBuilder", { enumerable: true, get: function () { return correlation_js_1.TrackingPayloadBuilder; } });
 Object.defineProperty(exports, "TrackingFieldSelector", { enumerable: true, get: function () { return correlation_js_1.TrackingFieldSelector; } });
 var transports_js_1 = require("./transports.js");
-Object.defineProperty(exports, "EndpointAdapterFactory", { enumerable: true, get: function () { return transports_js_1.EndpointAdapterFactory; } });
 Object.defineProperty(exports, "LOADSTRIKE_TRACE_ID_HEADER", { enumerable: true, get: function () { return transports_js_1.LOADSTRIKE_TRACE_ID_HEADER; } });
 Object.defineProperty(exports, "LOADSTRIKE_TRACE_ID_TRACKING_FIELD", { enumerable: true, get: function () { return transports_js_1.LOADSTRIKE_TRACE_ID_TRACKING_FIELD; } });
 Object.defineProperty(exports, "TrafficEndpointDefinition", { enumerable: true, get: function () { return transports_js_1.TrafficEndpointDefinition; } });
@@ -48,6 +47,7 @@ Object.defineProperty(exports, "RabbitMqEndpointDefinition", { enumerable: true,
 Object.defineProperty(exports, "NatsEndpointDefinition", { enumerable: true, get: function () { return transports_js_1.NatsEndpointDefinition; } });
 Object.defineProperty(exports, "RedisStreamsEndpointDefinition", { enumerable: true, get: function () { return transports_js_1.RedisStreamsEndpointDefinition; } });
 Object.defineProperty(exports, "AzureEventHubsEndpointDefinition", { enumerable: true, get: function () { return transports_js_1.AzureEventHubsEndpointDefinition; } });
+Object.defineProperty(exports, "SqsEndpointDefinition", { enumerable: true, get: function () { return transports_js_1.SqsEndpointDefinition; } });
 Object.defineProperty(exports, "DelegateStreamEndpointDefinition", { enumerable: true, get: function () { return transports_js_1.DelegateStreamEndpointDefinition; } });
 Object.defineProperty(exports, "PushDiffusionEndpointDefinition", { enumerable: true, get: function () { return transports_js_1.PushDiffusionEndpointDefinition; } });
 Object.defineProperty(exports, "HttpOAuth2ClientCredentialsOptions", { enumerable: true, get: function () { return transports_js_1.HttpOAuth2ClientCredentialsOptions; } });

package/dist/cjs/local.js

@@ -44,7 +44,6 @@ const node_crypto_1 = require("node:crypto");
 const correlation_js_1 = require("./correlation.js");
 const transports_js_1 = require("./transports.js");
 const DEFAULT_LICENSING_API_BASE_URL = "https://licensing.loadstrike.com";
-const INTERNAL_BLACKBOX_LICENSING_API_BASE_URL_ENVIRONMENT_VARIABLE = "LOADSTRIKE_INTERNAL_BLACKBOX_API_BASE_URL";
 let developmentLicensingApiBaseUrlOverride;
 const BUILT_IN_WORKER_PLUGIN_NAMES = new Set([
     "loadstrike failed responses",
@@ -66,7 +65,8 @@ const TRACKING_FEATURE_BY_KIND = {
     pushdiffusion: "endpoint.push_diffusion",
     delegatestream: "endpoint.delegate_stream",
     nats: "endpoint.nats",
-    redisstreams: "endpoint.redis_streams"
+    redisstreams: "endpoint.redis_streams",
+    sqs: "endpoint.sqs"
 };
 const CI_ENVIRONMENT_VARIABLES = [
     "GITHUB_ACTIONS",
@@ -506,25 +506,8 @@ function normalizeLicensingApiBaseUrl(value) {
     const normalized = (value ?? "").trim();
     return normalized || DEFAULT_LICENSING_API_BASE_URL;
 }
-function resolveInternalBlackboxLicensingApiBaseUrlOverride(value) {
-    const normalized = String(value ?? "").trim();
-    if (!normalized) {
-        return undefined;
-    }
-    try {
-        const parsed = new URL(normalized);
-        const host = parsed.hostname.replace(/^\[|\]$/g, "").toLowerCase();
-        if (["http:", "https:"].includes(parsed.protocol) && ["127.0.0.1", "localhost", "::1"].includes(host)) {
-            return normalized.replace(/\/+$/, "");
-        }
-    }
-    catch {
-        return undefined;
-    }
-    return undefined;
-}
 function resolveLicensingApiBaseUrl() {
-    return normalizeLicensingApiBaseUrl(resolveInternalBlackboxLicensingApiBaseUrlOverride(process.env[INTERNAL_BLACKBOX_LICENSING_API_BASE_URL_ENVIRONMENT_VARIABLE]) ?? developmentLicensingApiBaseUrlOverride);
+    return normalizeLicensingApiBaseUrl(developmentLicensingApiBaseUrlOverride);
 }
 function setDevelopmentLicensingApiBaseUrlOverride(value) {
     developmentLicensingApiBaseUrlOverride = value;
@@ -563,6 +546,9 @@ function collectRequestedFeatures(request) {
     if (countCustomWorkerPlugins(context) > 0) {
         features.add("extensions.worker_plugins.custom");
     }
+    if (asList(pickValue(context, "RuntimePolicies", "runtimePolicies")).length > 0) {
+        features.add("policy.runtime_controls");
+    }
     const reportingSinks = asList(pickValue(context, "ReportingSinks", "reportingSinks"));
     let hasCustomSink = false;
     for (const sink of reportingSinks) {
@@ -858,159 +844,171 @@ async function evaluateScenarioOutcome(scenario, requestCount, context) {
     validateTrackingConfiguration(tracking, sourceEndpoint, destinationEndpoint);
     const sourceAdapter = transports_js_1.EndpointAdapterFactory.create(sourceEndpoint);
     const destinationAdapter = destinationEndpoint ? transports_js_1.EndpointAdapterFactory.create(destinationEndpoint) : null;
-    const correlationTimeoutOverride = pickValue(tracking, "CorrelationTimeoutMs", "correlationTimeoutMs");
-    const correlationTimeoutSeconds = pickValue(tracking, "CorrelationTimeoutSeconds", "correlationTimeoutSeconds", "CorrelationTimeout");
-    const timeoutMs = correlationTimeoutOverride != null && String(correlationTimeoutOverride).trim() !== ""
-        ? asInt(correlationTimeoutOverride)
-        : Math.trunc((correlationTimeoutSeconds == null || String(correlationTimeoutSeconds).trim() === ""
-            ? 30
-            : asNumber(correlationTimeoutSeconds)) * 1000);
-    const timeoutCountsAsFailure = toBoolean(pickValue(tracking, "TimeoutCountsAsFailure", "timeoutCountsAsFailure"), true);
-    const correlationStore = mapCorrelationStore(tracking, buildTrackingRunNamespace(stringOrDefault(pickValue(context, "SessionId", "sessionId"), "session"), stringOrDefault(pickValue(scenario, "Name", "name"), "scenario"), sourceEndpoint.name, destinationEndpoint?.name));
-    const timeoutSweepIntervalOverride = pickValue(tracking, "TimeoutSweepIntervalMs", "timeoutSweepIntervalMs");
-    const timeoutSweepIntervalSeconds = pickValue(tracking, "TimeoutSweepIntervalSeconds", "timeoutSweepIntervalSeconds");
-    const timeoutSweepIntervalMs = timeoutSweepIntervalOverride != null && String(timeoutSweepIntervalOverride).trim() !== ""
-        ? asInt(timeoutSweepIntervalOverride)
-        : Math.trunc((timeoutSweepIntervalSeconds == null || String(timeoutSweepIntervalSeconds).trim() === ""
-            ? 1
-            : asNumber(timeoutSweepIntervalSeconds)) * 1000);
-    const timeoutBatchSizeValue = pickValue(tracking, "TimeoutBatchSize", "timeoutBatchSize");
-    const timeoutBatchSize = timeoutBatchSizeValue == null || String(timeoutBatchSizeValue).trim() === ""
-        ? 200
-        : asInt(timeoutBatchSizeValue);
-    const runtime = new correlation_js_1.CrossPlatformTrackingRuntime({
-        sourceTrackingField: sourceEndpoint.trackingField,
-        destinationTrackingField: destinationEndpoint?.trackingField,
-        destinationGatherByField: destinationEndpoint?.gatherByField,
-        trackingFieldValueCaseSensitive: toBoolean(pickValue(tracking, "TrackingFieldValueCaseSensitive", "trackingFieldValueCaseSensitive"), true),
-        gatherByFieldValueCaseSensitive: toBoolean(pickValue(tracking, "GatherByFieldValueCaseSensitive", "gatherByFieldValueCaseSensitive"), true),
-        correlationTimeoutMs: timeoutMs,
-        timeoutCountsAsFailure,
-        store: correlationStore ?? undefined
-    });
-    const runMode = stringOrDefault(pickValue(tracking, "RunMode", "runMode"), "GenerateAndCorrelate").trim().toLowerCase();
-    const sourceOnlyMode = !destinationEndpoint;
-    const restartOnFail = toBoolean(pickValue(scenario, "RestartIterationOnFail", "restartIterationOnFail"), false);
-    const restartMaxAttempts = Math.max(asInt(pickValue(context, "RestartIterationMaxAttempts", "restartIterationMaxAttempts")), 0);
-    const maxFailCount = Math.max(asInt(pickValue(scenario, "MaxFailCount", "maxFailCount")), 0);
-    const scenarioCompletionTimeoutSeconds = Math.max(asNumber(pickValue(context, "ScenarioCompletionTimeoutSeconds", "scenarioCompletionTimeoutSeconds")), 0);
-    const scenarioDeadlineMs = scenarioCompletionTimeoutSeconds > 0
-        ? Date.now() + Math.trunc(scenarioCompletionTimeoutSeconds * 1000)
-        : Number.POSITIVE_INFINITY;
-    let okCount = 0;
-    let failCount = 0;
-    let nowMs = Date.now();
-    let processedIterations = 0;
-    let lastSweepAtMs = nowMs;
-    for (let i = 0; i < requestCount; i += 1) {
-        if (Date.now() > scenarioDeadlineMs) {
-            break;
-        }
-        let iterationComplete = false;
-        let attempts = 0;
-        const maxAttempts = 1 + (restartOnFail ? restartMaxAttempts : 0);
-        while (!iterationComplete) {
-            attempts += 1;
+    let correlationStore = null;
+    try {
+        await sourceAdapter.initialize?.();
+        await destinationAdapter?.initialize?.();
+        const correlationTimeoutOverride = pickValue(tracking, "CorrelationTimeoutMs", "correlationTimeoutMs");
+        const correlationTimeoutSeconds = pickValue(tracking, "CorrelationTimeoutSeconds", "correlationTimeoutSeconds", "CorrelationTimeout");
+        const timeoutMs = correlationTimeoutOverride != null && String(correlationTimeoutOverride).trim() !== ""
+            ? asInt(correlationTimeoutOverride)
+            : Math.trunc((correlationTimeoutSeconds == null || String(correlationTimeoutSeconds).trim() === ""
+                ? 30
+                : asNumber(correlationTimeoutSeconds)) * 1000);
+        const timeoutCountsAsFailure = toBoolean(pickValue(tracking, "TimeoutCountsAsFailure", "timeoutCountsAsFailure"), true);
+        correlationStore = mapCorrelationStore(tracking, buildTrackingRunNamespace(stringOrDefault(pickValue(context, "SessionId", "sessionId"), "session"), stringOrDefault(pickValue(scenario, "Name", "name"), "scenario"), sourceEndpoint.name, destinationEndpoint?.name));
+        const timeoutSweepIntervalOverride = pickValue(tracking, "TimeoutSweepIntervalMs", "timeoutSweepIntervalMs");
+        const timeoutSweepIntervalSeconds = pickValue(tracking, "TimeoutSweepIntervalSeconds", "timeoutSweepIntervalSeconds");
+        const timeoutSweepIntervalMs = timeoutSweepIntervalOverride != null && String(timeoutSweepIntervalOverride).trim() !== ""
+            ? asInt(timeoutSweepIntervalOverride)
+            : Math.trunc((timeoutSweepIntervalSeconds == null || String(timeoutSweepIntervalSeconds).trim() === ""
+                ? 1
+                : asNumber(timeoutSweepIntervalSeconds)) * 1000);
+        const timeoutBatchSizeValue = pickValue(tracking, "TimeoutBatchSize", "timeoutBatchSize");
+        const timeoutBatchSize = timeoutBatchSizeValue == null || String(timeoutBatchSizeValue).trim() === ""
+            ? 200
+            : asInt(timeoutBatchSizeValue);
+        const runtime = new correlation_js_1.CrossPlatformTrackingRuntime({
+            sourceTrackingField: sourceEndpoint.trackingField,
+            destinationTrackingField: destinationEndpoint?.trackingField,
+            destinationGatherByField: destinationEndpoint?.gatherByField,
+            trackingFieldValueCaseSensitive: toBoolean(pickValue(tracking, "TrackingFieldValueCaseSensitive", "trackingFieldValueCaseSensitive"), true),
+            gatherByFieldValueCaseSensitive: toBoolean(pickValue(tracking, "GatherByFieldValueCaseSensitive", "gatherByFieldValueCaseSensitive"), true),
+            correlationTimeoutMs: timeoutMs,
+            timeoutCountsAsFailure,
+            store: correlationStore ?? undefined
+        });
+        const runMode = stringOrDefault(pickValue(tracking, "RunMode", "runMode"), "GenerateAndCorrelate").trim().toLowerCase();
+        const sourceOnlyMode = !destinationEndpoint;
+        const restartOnFail = toBoolean(pickValue(scenario, "RestartIterationOnFail", "restartIterationOnFail"), false);
+        const restartMaxAttempts = Math.max(asInt(pickValue(context, "RestartIterationMaxAttempts", "restartIterationMaxAttempts")), 0);
+        const maxFailCount = Math.max(asInt(pickValue(scenario, "MaxFailCount", "maxFailCount")), 0);
+        const scenarioCompletionTimeoutSeconds = Math.max(asNumber(pickValue(context, "ScenarioCompletionTimeoutSeconds", "scenarioCompletionTimeoutSeconds")), 0);
+        const scenarioDeadlineMs = scenarioCompletionTimeoutSeconds > 0
+            ? Date.now() + Math.trunc(scenarioCompletionTimeoutSeconds * 1000)
+            : Number.POSITIVE_INFINITY;
+        let okCount = 0;
+        let failCount = 0;
+        let nowMs = Date.now();
+        let processedIterations = 0;
+        let lastSweepAtMs = nowMs;
+        for (let i = 0; i < requestCount; i += 1) {
             if (Date.now() > scenarioDeadlineMs) {
-                iterationComplete = true;
                 break;
             }
-            let sourcePayload = runMode === "correlateexistingtraffic"
-                ? await sourceAdapter.consume()
-                : await sourceAdapter.produce();
-            sourcePayload = normalizePayload(sourcePayload, sourceEndpoint, i);
-            const sourceTrackingId = sourceOnlyMode
-                ? readTrackingId(sourcePayload, sourceEndpoint.trackingField)
-                : await runtime.onSourceProduced(sourcePayload, nowMs);
-            if (!sourceTrackingId) {
-                if (restartOnFail && attempts < maxAttempts) {
+            let iterationComplete = false;
+            let attempts = 0;
+            const maxAttempts = 1 + (restartOnFail ? restartMaxAttempts : 0);
+            while (!iterationComplete) {
+                attempts += 1;
+                if (Date.now() > scenarioDeadlineMs) {
+                    iterationComplete = true;
+                    break;
+                }
+                let sourcePayload = runMode === "correlateexistingtraffic"
+                    ? await sourceAdapter.consume()
+                    : await sourceAdapter.produce();
+                sourcePayload = normalizePayload(sourcePayload, sourceEndpoint, i);
+                const sourceTrackingId = sourceOnlyMode
+                    ? readTrackingId(sourcePayload, sourceEndpoint.trackingField)
+                    : await runtime.onSourceProduced(sourcePayload, nowMs);
+                if (!sourceTrackingId) {
+                    if (restartOnFail && attempts < maxAttempts) {
+                        nowMs += 1;
+                        continue;
+                    }
+                    failCount += 1;
+                    iterationComplete = true;
                     nowMs += 1;
-                    continue;
+                    break;
                 }
-                failCount += 1;
-                iterationComplete = true;
-                nowMs += 1;
-                break;
-            }
-            if (sourceOnlyMode || !destinationAdapter || !destinationEndpoint) {
-                okCount += 1;
-                iterationComplete = true;
-                nowMs += 1;
-                break;
-            }
-            let destinationPayload = await destinationAdapter.consume();
-            destinationPayload = normalizePayload(destinationPayload, destinationEndpoint, i);
-            let matched = await runtime.onDestinationConsumed(destinationPayload, nowMs + 1);
-            const correlationDeadlineMs = nowMs + timeoutMs;
-            while (!matched && Date.now() <= scenarioDeadlineMs && nowMs < correlationDeadlineMs) {
-                const remainingTimeoutMs = correlationDeadlineMs - nowMs;
-                if (remainingTimeoutMs <= 0) {
+                if (sourceOnlyMode || !destinationAdapter || !destinationEndpoint) {
+                    okCount += 1;
+                    iterationComplete = true;
+                    nowMs += 1;
                     break;
                 }
-                await sleep(Math.min(destinationEndpoint.pollIntervalMs ?? 250, remainingTimeoutMs));
-                destinationPayload = normalizePayload(await destinationAdapter.consume(), destinationEndpoint, i);
-                matched = await runtime.onDestinationConsumed(destinationPayload, nowMs + 1);
+                let destinationPayload = await destinationAdapter.consume();
+                destinationPayload = normalizePayload(destinationPayload, destinationEndpoint, i);
+                let matched = await runtime.onDestinationConsumed(destinationPayload, nowMs + 1);
+                const correlationDeadlineMs = nowMs + timeoutMs;
+                while (!matched && Date.now() <= scenarioDeadlineMs && nowMs < correlationDeadlineMs) {
+                    const remainingTimeoutMs = correlationDeadlineMs - nowMs;
+                    if (remainingTimeoutMs <= 0) {
+                        break;
+                    }
+                    await sleep(Math.min(destinationEndpoint.pollIntervalMs ?? 250, remainingTimeoutMs));
+                    destinationPayload = normalizePayload(await destinationAdapter.consume(), destinationEndpoint, i);
+                    matched = await runtime.onDestinationConsumed(destinationPayload, nowMs + 1);
+                    if (matched) {
+                        break;
+                    }
+                    nowMs += Math.max(destinationEndpoint.pollIntervalMs ?? 250, 1);
+                }
                 if (matched) {
-                    break;
+                    okCount += 1;
+                    iterationComplete = true;
                 }
-                nowMs += Math.max(destinationEndpoint.pollIntervalMs ?? 250, 1);
-            }
-            if (matched) {
-                okCount += 1;
-                iterationComplete = true;
-            }
-            else if (restartOnFail && attempts < maxAttempts) {
-                nowMs += 2;
-                continue;
-            }
-            else {
-                await runtime.sweepTimeouts(nowMs + timeoutMs + 1, timeoutBatchSize || undefined);
-                failCount += 1;
-                iterationComplete = true;
-            }
-            nowMs += 2;
-        }
-        processedIterations += 1;
-        if (timeoutSweepIntervalMs > 0 && nowMs - lastSweepAtMs >= timeoutSweepIntervalMs) {
-            const swept = await runtime.sweepTimeouts(nowMs, timeoutBatchSize || undefined);
-            if (swept > 0) {
-                if (timeoutCountsAsFailure) {
-                    failCount += swept;
+                else if (restartOnFail && attempts < maxAttempts) {
+                    nowMs += 2;
+                    continue;
                 }
                 else {
-                    okCount += swept;
+                    await runtime.sweepTimeouts(nowMs + timeoutMs + 1, timeoutBatchSize || undefined);
+                    failCount += 1;
+                    iterationComplete = true;
                 }
+                nowMs += 2;
+            }
+            processedIterations += 1;
+            if (timeoutSweepIntervalMs > 0 && nowMs - lastSweepAtMs >= timeoutSweepIntervalMs) {
+                const swept = await runtime.sweepTimeouts(nowMs, timeoutBatchSize || undefined);
+                if (swept > 0) {
+                    if (timeoutCountsAsFailure) {
+                        failCount += swept;
+                    }
+                    else {
+                        okCount += swept;
+                    }
+                }
+                lastSweepAtMs = nowMs;
+            }
+            if (maxFailCount > 0 && failCount >= maxFailCount) {
+                break;
             }
-            lastSweepAtMs = nowMs;
-        }
-        if (maxFailCount > 0 && failCount >= maxFailCount) {
-            break;
         }
-    }
-    let expired = 0;
-    let sweptChunk = 0;
-    do {
-        sweptChunk = await runtime.sweepTimeouts(nowMs + timeoutMs + 1, timeoutBatchSize || undefined);
-        expired += sweptChunk;
-    } while (timeoutBatchSize > 0 && sweptChunk >= timeoutBatchSize);
-    if (expired > 0) {
-        if (timeoutCountsAsFailure) {
-            failCount += expired;
+        let expired = 0;
+        let sweptChunk = 0;
+        do {
+            sweptChunk = await runtime.sweepTimeouts(nowMs + timeoutMs + 1, timeoutBatchSize || undefined);
+            expired += sweptChunk;
+        } while (timeoutBatchSize > 0 && sweptChunk >= timeoutBatchSize);
+        if (expired > 0) {
+            if (timeoutCountsAsFailure) {
+                failCount += expired;
+            }
+            else {
+                okCount += expired;
+            }
         }
-        else {
-            okCount += expired;
+        if (processedIterations < requestCount && Date.now() > scenarioDeadlineMs) {
+            const timedOutIterations = requestCount - processedIterations;
+            if (timeoutCountsAsFailure) {
+                failCount += timedOutIterations;
+            }
+            else {
+                okCount += timedOutIterations;
+            }
         }
+        return { requestCount: processedIterations, okCount, failCount };
     }
-    if (processedIterations < requestCount && Date.now() > scenarioDeadlineMs) {
-        const timedOutIterations = requestCount - processedIterations;
-        if (timeoutCountsAsFailure) {
-            failCount += timedOutIterations;
-        }
-        else {
-            okCount += timedOutIterations;
-        }
+    finally {
+        await Promise.allSettled([
+            sourceAdapter.dispose?.(),
+            destinationAdapter?.dispose?.(),
+            correlationStore?.close?.()
+        ].filter((value) => Boolean(value)));
     }
-    return { requestCount: processedIterations, okCount, failCount };
 }
 function validateTrackingConfiguration(tracking, sourceEndpoint, destinationEndpoint) {
     if (sourceEndpoint.gatherByField?.trim()) {

package/dist/cjs/runtime.js

@@ -6626,6 +6626,7 @@ function mapRuntimeTrackingEndpointSpec(spec, useLoadStrikeTraceIdHeader = false
         nats: asTrackingRecord(pickTrackingValue(spec, "Nats", "nats")),
         redisStreams: asTrackingRecord(pickTrackingValue(spec, "RedisStreams", "redisStreams")),
         azureEventHubs: asTrackingRecord(pickTrackingValue(spec, "AzureEventHubs", "azureEventHubs")),
+        sqs: asTrackingRecord(pickTrackingValue(spec, "Sqs", "sqs")),
         pushDiffusion: asTrackingRecord(pickTrackingValue(spec, "PushDiffusion", "pushDiffusion")),
         delegate: typeof delegateProduce === "function"
             || typeof delegateConsume === "function"