@loadstrike/loadstrike-sdk 0.1.10001 → 1.0.10801

package/dist/esm/sinks.js

@@ -235,13 +235,8 @@ export 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) {
@@ -283,13 +278,9 @@ export 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);
     }
@@ -358,17 +349,6 @@ export 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;
@@ -454,12 +434,6 @@ export 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));
     }
@@ -569,12 +543,6 @@ export 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));
     }
@@ -720,12 +688,6 @@ export 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));
     }
@@ -1001,12 +963,6 @@ export 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));
     }
@@ -1129,12 +1085,6 @@ export 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));
     }
@@ -1237,12 +1187,6 @@ export 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));
     }
@@ -1356,6 +1300,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"
@@ -1450,7 +1395,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,
@@ -1583,30 +1529,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({
@@ -2578,28 +2557,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 ?? {})
         }))
     };
 }
@@ -2608,20 +2587,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 LoadStrikePluginDataModel(plugin.pluginName, (plugin.tables ?? []).map((table) => new LoadStrikePluginDataTableModel(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 LoadStrikePluginDataModel(plugin.pluginName, plugin.tables.map((table) => new LoadStrikePluginDataTableModel(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 LoadStrikePluginDataModel(plugin.pluginName, (plugin.tables ?? []).map((table) => new LoadStrikePluginDataTableModel(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 LoadStrikePluginDataModel(plugin.pluginName, (plugin.tables ?? []).map((table) => new LoadStrikePluginDataTableModel(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) {
@@ -2655,3 +2705,47 @@ async function postWithTimeout(fetchImpl, url, init, timeoutMs, sinkName) {
         clearTimeout(timer);
     }
 }
+export const __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/esm/transports.js

@@ -211,6 +211,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 {
@@ -614,6 +617,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);
@@ -704,10 +713,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.");
@@ -897,8 +905,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,
@@ -1124,10 +1131,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()
@@ -1139,8 +1146,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);
@@ -1566,13 +1573,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;
         }
@@ -1798,6 +1807,10 @@ const AZURE_EVENT_HUBS_ENDPOINT_FLAT_KEYS = [
     "consumerGroup",
     "PartitionId",
     "partitionId",
+    "PartitionKey",
+    "partitionKey",
+    "PartitionCount",
+    "partitionCount",
     "StartFromEarliest",
     "startFromEarliest"
 ];
@@ -1910,7 +1923,11 @@ function resolveEndpointKind(raw) {
         "EventHubName",
         "eventHubName",
         "PartitionId",
-        "partitionId"
+        "partitionId",
+        "PartitionKey",
+        "partitionKey",
+        "PartitionCount",
+        "partitionCount"
     ])) {
         return "AzureEventHubs";
     }
@@ -2411,6 +2428,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;
@@ -2810,6 +2831,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,
@@ -2992,10 +3024,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;
             }
         }
@@ -3656,3 +3688,57 @@ function payloadBodyAsUtf8(body) {
         return String(body);
     }
 }
+export const __loadstrikeTestExports = {
+    AzureEventHubsEndpointAdapter,
+    AzureEventHubsEndpointDefinition,
+    AzureEventHubsEndpointDefinitionModel,
+    CallbackAdapter,
+    DelegateStreamEndpointDefinition,
+    DelegateStreamEndpointDefinitionModel,
+    HttpEndpointAdapter,
+    NatsEndpointDefinition,
+    NatsEndpointDefinitionModel,
+    PushDiffusionEndpointDefinition,
+    PushDiffusionEndpointDefinitionModel,
+    RabbitMqEndpointDefinition,
+    RabbitMqEndpointDefinitionModel,
+    RedisStreamsEndpointAdapter,
+    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/types/cluster.d.ts

@@ -60,6 +60,7 @@ export interface ClusterNodeStatsPayload {
     pluginsData?: unknown[];
     nodeInfo?: Record<string, unknown>;
     testInfo?: Record<string, unknown>;
+    logFiles?: string[];
 }
 export interface ClusterRunResultMessage {
     commandId: string;
@@ -110,3 +111,32 @@ export declare class DistributedClusterAgent {
     dispose(): Promise<void>;
     pollAndExecuteOnce(execute: (dispatch: ClusterScenarioDispatch) => Promise<ClusterNodeResult> | ClusterNodeResult): Promise<boolean>;
 }
+declare function parseRunCommand(value: Record<string, unknown>): ClusterRunCommandMessage;
+declare function parseRunResult(value: Record<string, unknown>): ClusterRunResultMessage;
+declare function convertRunResult(result: ClusterRunResultMessage): ClusterNodeResult;
+declare function sanitizeToken(value: string): string;
+declare function buildWeightedCycle(weights: number[]): number[];
+declare function stringOrDefault(value: unknown, fallback: string): string;
+declare function numberOrDefault(value: unknown, fallback: number): number;
+declare function recordOrUndefined(value: unknown): Record<string, unknown> | undefined;
+declare function arrayOrUndefined(value: unknown): unknown[] | undefined;
+declare function booleanOrDefault(value: unknown, fallback: boolean): boolean;
+declare function sleep(ms: number): Promise<void>;
+export declare const __loadstrikeTestExports: {
+    DistributedClusterAgent: typeof DistributedClusterAgent;
+    DistributedClusterCoordinator: typeof DistributedClusterCoordinator;
+    LocalClusterCoordinator: typeof LocalClusterCoordinator;
+    arrayOrUndefined: typeof arrayOrUndefined;
+    booleanOrDefault: typeof booleanOrDefault;
+    buildWeightedCycle: typeof buildWeightedCycle;
+    convertRunResult: typeof convertRunResult;
+    numberOrDefault: typeof numberOrDefault;
+    parseRunCommand: typeof parseRunCommand;
+    parseRunResult: typeof parseRunResult;
+    planAgentScenarioAssignments: typeof planAgentScenarioAssignments;
+    recordOrUndefined: typeof recordOrUndefined;
+    sanitizeToken: typeof sanitizeToken;
+    sleep: typeof sleep;
+    stringOrDefault: typeof stringOrDefault;
+};
+export {};