@loadstrike/loadstrike-sdk 0.1.10001 → 1.0.10801

package/dist/cjs/sinks.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.OtelCollectorReportingSink = exports.SplunkReportingSink = exports.DatadogReportingSink = exports.TimescaleDbReportingSink = exports.GrafanaLokiReportingSink = exports.InfluxDbReportingSink = exports.CompositeReportingSink = exports.ConsoleReportingSink = exports.MemoryReportingSink = exports.OtelCollectorReportingSinkOptions = exports.SplunkReportingSinkOptions = exports.DatadogReportingSinkOptions = exports.TimescaleDbReportingSinkOptions = exports.GrafanaLokiReportingSinkOptions = exports.InfluxDbReportingSinkOptions = void 0;
+exports.__loadstrikeTestExports = exports.OtelCollectorReportingSink = exports.SplunkReportingSink = exports.DatadogReportingSink = exports.TimescaleDbReportingSink = exports.GrafanaLokiReportingSink = exports.InfluxDbReportingSink = exports.CompositeReportingSink = exports.ConsoleReportingSink = exports.MemoryReportingSink = exports.OtelCollectorReportingSinkOptions = exports.SplunkReportingSinkOptions = exports.DatadogReportingSinkOptions = exports.TimescaleDbReportingSinkOptions = exports.GrafanaLokiReportingSinkOptions = exports.InfluxDbReportingSinkOptions = void 0;
 const runtime_js_1 = require("./runtime.js");
 const pg_1 = require("pg");
 const DEFAULT_INFLUX_CONFIGURATION_SECTION_PATH = "LoadStrike:ReportingSinks:InfluxDb";
@@ -244,13 +244,8 @@ class MemoryReportingSink {
     SaveRealtimeMetrics(metrics) {
         this.saveRealtimeMetrics(metrics);
     }
-    saveFinalStats(result) {
-        this.finalResults.push(cloneNodeStats(result));
-    }
-    SaveFinalStats(result) {
-        this.saveFinalStats(result);
-    }
     saveRunResult(result) {
+        this.finalResults.push(cloneRunResult(result));
         this.runResults.push(deepCloneRecord(result));
     }
     SaveRunResult(result) {
@@ -293,13 +288,9 @@ class ConsoleReportingSink {
     SaveRealtimeMetrics(metrics) {
         this.saveRealtimeMetrics(metrics);
     }
-    saveFinalStats(result) {
+    saveRunResult(result) {
         this.writeLine(`final requests=${result.allRequestCount} ok=${result.allOkCount} fail=${result.allFailCount} failedThresholds=${result.failedThresholds}`);
     }
-    SaveFinalStats(result) {
-        this.saveFinalStats(result);
-    }
-    saveRunResult(_result) { }
     SaveRunResult(result) {
         this.saveRunResult(result);
     }
@@ -369,17 +360,6 @@ class CompositeReportingSink {
     async SaveRealtimeMetrics(metrics) {
         await this.saveRealtimeMetrics(metrics);
     }
-    async saveFinalStats(result) {
-        for (const sink of this.sinks) {
-            const saveFinalStats = sink.saveFinalStats ?? sink.SaveFinalStats;
-            if (saveFinalStats) {
-                await saveFinalStats.call(sink, result);
-            }
-        }
-    }
-    async SaveFinalStats(result) {
-        await this.saveFinalStats(result);
-    }
     async saveRunResult(result) {
         for (const sink of this.sinks) {
             const saveRunResult = sink.saveRunResult ?? sink.SaveRunResult;
@@ -466,12 +446,6 @@ class InfluxDbReportingSink {
     async SaveRealtimeMetrics(metrics) {
         await this.saveRealtimeMetrics(metrics);
     }
-    async saveFinalStats(result) {
-        await this.persistEvents(createFinalStatsEvents(this.getSession(), result));
-    }
-    async SaveFinalStats(result) {
-        await this.saveFinalStats(result);
-    }
     async saveRunResult(result) {
         await this.persistEvents(createRunResultEvents(this.getSession(), result));
     }
@@ -582,12 +556,6 @@ class GrafanaLokiReportingSink {
     async SaveRealtimeMetrics(metrics) {
         await this.saveRealtimeMetrics(metrics);
     }
-    async saveFinalStats(result) {
-        await this.persistEvents(createFinalStatsEvents(this.getSession(), result));
-    }
-    async SaveFinalStats(result) {
-        await this.saveFinalStats(result);
-    }
     async saveRunResult(result) {
         await this.persistEvents(createRunResultEvents(this.getSession(), result));
     }
@@ -734,12 +702,6 @@ class TimescaleDbReportingSink {
     async SaveRealtimeMetrics(metrics) {
         await this.saveRealtimeMetrics(metrics);
     }
-    async saveFinalStats(result) {
-        await this.persistEvents(createFinalStatsEvents(this.getSession(), result));
-    }
-    async SaveFinalStats(result) {
-        await this.saveFinalStats(result);
-    }
     async saveRunResult(result) {
         await this.persistEvents(createRunResultEvents(this.getSession(), result));
     }
@@ -1016,12 +978,6 @@ class DatadogReportingSink {
     async SaveRealtimeMetrics(metrics) {
         await this.saveRealtimeMetrics(metrics);
     }
-    async saveFinalStats(result) {
-        await this.persistEvents(createFinalStatsEvents(this.getSession(), result));
-    }
-    async SaveFinalStats(result) {
-        await this.saveFinalStats(result);
-    }
     async saveRunResult(result) {
         await this.persistEvents(createRunResultEvents(this.getSession(), result));
     }
@@ -1145,12 +1101,6 @@ class SplunkReportingSink {
     async SaveRealtimeMetrics(metrics) {
         await this.saveRealtimeMetrics(metrics);
     }
-    async saveFinalStats(result) {
-        await this.persistEvents(createFinalStatsEvents(this.getSession(), result));
-    }
-    async SaveFinalStats(result) {
-        await this.saveFinalStats(result);
-    }
     async saveRunResult(result) {
         await this.persistEvents(createRunResultEvents(this.getSession(), result));
     }
@@ -1254,12 +1204,6 @@ class OtelCollectorReportingSink {
     async SaveRealtimeMetrics(metrics) {
         await this.saveRealtimeMetrics(metrics);
     }
-    async saveFinalStats(result) {
-        await this.persistEvents(createFinalStatsEvents(this.getSession(), result));
-    }
-    async SaveFinalStats(result) {
-        await this.saveFinalStats(result);
-    }
     async saveRunResult(result) {
         await this.persistEvents(createRunResultEvents(this.getSession(), result));
     }
@@ -1374,6 +1318,7 @@ function createFinalStatsEvents(session, stats) {
 function createRunResultEvents(session, result) {
     const occurredUtc = new Date();
     const events = [
+        ...createFinalStatsEvents(session, runResultToNodeStats(result)),
         createReportingEvent(session, occurredUtc, "run.result.final", null, null, {
             phase: "final",
             entity: "run-result"
@@ -1468,7 +1413,8 @@ function createStepEvents(session, occurredUtc, phase, scenario) {
     return events;
 }
 function createStatusCodeEvents(session, occurredUtc, phase, scenarioName, stepName, resultKind, statusCodes) {
-    return statusCodes.map((statusCode) => createReportingEvent(session, occurredUtc, `status-code.${phase}`, scenarioName, stepName, {
+    const normalizedStatusCodes = Array.isArray(statusCodes) ? statusCodes : [];
+    return normalizedStatusCodes.map((statusCode) => createReportingEvent(session, occurredUtc, `status-code.${phase}`, scenarioName, stepName, {
         phase,
         entity: stepName ? "step-status-code" : "scenario-status-code",
         result_kind: resultKind,
@@ -1601,30 +1547,63 @@ function createReportingEvent(session, occurredUtc, eventType, scenarioName, ste
     };
 }
 function addMeasurementFields(fields, prefix, measurement) {
-    fields[`${prefix}_request_count`] = measurement.request.count;
-    fields[`${prefix}_request_percent`] = measurement.request.percent;
-    fields[`${prefix}_request_rps`] = measurement.request.rps;
-    fields[`${prefix}_latency_min_ms`] = measurement.latency.minMs;
-    fields[`${prefix}_latency_mean_ms`] = measurement.latency.meanMs;
-    fields[`${prefix}_latency_max_ms`] = measurement.latency.maxMs;
-    fields[`${prefix}_latency_p50_ms`] = measurement.latency.percent50;
-    fields[`${prefix}_latency_p75_ms`] = measurement.latency.percent75;
-    fields[`${prefix}_latency_p95_ms`] = measurement.latency.percent95;
-    fields[`${prefix}_latency_p99_ms`] = measurement.latency.percent99;
-    fields[`${prefix}_latency_std_dev`] = measurement.latency.stdDev;
-    fields[`${prefix}_latency_le_800_count`] = measurement.latency.latencyCount.lessOrEq800;
-    fields[`${prefix}_latency_gt_800_lt_1200_count`] = measurement.latency.latencyCount.more800Less1200;
-    fields[`${prefix}_latency_ge_1200_count`] = measurement.latency.latencyCount.moreOrEq1200;
-    fields[`${prefix}_bytes_all`] = measurement.dataTransfer.allBytes;
-    fields[`${prefix}_bytes_min`] = measurement.dataTransfer.minBytes;
-    fields[`${prefix}_bytes_mean`] = measurement.dataTransfer.meanBytes;
-    fields[`${prefix}_bytes_max`] = measurement.dataTransfer.maxBytes;
-    fields[`${prefix}_bytes_p50`] = measurement.dataTransfer.percent50;
-    fields[`${prefix}_bytes_p75`] = measurement.dataTransfer.percent75;
-    fields[`${prefix}_bytes_p95`] = measurement.dataTransfer.percent95;
-    fields[`${prefix}_bytes_p99`] = measurement.dataTransfer.percent99;
-    fields[`${prefix}_bytes_std_dev`] = measurement.dataTransfer.stdDev;
-    fields[`${prefix}_status_code_count`] = measurement.statusCodes.length;
+    const request = measurement?.request ?? { count: 0, percent: 0, rps: 0 };
+    const latency = measurement?.latency ?? {
+        minMs: 0,
+        meanMs: 0,
+        maxMs: 0,
+        percent50: 0,
+        percent75: 0,
+        percent95: 0,
+        percent99: 0,
+        stdDev: 0,
+        latencyCount: {
+            lessOrEq800: 0,
+            more800Less1200: 0,
+            moreOrEq1200: 0
+        }
+    };
+    const latencyCount = latency.latencyCount ?? {
+        lessOrEq800: 0,
+        more800Less1200: 0,
+        moreOrEq1200: 0
+    };
+    const dataTransfer = measurement?.dataTransfer ?? {
+        allBytes: 0,
+        minBytes: 0,
+        meanBytes: 0,
+        maxBytes: 0,
+        percent50: 0,
+        percent75: 0,
+        percent95: 0,
+        percent99: 0,
+        stdDev: 0
+    };
+    const statusCodes = Array.isArray(measurement?.statusCodes) ? measurement.statusCodes : [];
+    fields[`${prefix}_request_count`] = request.count ?? 0;
+    fields[`${prefix}_request_percent`] = request.percent ?? 0;
+    fields[`${prefix}_request_rps`] = request.rps ?? 0;
+    fields[`${prefix}_latency_min_ms`] = latency.minMs ?? 0;
+    fields[`${prefix}_latency_mean_ms`] = latency.meanMs ?? 0;
+    fields[`${prefix}_latency_max_ms`] = latency.maxMs ?? 0;
+    fields[`${prefix}_latency_p50_ms`] = latency.percent50 ?? 0;
+    fields[`${prefix}_latency_p75_ms`] = latency.percent75 ?? 0;
+    fields[`${prefix}_latency_p95_ms`] = latency.percent95 ?? 0;
+    fields[`${prefix}_latency_p99_ms`] = latency.percent99 ?? 0;
+    fields[`${prefix}_latency_std_dev`] = latency.stdDev ?? 0;
+    fields[`${prefix}_latency_le_800_count`] = latencyCount.lessOrEq800 ?? 0;
+    fields[`${prefix}_latency_gt_800_lt_1200_count`] = latencyCount.more800Less1200 ?? 0;
+    fields[`${prefix}_latency_ge_1200_count`] = latencyCount.moreOrEq1200 ?? 0;
+    fields[`${prefix}_bytes_all`] = dataTransfer.allBytes ?? 0;
+    fields[`${prefix}_bytes_min`] = dataTransfer.minBytes ?? 0;
+    fields[`${prefix}_bytes_mean`] = dataTransfer.meanBytes ?? 0;
+    fields[`${prefix}_bytes_max`] = dataTransfer.maxBytes ?? 0;
+    fields[`${prefix}_bytes_p50`] = dataTransfer.percent50 ?? 0;
+    fields[`${prefix}_bytes_p75`] = dataTransfer.percent75 ?? 0;
+    fields[`${prefix}_bytes_p95`] = dataTransfer.percent95 ?? 0;
+    fields[`${prefix}_bytes_p99`] = dataTransfer.percent99 ?? 0;
+    fields[`${prefix}_bytes_std_dev`] = dataTransfer.stdDev ?? 0;
+    fields[`${prefix}_status_code_count`] = statusCodes.length;
 }
 function buildInfluxWriteUri(options) {
     const query = new URLSearchParams({
@@ -2596,28 +2575,28 @@ function cloneSessionStartInfo(session) {
     };
 }
 function cloneNodeInfo(nodeInfo) {
-    return { ...nodeInfo };
+    return { ...(nodeInfo ?? {}) };
 }
 function cloneTestInfo(testInfo) {
-    return { ...testInfo };
+    return { ...(testInfo ?? {}) };
 }
 function cloneMetricStats(metrics) {
     return {
-        counters: metrics.counters.map((value) => ({ ...value })),
-        gauges: metrics.gauges.map((value) => ({ ...value })),
-        durationMs: metrics.durationMs
+        counters: (metrics?.counters ?? []).map((value) => ({ ...value })),
+        gauges: (metrics?.gauges ?? []).map((value) => ({ ...value })),
+        durationMs: metrics?.durationMs ?? 0
     };
 }
 function cloneScenarioStats(value) {
     return {
         ...value,
-        ok: deepCloneRecord(value.ok),
-        fail: deepCloneRecord(value.fail),
-        loadSimulationStats: { ...value.loadSimulationStats },
-        stepStats: value.stepStats.map((step) => ({
+        ok: deepCloneRecord(value?.ok ?? {}),
+        fail: deepCloneRecord(value?.fail ?? {}),
+        loadSimulationStats: { ...(value?.loadSimulationStats ?? {}) },
+        stepStats: (value?.stepStats ?? []).map((step) => ({
             ...step,
-            ok: deepCloneRecord(step.ok),
-            fail: deepCloneRecord(step.fail)
+            ok: deepCloneRecord(step?.ok ?? {}),
+            fail: deepCloneRecord(step?.fail ?? {})
         }))
     };
 }
@@ -2626,20 +2605,91 @@ function cloneNodeStats(result) {
         ...result,
         nodeInfo: cloneNodeInfo(result.nodeInfo),
         testInfo: cloneTestInfo(result.testInfo),
-        thresholds: result.thresholds.map((value) => ({ ...value })),
-        thresholdResults: result.thresholdResults.map((value) => ({ ...value })),
+        thresholds: (result.thresholds ?? []).map((value) => ({ ...value })),
+        thresholdResults: (result.thresholdResults ?? []).map((value) => ({ ...value })),
         metrics: cloneMetricStats(result.metrics),
-        metricValues: result.metricValues.map((value) => ({ ...value })),
-        scenarioStats: result.scenarioStats.map((value) => cloneScenarioStats(value)),
-        stepStats: result.stepStats.map((step) => ({
+        metricValues: Array.isArray(result.metricValues) ? result.metricValues.map((value) => ({ ...value })) : [],
+        scenarioStats: (result.scenarioStats ?? []).map((value) => cloneScenarioStats(value)),
+        stepStats: (result.stepStats ?? []).map((step) => ({
+            ...step,
+            ok: deepCloneRecord(step?.ok ?? {}),
+            fail: deepCloneRecord(step?.fail ?? {})
+        })),
+        pluginsData: (result.pluginsData ?? []).map((plugin) => new runtime_js_1.LoadStrikePluginData(plugin.pluginName, (plugin.tables ?? []).map((table) => new runtime_js_1.LoadStrikePluginDataTable(table.tableName, [...(table.headers ?? [])], (table.rows ?? []).map((row) => deepCloneRecord(row)))), [...(plugin.hints ?? [])])),
+        disabledSinks: [...(result.disabledSinks ?? [])],
+        sinkErrors: (result.sinkErrors ?? []).map((value) => ({ ...value })),
+        reportFiles: [...(result.reportFiles ?? [])],
+        logFiles: [...(result.logFiles ?? [])]
+    };
+}
+function cloneRunResult(result) {
+    return {
+        ...result,
+        nodeInfo: cloneNodeInfo(result.nodeInfo),
+        testInfo: cloneTestInfo(result.testInfo),
+        thresholds: (result.thresholds ?? []).map((value) => ({ ...value })),
+        thresholdResults: (result.thresholdResults ?? []).map((value) => ({ ...value })),
+        metricStats: cloneMetricStats(result.metricStats),
+        metrics: Array.isArray(result.metrics) ? result.metrics.map((value) => ({ ...value })) : [],
+        scenarioStats: (result.scenarioStats ?? []).map((value) => cloneScenarioStats(value)),
+        stepStats: (result.stepStats ?? []).map((step) => ({
             ...step,
-            ok: deepCloneRecord(step.ok),
-            fail: deepCloneRecord(step.fail)
+            ok: deepCloneRecord(step?.ok ?? {}),
+            fail: deepCloneRecord(step?.fail ?? {})
         })),
-        pluginsData: result.pluginsData.map((plugin) => new runtime_js_1.LoadStrikePluginData(plugin.pluginName, plugin.tables.map((table) => new runtime_js_1.LoadStrikePluginDataTable(table.tableName, [...table.headers], table.rows.map((row) => deepCloneRecord(row)))), [...plugin.hints])),
-        disabledSinks: [...result.disabledSinks],
-        sinkErrors: result.sinkErrors.map((value) => ({ ...value })),
-        reportFiles: [...result.reportFiles]
+        scenarioDurationsMs: { ...(result.scenarioDurationsMs ?? {}) },
+        pluginsData: (result.pluginsData ?? []).map((plugin) => new runtime_js_1.LoadStrikePluginData(plugin.pluginName, (plugin.tables ?? []).map((table) => new runtime_js_1.LoadStrikePluginDataTable(table.tableName, [...(table.headers ?? [])], (table.rows ?? []).map((row) => deepCloneRecord(row)))), [...(plugin.hints ?? [])])),
+        disabledSinks: [...(result.disabledSinks ?? [])],
+        sinkErrors: (result.sinkErrors ?? []).map((value) => ({ ...value })),
+        reportFiles: [...(result.reportFiles ?? [])],
+        logFiles: [...(result.logFiles ?? [])],
+        correlationRows: (result.correlationRows ?? []).map((row) => deepCloneRecord(row)),
+        failedCorrelationRows: (result.failedCorrelationRows ?? []).map((row) => deepCloneRecord(row))
+    };
+}
+function runResultToNodeStats(result) {
+    return {
+        startedUtc: result.startedUtc,
+        completedUtc: result.completedUtc,
+        allBytes: result.allBytes ?? 0,
+        allRequestCount: result.allRequestCount ?? 0,
+        allOkCount: result.allOkCount ?? 0,
+        allFailCount: result.allFailCount ?? 0,
+        failedThresholds: result.failedThresholds ?? 0,
+        durationMs: result.durationMs ?? 0,
+        nodeInfo: cloneNodeInfo(result.nodeInfo),
+        testInfo: cloneTestInfo(result.testInfo),
+        thresholds: (result.thresholds ?? []).map((value) => ({ ...value })),
+        thresholdResults: (result.thresholdResults ?? []).map((value) => ({ ...value })),
+        metrics: cloneMetricStats(result.metricStats),
+        metricValues: Array.isArray(result.metrics) ? result.metrics.map((value) => ({ ...value })) : [],
+        scenarioStats: (result.scenarioStats ?? []).map((value) => cloneScenarioStats(value)),
+        stepStats: (result.stepStats ?? []).map((step) => ({
+            ...step,
+            ok: deepCloneRecord(step?.ok ?? {}),
+            fail: deepCloneRecord(step?.fail ?? {})
+        })),
+        pluginsData: (result.pluginsData ?? []).map((plugin) => new runtime_js_1.LoadStrikePluginData(plugin.pluginName, (plugin.tables ?? []).map((table) => new runtime_js_1.LoadStrikePluginDataTable(table.tableName, [...(table.headers ?? [])], (table.rows ?? []).map((row) => deepCloneRecord(row)))), [...(plugin.hints ?? [])])),
+        disabledSinks: [...(result.disabledSinks ?? [])],
+        sinkErrors: (result.sinkErrors ?? []).map((value) => ({ ...value })),
+        reportFiles: [...(result.reportFiles ?? [])],
+        logFiles: [...(result.logFiles ?? [])],
+        findScenarioStats: (scenarioName) => result.scenarioStats.find((value) => value.scenarioName === scenarioName),
+        getScenarioStats: (scenarioName) => {
+            const scenario = result.scenarioStats.find((value) => value.scenarioName === scenarioName);
+            if (!scenario) {
+                throw new Error(`Scenario stats not found: ${scenarioName}`);
+            }
+            return scenario;
+        },
+        FindScenarioStats: (scenarioName) => result.scenarioStats.find((value) => value.scenarioName === scenarioName),
+        GetScenarioStats: (scenarioName) => {
+            const scenario = result.scenarioStats.find((value) => value.scenarioName === scenarioName);
+            if (!scenario) {
+                throw new Error(`Scenario stats not found: ${scenarioName}`);
+            }
+            return scenario;
+        }
     };
 }
 function deepCloneRecord(value) {
@@ -2673,3 +2723,47 @@ async function postWithTimeout(fetchImpl, url, init, timeoutMs, sinkName) {
         clearTimeout(timer);
     }
 }
+exports.__loadstrikeTestExports = {
+    GrafanaLokiReportingSink,
+    InfluxDbReportingSink,
+    OtelCollectorReportingSink,
+    SplunkReportingSink,
+    TimescaleDbReportingSink,
+    cleanNullableText,
+    cloneBaseContext,
+    cloneMetricStats,
+    cloneNodeStats,
+    cloneScenarioStats,
+    cloneSessionStartInfo,
+    createFinalStatsEvents,
+    createRealtimeStatsEvents,
+    createRunResultEvents,
+    createReportingEvent,
+    mergeDatadogOptions,
+    mergeInfluxOptions,
+    mergeGrafanaLokiOptions,
+    mergeOtelCollectorOptions,
+    mergeSplunkOptions,
+    mergeTimescaleDbOptions,
+    normalizeConfigKey,
+    normalizePath,
+    normalizePluginFieldName,
+    normalizePluginFieldValue,
+    normalizeStringMap,
+    optionNumber,
+    optionString,
+    pickBooleanValue,
+    pickRecordValue,
+    postWithTimeout,
+    quoteIdentifier,
+    replaceLiteral,
+    resolveConfigSection,
+    resolveTimeoutMs,
+    sanitizeGrafanaLabelKey,
+    sanitizeGrafanaLabelValue,
+    sinkSessionMetadataFromContext,
+    toOtelAnyValue,
+    trimTrailingSlashes,
+    tryReadText,
+    validateIdentifier
+};

package/dist/cjs/transports.js

@@ -33,7 +33,7 @@ var __importStar = (this && this.__importStar) || (function () {
     };
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.EndpointAdapterFactory = exports.KafkaSaslOptions = exports.HttpAuthOptions = exports.HttpOAuth2ClientCredentialsOptions = exports.PushDiffusionEndpointDefinition = exports.DelegateStreamEndpointDefinition = exports.AzureEventHubsEndpointDefinition = exports.RedisStreamsEndpointDefinition = exports.NatsEndpointDefinition = exports.RabbitMqEndpointDefinition = exports.KafkaEndpointDefinition = exports.HttpEndpointDefinition = exports.TrafficEndpointDefinition = void 0;
+exports.__loadstrikeTestExports = exports.EndpointAdapterFactory = exports.KafkaSaslOptions = exports.HttpAuthOptions = exports.HttpOAuth2ClientCredentialsOptions = exports.PushDiffusionEndpointDefinition = exports.DelegateStreamEndpointDefinition = exports.AzureEventHubsEndpointDefinition = exports.RedisStreamsEndpointDefinition = exports.NatsEndpointDefinition = exports.RabbitMqEndpointDefinition = exports.KafkaEndpointDefinition = exports.HttpEndpointDefinition = exports.TrafficEndpointDefinition = void 0;
 const node_crypto_1 = require("node:crypto");
 const correlation_js_1 = require("./correlation.js");
 class TrafficEndpointDefinitionModel {
@@ -247,6 +247,9 @@ class AzureEventHubsEndpointDefinitionModel extends TrafficEndpointDefinitionMod
         super.Validate();
         requireNonEmptyString(this.ConnectionString, "ConnectionString must be provided for Azure Event Hubs endpoint.");
         requireNonEmptyString(this.EventHubName, "EventHubName must be provided for Azure Event Hubs endpoint.");
+        if (this.PartitionCount != null && this.PartitionCount < 0) {
+            throw new RangeError("PartitionCount must be zero or greater when configured.");
+        }
     }
 }
 class DelegateStreamEndpointDefinitionModel extends TrafficEndpointDefinitionModel {
@@ -650,6 +653,12 @@ function initializeAzureEventHubsEndpointDefinitionModel(target, initial) {
     if (hasAnyEndpointField(raw, ["PartitionId", "partitionId"])) {
         target.PartitionId = pickOptionalEndpointString(raw, "PartitionId", "partitionId");
     }
+    if (hasAnyEndpointField(raw, ["PartitionKey", "partitionKey"])) {
+        target.PartitionKey = pickOptionalEndpointString(raw, "PartitionKey", "partitionKey");
+    }
+    if (hasAnyEndpointField(raw, ["PartitionCount", "partitionCount"])) {
+        target.PartitionCount = pickOptionalEndpointNumber(raw, "PartitionCount", "partitionCount");
+    }
 }
 function initializeDelegateStreamEndpointDefinitionModel(target, initial) {
     const raw = asRecordOrEmpty(initial);
@@ -740,10 +749,9 @@ function validateHttpAuthOptionsModel(target) {
             }
             return;
         case "oauth2clientcredentials":
-            if (!target.OAuth2ClientCredentials) {
-                throw new Error("OAuth2ClientCredentials must be provided for OAuth2 client-credentials auth.");
-            }
-            toHttpOAuth2ClientCredentialsOptionsModel(target.OAuth2ClientCredentials).Validate();
+            requireNonEmptyString(target.TokenUrl ?? target.OAuth2ClientCredentials?.TokenEndpoint, "TokenEndpoint must be provided for OAuth2 client credentials flow.");
+            requireNonEmptyString(target.ClientId ?? target.OAuth2ClientCredentials?.ClientId, "ClientId must be provided for OAuth2 client credentials flow.");
+            requireNonEmptyString(target.ClientSecret ?? target.OAuth2ClientCredentials?.ClientSecret, "ClientSecret must be provided for OAuth2 client credentials flow.");
             return;
         default:
             throw new RangeError("Unsupported HTTP auth type.");
@@ -933,8 +941,7 @@ const protocolBus = new class InMemoryProtocolBus {
     produceEventHub(endpoint, payload) {
         const options = endpoint.azureEventHubs ?? {};
         const hubName = optionString(options, "EventHubName", "eventHubName") || endpoint.name;
-        const partitionId = optionString(options, "PartitionId", "partitionId")
-            || partitionFromKey(optionString(options, "PartitionKey", "partitionKey"), optionNumber(options, "PartitionCount", "partitionCount") || 4);
+        const partitionId = resolveEventHubProducePartitionId(options) || "";
         const rows = this.eventHubs.get(hubName) ?? [];
         rows.push({
             partitionId,
@@ -1160,10 +1167,10 @@ class HttpEndpointAdapter extends CallbackAdapter {
         body.set("grant_type", "client_credentials");
         body.set("client_id", clientId);
         body.set("client_secret", clientSecret);
-        const scopes = Array.isArray(oauthOptions?.scopes)
-            ? oauthOptions.scopes.map((x) => String(x).trim()).filter((x) => x.length > 0)
-            : Array.isArray(auth?.scopes)
-                ? auth.scopes.map((x) => String(x).trim()).filter((x) => x.length > 0)
+        const scopes = Array.isArray(auth?.scopes)
+            ? auth.scopes.map((x) => String(x).trim()).filter((x) => x.length > 0)
+            : Array.isArray(oauthOptions?.scopes)
+                ? oauthOptions.scopes.map((x) => String(x).trim()).filter((x) => x.length > 0)
                 : [];
         const scopeText = typeof auth?.scope === "string" && auth.scope.trim()
             ? auth.scope.trim()
@@ -1175,8 +1182,8 @@ class HttpEndpointAdapter extends CallbackAdapter {
             body.set("audience", auth.audience.trim());
         }
         for (const [key, value] of Object.entries({
-            ...(auth?.additionalFormFields ?? {}),
-            ...(oauthOptions?.additionalFormFields ?? {})
+            ...(oauthOptions?.additionalFormFields ?? {}),
+            ...(auth?.additionalFormFields ?? {})
         })) {
             if (key.trim() && typeof value === "string") {
                 body.set(key.trim(), value);
@@ -1602,13 +1609,15 @@ class AzureEventHubsEndpointAdapter extends CallbackAdapter {
         const options = this.endpoint.azureEventHubs ?? {};
         const resolved = prepareProducedPayload(this.endpoint, payload);
         const wire = toWirePayload(resolved, this.endpoint);
-        const batch = await producer.createBatch();
+        const partitionId = resolveEventHubProducePartitionId(options);
+        const batch = partitionId
+            ? await producer.createBatch({ partitionId })
+            : await producer.createBatch();
         const eventData = {
             body: Buffer.from(wire.body),
             contentType: wire.contentType,
             properties: { ...wire.headers }
         };
-        const partitionId = optionString(options, "PartitionId", "partitionId");
         if (partitionId) {
             eventData.partitionId = partitionId;
         }
@@ -1835,6 +1844,10 @@ const AZURE_EVENT_HUBS_ENDPOINT_FLAT_KEYS = [
     "consumerGroup",
     "PartitionId",
     "partitionId",
+    "PartitionKey",
+    "partitionKey",
+    "PartitionCount",
+    "partitionCount",
     "StartFromEarliest",
     "startFromEarliest"
 ];
@@ -1947,7 +1960,11 @@ function resolveEndpointKind(raw) {
         "EventHubName",
         "eventHubName",
         "PartitionId",
-        "partitionId"
+        "partitionId",
+        "PartitionKey",
+        "partitionKey",
+        "PartitionCount",
+        "partitionCount"
     ])) {
         return "AzureEventHubs";
     }
@@ -2448,6 +2465,10 @@ function validateAzureEventHubsEndpoint(endpoint, mode, hasModeDelegate) {
     }
     requireNonEmptyString(optionString(options, "ConnectionString", "connectionString"), "ConnectionString must be provided for Azure Event Hubs endpoint.");
     requireNonEmptyString(optionString(options, "EventHubName", "eventHubName"), "EventHubName must be provided for Azure Event Hubs endpoint.");
+    const partitionCount = optionNumber(options, "PartitionCount", "partitionCount");
+    if (partitionCount < 0) {
+        throw new RangeError("PartitionCount must be zero or greater when configured.");
+    }
 }
 function validatePushDiffusionEndpoint(endpoint, mode) {
     const options = endpoint.pushDiffusion;
@@ -2847,6 +2868,17 @@ function partitionFromKey(value, partitionCount) {
     const normalized = Math.abs(hash) % count;
     return String(normalized);
 }
+function resolveEventHubProducePartitionId(options) {
+    const explicitPartitionId = optionString(options, "PartitionId", "partitionId");
+    if (explicitPartitionId) {
+        return explicitPartitionId;
+    }
+    const partitionKey = optionString(options, "PartitionKey", "partitionKey");
+    if (!partitionKey) {
+        return undefined;
+    }
+    return partitionFromKey(partitionKey, optionNumber(options, "PartitionCount", "partitionCount") || 4);
+}
 function attachPayloadHelpers(payload) {
     const target = {
         ...payload,
@@ -3029,10 +3061,10 @@ function deserializeBrokerPayloadBody(body, contentType, messagePayloadType) {
 function extractTrackingValue(payload, selector) {
     const normalized = selector.trim().toLowerCase();
     if (normalized.startsWith("header:")) {
-        const headerName = selector.slice("header:".length).trim().toLowerCase();
+        const headerName = selector.slice("header:".length).trim();
         for (const [key, value] of Object.entries(payload.headers ?? {})) {
             const resolved = String(value ?? "").trim();
-            if (key.toLowerCase() === headerName && resolved) {
+            if (key === headerName && resolved) {
                 return resolved;
             }
         }
@@ -3693,3 +3725,57 @@ function payloadBodyAsUtf8(body) {
         return String(body);
     }
 }
+exports.__loadstrikeTestExports = {
+    AzureEventHubsEndpointAdapter,
+    AzureEventHubsEndpointDefinition: exports.AzureEventHubsEndpointDefinition,
+    AzureEventHubsEndpointDefinitionModel,
+    CallbackAdapter,
+    DelegateStreamEndpointDefinition: exports.DelegateStreamEndpointDefinition,
+    DelegateStreamEndpointDefinitionModel,
+    HttpEndpointAdapter,
+    NatsEndpointDefinition: exports.NatsEndpointDefinition,
+    NatsEndpointDefinitionModel,
+    PushDiffusionEndpointDefinition: exports.PushDiffusionEndpointDefinition,
+    PushDiffusionEndpointDefinitionModel,
+    RabbitMqEndpointDefinition: exports.RabbitMqEndpointDefinition,
+    RabbitMqEndpointDefinitionModel,
+    RedisStreamsEndpointAdapter,
+    RedisStreamsEndpointDefinition: exports.RedisStreamsEndpointDefinition,
+    RedisStreamsEndpointDefinitionModel,
+    bufferToUint8Array,
+    applyHttpAuthHeaders,
+    buildHttpRequestBody,
+    buildConfluentKafkaClientOptions,
+    buildKafkaClientOptions,
+    canonicalizeHttpResponseSource,
+    createDelegateRequestEndpointView,
+    createNatsHeaders,
+    createRedisStreamPayload,
+    deserializeBrokerPayloadBody,
+    extractTrackingValue,
+    fromKafkaHeaders,
+    headerValue,
+    attachDotNetTrackingPayloadAliases,
+    attachPayloadHelpers,
+    attachStructuredProducedMessageRequestAliases,
+    inferLegacyHttpResponseSource,
+    injectTrackingValue,
+    mapConfluentKafkaSaslMechanism,
+    mapConfluentKafkaSecurityProtocol,
+    mapKafkaSaslMechanism,
+    parseBodyObject,
+    partitionFromKey,
+    payloadBodyAsUtf8,
+    prepareProducedPayload,
+    protocolBus: protocolBus,
+    readFirstRedisStreamEntry,
+    resolveConnectionMetadata,
+    resolveKafkaOAuthBearerToken,
+    serializePayloadBody,
+    setJsonBodyValue,
+    shouldUseConfluentKafkaClient,
+    toHeaderRecord,
+    toKafkaHeadersWithContentType,
+    validateHttpEndpoint,
+    validateTrackingSelectorPath
+};

package/dist/esm/cluster.js

@@ -401,3 +401,20 @@ function booleanOrDefault(value, fallback) {
 async function sleep(ms) {
     await new Promise((resolve) => setTimeout(resolve, ms));
 }
+export const __loadstrikeTestExports = {
+    DistributedClusterAgent,
+    DistributedClusterCoordinator,
+    LocalClusterCoordinator,
+    arrayOrUndefined,
+    booleanOrDefault,
+    buildWeightedCycle,
+    convertRunResult,
+    numberOrDefault,
+    parseRunCommand,
+    parseRunResult,
+    planAgentScenarioAssignments,
+    recordOrUndefined,
+    sanitizeToken,
+    sleep,
+    stringOrDefault
+};