@opentap/runner-client 4.1.0 → 4.3.0-beta.3.32232370336

package/dist/cjs/BaseClient.js

@@ -52,6 +52,28 @@ var nats_ws_1 = require("nats.ws");
 var DTOs_1 = require("./DTOs");
 var events_1 = require("events");
 var uuid_1 = require("uuid");
+// A browser extension like CrowdStrike may have already replaced globalThis.WebSocket before this module loads,
+// so reading the global is unreliable. A freshly created same-origin iframe exposes the browser's
+// pristine, un-patched WebSocket; NATS traffic is pinned to it to bypass any extension shim.
+// The iframe is kept attached because a detached iframe's realm becomes unusable in some browsers.
+function getNativeWebSocket() {
+    var _a, _b, _c;
+    var globalWebSocket = typeof globalThis !== 'undefined' ? globalThis.WebSocket : undefined;
+    var doc = globalThis.document;
+    if (!doc || typeof doc.createElement !== 'function') {
+        return globalWebSocket;
+    }
+    try {
+        var iframe = doc.createElement('iframe');
+        iframe.style.display = 'none';
+        ((_a = doc.body) !== null && _a !== void 0 ? _a : doc.documentElement).appendChild(iframe);
+        return (_c = (_b = iframe.contentWindow) === null || _b === void 0 ? void 0 : _b.WebSocket) !== null && _c !== void 0 ? _c : globalWebSocket;
+    }
+    catch (_d) {
+        return globalWebSocket;
+    }
+}
+var OriginalWebSocket = getNativeWebSocket();
 var DEFAULT_TIMEOUT = 40000; // default timeout of 40 seconds
 var Events;
 (function (Events) {
@@ -64,6 +86,15 @@ var BaseClient = /** @class */ (function () {
         this.baseSubject = baseSubject;
         this.connectionOptions = __assign({}, options);
         this.connectionOptions.timeout = (options === null || options === void 0 ? void 0 : options.timeout) || DEFAULT_TIMEOUT;
+        // Force NATS to use the WebSocket captured at load time so an extension that later
+        // overrides globalThis.WebSocket cannot intercept NATS traffic. Respect a caller-provided factory.
+        // wsFactory is read by nats.ws's WsTransport at runtime but lives on WsConnectionOptions, which
+        // isn't re-exported, so it's absent from the public ConnectionOptions type and requires this cast.
+        if (OriginalWebSocket && !this.connectionOptions.wsFactory) {
+            this.connectionOptions.wsFactory = function (url) {
+                return Promise.resolve({ socket: new OriginalWebSocket(url), encrypted: url.indexOf('wss://') === 0 });
+            };
+        }
         this.eventEmitter = new events_1.EventEmitter();
     }
     Object.defineProperty(BaseClient.prototype, "accessToken", {
@@ -102,8 +133,13 @@ var BaseClient = /** @class */ (function () {
         enumerable: false,
         configurable: true
     });
-    BaseClient.prototype.withTimeout = function (promise, timeout) {
-        return Promise.race([promise, new Promise(function (_, reject) { return setTimeout(function () { return reject(new Error(nats_ws_1.ErrorCode.Timeout)); }, timeout); })]);
+    BaseClient.prototype.withTimeout = function (promise, timeout, subject) {
+        return Promise.race([
+            promise,
+            new Promise(function (_, reject) {
+                return setTimeout(function () { return reject(new Error("".concat(nats_ws_1.ErrorCode.Timeout, ": '").concat(subject, "' timed out after ").concat(timeout, "ms"))); }, timeout);
+            }),
+        ]);
     };
     /**
      * Send a request to the nats server.
@@ -152,7 +188,14 @@ var BaseClient = /** @class */ (function () {
                 };
                 subscription = this.connection.subscribe(replySubject);
                 responsePromise = new Promise(function (resolve, reject) {
+                    // Chunk payloads stored at their sequence index so order is preserved
+                    // regardless of the order in which messages arrive.
                     var messages = [];
+                    var receivedCount = 0;
+                    // Total number of chunks and the size of the final chunk, both known
+                    // once the final (smaller) chunk arrives.
+                    var expectedCount;
+                    var finalChunkLength = 0;
                     subscription.callback = function (error, message) {
                         var _a, _b, _c, _d;
                         if (error) {
@@ -168,44 +211,50 @@ var BaseClient = /** @class */ (function () {
                             subscription.unsubscribe();
                             return resolve({ byteArray: message.data, isErrorResponse: ((_c = message.headers) === null || _c === void 0 ? void 0 : _c.get('OpenTapNatsError')) === 'true' });
                         }
-                        // Put all the response chunks in an array in the order they are received
-                        messages.push(message);
-                        // If the chunk has a size equal to the chunkSize, we should expect another message
-                        if (message.data.length === chunkSize) {
+                        // Place each chunk at its sequence position (ChunkNumber is 1-based).
+                        messages[parseInt(chunkNumber) - 1] = message.data;
+                        receivedCount += 1;
+                        // The final chunk is the only one smaller than chunkSize; its sequence
+                        // number tells us how many chunks to expect and its length fixes the total size.
+                        if (message.data.length < chunkSize) {
+                            expectedCount = parseInt(chunkNumber);
+                            finalChunkLength = message.data.length;
+                        }
+                        // Keep waiting until every expected chunk has been received.
+                        if (expectedCount === undefined || receivedCount < expectedCount) {
                             return;
                         }
-                        // If the chunk has a length smaller than the chunkSize, the message is complete
-                        // If the final message was received, we can safely unsubscribe
+                        // All expected chunks received; we can safely unsubscribe.
                         subscription.unsubscribe();
-                        // Check if the number of the final message is equal to the number of
-                        // messages we received. If this is not the case, we dropped a chunk,
-                        // likely due to a network error. In this case, the entire message is invalid.
-                        if (parseInt(chunkNumber) !== messages.length) {
-                            return reject(Error("Expected {finalMessageNumber} chunks, but received ".concat(messages.length, ". ") +
-                                "The connection may have been interrupted."));
-                        }
-                        // Concatenate the payloads
-                        // When there are many chunks, doing a single allocation
-                        // is significantly faster than concatenating arrays in sequence
-                        var flattenedSize = messages.reduce(function (sum, array) { return sum + array.data.length; }, 0);
+                        // Every chunk before the last is exactly chunkSize, so the total size is
+                        // deterministic and each chunk's offset is simply its index * chunkSize.
+                        var flattenedSize = (expectedCount - 1) * chunkSize + finalChunkLength;
                         var flattenedArray = new Uint8Array(flattenedSize);
-                        var k = 0;
-                        messages.forEach(function (m) {
-                            for (var i = 0; i < m.data.length; i++) {
-                                flattenedArray[k++] = m.data[i];
+                        for (var i = 0; i < expectedCount; i++) {
+                            var chunk_1 = messages[i];
+                            // A missing chunk means we dropped one, likely due to a network error,
+                            // so the entire message is invalid.
+                            if (!chunk_1) {
+                                return reject(Error("Expected ".concat(expectedCount, " chunks, but chunk ").concat(i + 1, " is missing. ") +
+                                    "The connection may have been interrupted."));
                             }
-                        });
+                            flattenedArray.set(chunk_1, i * chunkSize);
+                        }
                         return resolve({ byteArray: flattenedArray, isErrorResponse: ((_d = message.headers) === null || _d === void 0 ? void 0 : _d.get('OpenTapNatsError')) === 'true' });
                     };
                 }).then(function (_a) {
                     var byteArray = _a.byteArray, isErrorResponse = _a.isErrorResponse;
+                    var jsonCodec = (0, nats_ws_1.JSONCodec)();
+                    if (isErrorResponse) {
+                        return Promise.reject(DTOs_1.ErrorResponse.fromJS(jsonCodec.decode(byteArray)));
+                    }
                     if (byteArray.length === 0) {
                         return Promise.resolve(undefined);
                     }
-                    var jsonCodec = (0, nats_ws_1.JSONCodec)();
-                    // If a raw response is requested, we should avoid decoding the bytes.
-                    var response = (options === null || options === void 0 ? void 0 : options.rawResponse) ? byteArray : jsonCodec.decode(byteArray);
-                    return isErrorResponse ? Promise.reject(DTOs_1.ErrorResponse.fromJS(response)) : Promise.resolve(response);
+                    if (options === null || options === void 0 ? void 0 : options.rawResponse) {
+                        return Promise.resolve(byteArray);
+                    }
+                    return Promise.resolve(jsonCodec.decode(byteArray));
                 });
                 chunk = getChunk(0);
                 this.connection.publish(subject, chunk, opts);
@@ -223,7 +272,7 @@ var BaseClient = /** @class */ (function () {
                     this.connection.publish(subject, nats_ws_1.Empty, opts);
                 }
                 // Now that we have sent the terminating chunk, the result should arrive on our promise.
-                return [2 /*return*/, this.withTimeout(responsePromise, timeout).catch(function (err) {
+                return [2 /*return*/, this.withTimeout(responsePromise, timeout, subject).catch(function (err) {
                         subscription.unsubscribe();
                         return Promise.reject(_this.natsErrorHandler(err, subject));
                     })];

package/dist/cjs/DTOs.d.ts

@@ -876,7 +876,8 @@ export declare enum SessionState {
     Breaking = "Breaking",
     Aborting = "Aborting",
     WaitingForUserInput = "WaitingForUserInput",
-    Loading = "Loading"
+    Loading = "Loading",
+    Restarting = "Restarting"
 }
 export declare class TestPlan implements ITestPlan {
     childTestSteps?: TestStep[] | undefined;
@@ -1207,7 +1208,9 @@ export declare enum SessionEventName {
     TestStepChanged = "TestStepChanged",
     Break = "Break",
     UserInputRequested = "UserInputRequested",
-    UserInputRequestCompleted = "UserInputRequestCompleted"
+    UserInputRequestCompleted = "UserInputRequestCompleted",
+    RestartInitiated = "RestartInitiated",
+    RestartCompleted = "RestartCompleted"
 }
 export declare abstract class SessionEvent {
     readonly eventName: SessionEventName;
@@ -1315,6 +1318,21 @@ export declare class UserInputRequestCompletedEvent extends SessionEvent {
     static fromJS(data: any): UserInputRequestCompletedEvent;
     toJSON(data?: any): any;
 }
+export declare class RestartInitiatedEvent extends SessionEvent {
+    oldProcessId?: number;
+    constructor(data?: Partial<RestartInitiatedEvent>);
+    init(_data?: any): void;
+    static fromJS(data: any): RestartInitiatedEvent;
+    toJSON(data?: any): any;
+}
+export declare class RestartCompletedEvent extends SessionEvent {
+    success?: boolean;
+    newProcessId?: number;
+    constructor(data?: Partial<RestartCompletedEvent>);
+    init(_data?: any): void;
+    static fromJS(data: any): RestartCompletedEvent;
+    toJSON(data?: any): any;
+}
 export type ISessionEvent = SessionEvent;
 export interface IOnTestPlanRun {
     status?: string;

package/dist/cjs/DTOs.js

@@ -17,7 +17,7 @@ var __extends = (this && this.__extends) || (function () {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SettingsTapPackage = exports.SaveTestPlanToRepositoryResponse = exports.GetTestPlanReferenceResponse = exports.RepositorySettingsPackageDefinition = exports.RepositoryPackageDefinition = exports.RepositoryPackageReference = exports.ProfileGroup = exports.ListItemType = exports.ComponentSettings = exports.FileDescriptor = exports.Resource = exports.PluginTypeSelectorControl = exports.PictureControl = exports.PasswordControl = exports.MultiSelectControl = exports.DropdownControl = exports.EnabledControl = exports.DirectoryPathControl = exports.FilePathControl = exports.ComboBoxControl = exports.TextBoxControl = exports.DataGridControl = exports.DataGridReferenceControl = exports.CheckBoxControl = exports.AvailableValue = exports.ButtonsControl = exports.ButtonControl = exports.ExternalParameter = exports.MetaData = exports.DisplayAttribute = exports.VisualStatus = exports.UnitAttribute = exports.Icon = exports.ColumnDisplayName = exports.LayoutMode = exports.Layout = exports.Setting = exports.ComponentSettingsListItem = exports.AnnotatedObject = exports.ComponentSettingsList = exports.DisplayMode = exports.ComponentSettingsIdentifier = exports.ComponentSettingsBase = exports.Links = exports.Session = exports.ImageResolveErrorResponse = exports.NoResponderError = exports.ErrorResponse = exports.PackageSpecifier = exports.Image = void 0;
 exports.StartedEvent = exports.StartingEvent = exports.SessionTimeoutEvent = exports.TestPlanSettingsChangedEvent = exports.TypeCacheInvalidatedEvent = exports.SessionEvent = exports.SessionEventName = exports.MetricValue = exports.MetricsConfiguration = exports.SessionMetricInfo = exports.MetricsPollRate = exports.SessionMetricType = exports.SessionMetricKind = exports.ResultColumn = exports.Result = exports.LogEvent = exports.TestStepRun = exports.WatchDog = exports.Parameter = exports.BreakPoints = exports.InstalledFile = exports.Interaction = exports.CommonContext = exports.CommonSettings = exports.ValidationError = exports.TestStepValidationError = exports.TestStepCopy = exports.TestStepType = exports.TestStep = exports.TestPlan = exports.SessionState = exports.Verdict = exports.EditStatus = exports.RunStatus = exports.MissingLicenseResponse = exports.LogEntry = exports.LogList = exports.ApiException = exports.ImageCreationFailed = exports.ImageCreated = exports.ImageCreating = exports.ImageEventArgs = exports.SessionInactivityLimitHit = exports.SessionShutdownFailed = exports.SessionShutdown = exports.SessionShutdownInitiated = exports.SessionStartFailed = exports.SessionStarted = exports.SessionStartInitiated = exports.SessionEventArgs = void 0;
-exports.RunnerStatus = exports.StreamInfoStatus = exports.RemoteConnection = exports.SaveDefaultSettings = exports.AssetUpdatedEvent = exports.MetadataUpdatedEvent = exports.RunnerDeletedEvent = exports.RunnerUpdatedEvent = exports.RunnerRegisteredEvent = exports.RunnerEvent = exports.TestPlanRunCompletedEvent = exports.TestPlanRunStartEvent = exports.TestPlanRunEvent = exports.OnTestStepRun = exports.OnTestPlanRun = exports.UserInputRequestCompletedEvent = exports.UserInputRequestedEvent = exports.BreakEvent = exports.TestStepChangedEvent = exports.SettingsChangedEvent = exports.HeartbeatEvent = exports.SessionStateChangedEvent = exports.TestPlanReferenceChangedEvent = exports.TestPlanChangedEvent = exports.StoppedEvent = exports.StoppingEvent = void 0;
+exports.RunnerStatus = exports.StreamInfoStatus = exports.RemoteConnection = exports.SaveDefaultSettings = exports.AssetUpdatedEvent = exports.MetadataUpdatedEvent = exports.RunnerDeletedEvent = exports.RunnerUpdatedEvent = exports.RunnerRegisteredEvent = exports.RunnerEvent = exports.TestPlanRunCompletedEvent = exports.TestPlanRunStartEvent = exports.TestPlanRunEvent = exports.OnTestStepRun = exports.OnTestPlanRun = exports.RestartCompletedEvent = exports.RestartInitiatedEvent = exports.UserInputRequestCompletedEvent = exports.UserInputRequestedEvent = exports.BreakEvent = exports.TestStepChangedEvent = exports.SettingsChangedEvent = exports.HeartbeatEvent = exports.SessionStateChangedEvent = exports.TestPlanReferenceChangedEvent = exports.TestPlanChangedEvent = exports.StoppedEvent = exports.StoppingEvent = void 0;
 var Image = /** @class */ (function () {
     function Image(data) {
         this.packages = [];
@@ -2444,6 +2444,7 @@ var SessionState;
     SessionState["Aborting"] = "Aborting";
     SessionState["WaitingForUserInput"] = "WaitingForUserInput";
     SessionState["Loading"] = "Loading";
+    SessionState["Restarting"] = "Restarting";
 })(SessionState || (exports.SessionState = SessionState = {}));
 var TestPlan = /** @class */ (function () {
     function TestPlan(data) {
@@ -3326,6 +3327,8 @@ var SessionEventName;
     SessionEventName["Break"] = "Break";
     SessionEventName["UserInputRequested"] = "UserInputRequested";
     SessionEventName["UserInputRequestCompleted"] = "UserInputRequestCompleted";
+    SessionEventName["RestartInitiated"] = "RestartInitiated";
+    SessionEventName["RestartCompleted"] = "RestartCompleted";
 })(SessionEventName || (exports.SessionEventName = SessionEventName = {}));
 var SessionEvent = /** @class */ (function () {
     function SessionEvent(eventName) {
@@ -3659,6 +3662,60 @@ var UserInputRequestCompletedEvent = /** @class */ (function (_super) {
     return UserInputRequestCompletedEvent;
 }(SessionEvent));
 exports.UserInputRequestCompletedEvent = UserInputRequestCompletedEvent;
+var RestartInitiatedEvent = /** @class */ (function (_super) {
+    __extends(RestartInitiatedEvent, _super);
+    function RestartInitiatedEvent(data) {
+        var _this = _super.call(this, SessionEventName.RestartInitiated) || this;
+        if (data) {
+            Object.assign(_this, data);
+        }
+        return _this;
+    }
+    RestartInitiatedEvent.prototype.init = function (_data) {
+        var data = ensureObject(_data);
+        this.oldProcessId = data['OldProcessId'];
+    };
+    RestartInitiatedEvent.fromJS = function (data) {
+        var result = new RestartInitiatedEvent();
+        result.init(data);
+        return result;
+    };
+    RestartInitiatedEvent.prototype.toJSON = function (data) {
+        var json = ensureObject(data);
+        json['OldProcessId'] = this.oldProcessId;
+        return json;
+    };
+    return RestartInitiatedEvent;
+}(SessionEvent));
+exports.RestartInitiatedEvent = RestartInitiatedEvent;
+var RestartCompletedEvent = /** @class */ (function (_super) {
+    __extends(RestartCompletedEvent, _super);
+    function RestartCompletedEvent(data) {
+        var _this = _super.call(this, SessionEventName.RestartCompleted) || this;
+        if (data) {
+            Object.assign(_this, data);
+        }
+        return _this;
+    }
+    RestartCompletedEvent.prototype.init = function (_data) {
+        var data = ensureObject(_data);
+        this.success = data['Success'];
+        this.newProcessId = data['NewProcessId'];
+    };
+    RestartCompletedEvent.fromJS = function (data) {
+        var result = new RestartCompletedEvent();
+        result.init(data);
+        return result;
+    };
+    RestartCompletedEvent.prototype.toJSON = function (data) {
+        var json = ensureObject(data);
+        json['Success'] = this.success;
+        json['NewProcessId'] = this.newProcessId;
+        return json;
+    };
+    return RestartCompletedEvent;
+}(SessionEvent));
+exports.RestartCompletedEvent = RestartCompletedEvent;
 var OnTestPlanRun = /** @class */ (function () {
     function OnTestPlanRun(data) {
         if (data) {

package/dist/cjs/RunnerClient.d.ts

@@ -58,6 +58,12 @@ export declare class RunnerClient extends BaseClient {
      * @returns {{Promise<void>}}
      */
     shutdownSession(sessionId: string, force?: boolean): Promise<void>;
+    /**
+     * Restart a session
+     * @param sessionId the ID of the session to restart
+     * @returns {{Promise<void>}}
+     */
+    restartSession(sessionId: string): Promise<void>;
     /**
      * Start a session
      * @returns {{Promise<Session>}}

package/dist/cjs/RunnerClient.js

@@ -232,6 +232,16 @@ var RunnerClient = /** @class */ (function (_super) {
                 .catch(this.error())
             : Promise.reject('sessionId is not defined');
     };
+    /**
+     * Restart a session
+     * @param sessionId the ID of the session to restart
+     * @returns {{Promise<void>}}
+     */
+    RunnerClient.prototype.restartSession = function (sessionId) {
+        return (sessionId === null || sessionId === void 0 ? void 0 : sessionId.length) > 0
+            ? this.request('RestartSession', { SessionId: sessionId }).then(this.success()).catch(this.error())
+            : Promise.reject('sessionId is not defined');
+    };
     /**
      * Start a session
      * @returns {{Promise<Session>}}

package/dist/cjs/SessionClient.js

@@ -847,6 +847,8 @@ var SessionClient = /** @class */ (function (_super) {
         _a[DTOs_1.SessionEventName.Break] = function (payload) { return DTOs_1.BreakEvent.fromJS(payload); },
         _a[DTOs_1.SessionEventName.UserInputRequested] = function (payload) { return DTOs_1.UserInputRequestedEvent.fromJS(payload); },
         _a[DTOs_1.SessionEventName.UserInputRequestCompleted] = function (payload) { return DTOs_1.UserInputRequestCompletedEvent.fromJS(payload); },
+        _a[DTOs_1.SessionEventName.RestartInitiated] = function (payload) { return DTOs_1.RestartInitiatedEvent.fromJS(payload); },
+        _a[DTOs_1.SessionEventName.RestartCompleted] = function (payload) { return DTOs_1.RestartCompletedEvent.fromJS(payload); },
         _a);
     return SessionClient;
 }(BaseClient_1.BaseClient));

package/dist/mjs/BaseClient.js

@@ -11,6 +11,28 @@ import { AckPolicy, Empty, ErrorCode, JSONCodec, StringCodec, connect, headers,
 import { ComponentSettingsBase, ComponentSettingsIdentifier, ComponentSettingsListItem, DataGridControl, ErrorResponse, FileDescriptor, ListItemType, NoResponderError, ProfileGroup, } from './DTOs';
 import { EventEmitter } from 'events';
 import { v4 as uuidv4 } from 'uuid';
+// A browser extension like CrowdStrike may have already replaced globalThis.WebSocket before this module loads,
+// so reading the global is unreliable. A freshly created same-origin iframe exposes the browser's
+// pristine, un-patched WebSocket; NATS traffic is pinned to it to bypass any extension shim.
+// The iframe is kept attached because a detached iframe's realm becomes unusable in some browsers.
+function getNativeWebSocket() {
+    var _a, _b, _c;
+    const globalWebSocket = typeof globalThis !== 'undefined' ? globalThis.WebSocket : undefined;
+    const doc = globalThis.document;
+    if (!doc || typeof doc.createElement !== 'function') {
+        return globalWebSocket;
+    }
+    try {
+        const iframe = doc.createElement('iframe');
+        iframe.style.display = 'none';
+        ((_a = doc.body) !== null && _a !== void 0 ? _a : doc.documentElement).appendChild(iframe);
+        return (_c = (_b = iframe.contentWindow) === null || _b === void 0 ? void 0 : _b.WebSocket) !== null && _c !== void 0 ? _c : globalWebSocket;
+    }
+    catch (_d) {
+        return globalWebSocket;
+    }
+}
+const OriginalWebSocket = getNativeWebSocket();
 const DEFAULT_TIMEOUT = 40000; // default timeout of 40 seconds
 var Events;
 (function (Events) {
@@ -47,10 +69,20 @@ export class BaseClient {
         this.baseSubject = baseSubject;
         this.connectionOptions = Object.assign({}, options);
         this.connectionOptions.timeout = (options === null || options === void 0 ? void 0 : options.timeout) || DEFAULT_TIMEOUT;
+        // Force NATS to use the WebSocket captured at load time so an extension that later
+        // overrides globalThis.WebSocket cannot intercept NATS traffic. Respect a caller-provided factory.
+        // wsFactory is read by nats.ws's WsTransport at runtime but lives on WsConnectionOptions, which
+        // isn't re-exported, so it's absent from the public ConnectionOptions type and requires this cast.
+        if (OriginalWebSocket && !this.connectionOptions.wsFactory) {
+            this.connectionOptions.wsFactory = (url) => Promise.resolve({ socket: new OriginalWebSocket(url), encrypted: url.indexOf('wss://') === 0 });
+        }
         this.eventEmitter = new EventEmitter();
     }
-    withTimeout(promise, timeout) {
-        return Promise.race([promise, new Promise((_, reject) => setTimeout(() => reject(new Error(ErrorCode.Timeout)), timeout))]);
+    withTimeout(promise, timeout, subject) {
+        return Promise.race([
+            promise,
+            new Promise((_, reject) => setTimeout(() => reject(new Error(`${ErrorCode.Timeout}: '${subject}' timed out after ${timeout}ms`)), timeout)),
+        ]);
     }
     /**
      * Send a request to the nats server.
@@ -106,7 +138,14 @@ export class BaseClient {
             // The response from the session will arrive in this promise after we have finished
             // publishing chunks
             const responsePromise = new Promise((resolve, reject) => {
+                // Chunk payloads stored at their sequence index so order is preserved
+                // regardless of the order in which messages arrive.
                 const messages = [];
+                let receivedCount = 0;
+                // Total number of chunks and the size of the final chunk, both known
+                // once the final (smaller) chunk arrives.
+                let expectedCount;
+                let finalChunkLength = 0;
                 subscription.callback = (error, message) => {
                     var _a, _b, _c, _d;
                     if (error) {
@@ -122,43 +161,49 @@ export class BaseClient {
                         subscription.unsubscribe();
                         return resolve({ byteArray: message.data, isErrorResponse: ((_c = message.headers) === null || _c === void 0 ? void 0 : _c.get('OpenTapNatsError')) === 'true' });
                     }
-                    // Put all the response chunks in an array in the order they are received
-                    messages.push(message);
-                    // If the chunk has a size equal to the chunkSize, we should expect another message
-                    if (message.data.length === chunkSize) {
+                    // Place each chunk at its sequence position (ChunkNumber is 1-based).
+                    messages[parseInt(chunkNumber) - 1] = message.data;
+                    receivedCount += 1;
+                    // The final chunk is the only one smaller than chunkSize; its sequence
+                    // number tells us how many chunks to expect and its length fixes the total size.
+                    if (message.data.length < chunkSize) {
+                        expectedCount = parseInt(chunkNumber);
+                        finalChunkLength = message.data.length;
+                    }
+                    // Keep waiting until every expected chunk has been received.
+                    if (expectedCount === undefined || receivedCount < expectedCount) {
                         return;
                     }
-                    // If the chunk has a length smaller than the chunkSize, the message is complete
-                    // If the final message was received, we can safely unsubscribe
+                    // All expected chunks received; we can safely unsubscribe.
                     subscription.unsubscribe();
-                    // Check if the number of the final message is equal to the number of
-                    // messages we received. If this is not the case, we dropped a chunk,
-                    // likely due to a network error. In this case, the entire message is invalid.
-                    if (parseInt(chunkNumber) !== messages.length) {
-                        return reject(Error(`Expected {finalMessageNumber} chunks, but received ${messages.length}. ` +
-                            `The connection may have been interrupted.`));
-                    }
-                    // Concatenate the payloads
-                    // When there are many chunks, doing a single allocation
-                    // is significantly faster than concatenating arrays in sequence
-                    const flattenedSize = messages.reduce((sum, array) => sum + array.data.length, 0);
+                    // Every chunk before the last is exactly chunkSize, so the total size is
+                    // deterministic and each chunk's offset is simply its index * chunkSize.
+                    const flattenedSize = (expectedCount - 1) * chunkSize + finalChunkLength;
                     const flattenedArray = new Uint8Array(flattenedSize);
-                    let k = 0;
-                    messages.forEach(m => {
-                        for (let i = 0; i < m.data.length; i++) {
-                            flattenedArray[k++] = m.data[i];
+                    for (let i = 0; i < expectedCount; i++) {
+                        const chunk = messages[i];
+                        // A missing chunk means we dropped one, likely due to a network error,
+                        // so the entire message is invalid.
+                        if (!chunk) {
+                            return reject(Error(`Expected ${expectedCount} chunks, but chunk ${i + 1} is missing. ` +
+                                `The connection may have been interrupted.`));
                         }
-                    });
+                        flattenedArray.set(chunk, i * chunkSize);
+                    }
                     return resolve({ byteArray: flattenedArray, isErrorResponse: ((_d = message.headers) === null || _d === void 0 ? void 0 : _d.get('OpenTapNatsError')) === 'true' });
                 };
             }).then(({ byteArray, isErrorResponse }) => {
+                const jsonCodec = JSONCodec();
+                if (isErrorResponse) {
+                    return Promise.reject(ErrorResponse.fromJS(jsonCodec.decode(byteArray)));
+                }
                 if (byteArray.length === 0) {
                     return Promise.resolve(undefined);
                 }
-                const jsonCodec = JSONCodec();
-                // If a raw response is requested, we should avoid decoding the bytes.
-                const response = (options === null || options === void 0 ? void 0 : options.rawResponse) ? byteArray : jsonCodec.decode(byteArray);
-                return isErrorResponse ? Promise.reject(ErrorResponse.fromJS(response)) : Promise.resolve(response);
+                if (options === null || options === void 0 ? void 0 : options.rawResponse) {
+                    return Promise.resolve(byteArray);
+                }
+                return Promise.resolve(jsonCodec.decode(byteArray));
             });
             // 'Manually' publish the first chunk. This is to ensure that **at least** one chunk is published.
             // This is a special case for empty payloads (which are common)
@@ -178,7 +223,7 @@ export class BaseClient {
                 this.connection.publish(subject, Empty, opts);
             }
             // Now that we have sent the terminating chunk, the result should arrive on our promise.
-            return this.withTimeout(responsePromise, timeout).catch(err => {
+            return this.withTimeout(responsePromise, timeout, subject).catch(err => {
                 subscription.unsubscribe();
                 return Promise.reject(this.natsErrorHandler(err, subject));
             });

package/dist/mjs/DTOs.d.ts

@@ -876,7 +876,8 @@ export declare enum SessionState {
     Breaking = "Breaking",
     Aborting = "Aborting",
     WaitingForUserInput = "WaitingForUserInput",
-    Loading = "Loading"
+    Loading = "Loading",
+    Restarting = "Restarting"
 }
 export declare class TestPlan implements ITestPlan {
     childTestSteps?: TestStep[] | undefined;
@@ -1207,7 +1208,9 @@ export declare enum SessionEventName {
     TestStepChanged = "TestStepChanged",
     Break = "Break",
     UserInputRequested = "UserInputRequested",
-    UserInputRequestCompleted = "UserInputRequestCompleted"
+    UserInputRequestCompleted = "UserInputRequestCompleted",
+    RestartInitiated = "RestartInitiated",
+    RestartCompleted = "RestartCompleted"
 }
 export declare abstract class SessionEvent {
     readonly eventName: SessionEventName;
@@ -1315,6 +1318,21 @@ export declare class UserInputRequestCompletedEvent extends SessionEvent {
     static fromJS(data: any): UserInputRequestCompletedEvent;
     toJSON(data?: any): any;
 }
+export declare class RestartInitiatedEvent extends SessionEvent {
+    oldProcessId?: number;
+    constructor(data?: Partial<RestartInitiatedEvent>);
+    init(_data?: any): void;
+    static fromJS(data: any): RestartInitiatedEvent;
+    toJSON(data?: any): any;
+}
+export declare class RestartCompletedEvent extends SessionEvent {
+    success?: boolean;
+    newProcessId?: number;
+    constructor(data?: Partial<RestartCompletedEvent>);
+    init(_data?: any): void;
+    static fromJS(data: any): RestartCompletedEvent;
+    toJSON(data?: any): any;
+}
 export type ISessionEvent = SessionEvent;
 export interface IOnTestPlanRun {
     status?: string;

package/dist/mjs/DTOs.js

@@ -2143,6 +2143,7 @@ export var SessionState;
     SessionState["Aborting"] = "Aborting";
     SessionState["WaitingForUserInput"] = "WaitingForUserInput";
     SessionState["Loading"] = "Loading";
+    SessionState["Restarting"] = "Restarting";
 })(SessionState || (SessionState = {}));
 export class TestPlan {
     constructor(data) {
@@ -2925,6 +2926,8 @@ export var SessionEventName;
     SessionEventName["Break"] = "Break";
     SessionEventName["UserInputRequested"] = "UserInputRequested";
     SessionEventName["UserInputRequestCompleted"] = "UserInputRequestCompleted";
+    SessionEventName["RestartInitiated"] = "RestartInitiated";
+    SessionEventName["RestartCompleted"] = "RestartCompleted";
 })(SessionEventName || (SessionEventName = {}));
 export class SessionEvent {
     constructor(eventName) {
@@ -3200,6 +3203,52 @@ export class UserInputRequestCompletedEvent extends SessionEvent {
         return json;
     }
 }
+export class RestartInitiatedEvent extends SessionEvent {
+    constructor(data) {
+        super(SessionEventName.RestartInitiated);
+        if (data) {
+            Object.assign(this, data);
+        }
+    }
+    init(_data) {
+        const data = ensureObject(_data);
+        this.oldProcessId = data['OldProcessId'];
+    }
+    static fromJS(data) {
+        const result = new RestartInitiatedEvent();
+        result.init(data);
+        return result;
+    }
+    toJSON(data) {
+        const json = ensureObject(data);
+        json['OldProcessId'] = this.oldProcessId;
+        return json;
+    }
+}
+export class RestartCompletedEvent extends SessionEvent {
+    constructor(data) {
+        super(SessionEventName.RestartCompleted);
+        if (data) {
+            Object.assign(this, data);
+        }
+    }
+    init(_data) {
+        const data = ensureObject(_data);
+        this.success = data['Success'];
+        this.newProcessId = data['NewProcessId'];
+    }
+    static fromJS(data) {
+        const result = new RestartCompletedEvent();
+        result.init(data);
+        return result;
+    }
+    toJSON(data) {
+        const json = ensureObject(data);
+        json['Success'] = this.success;
+        json['NewProcessId'] = this.newProcessId;
+        return json;
+    }
+}
 export class OnTestPlanRun {
     constructor(data) {
         if (data) {

package/dist/mjs/RunnerClient.d.ts

@@ -58,6 +58,12 @@ export declare class RunnerClient extends BaseClient {
      * @returns {{Promise<void>}}
      */
     shutdownSession(sessionId: string, force?: boolean): Promise<void>;
+    /**
+     * Restart a session
+     * @param sessionId the ID of the session to restart
+     * @returns {{Promise<void>}}
+     */
+    restartSession(sessionId: string): Promise<void>;
     /**
      * Start a session
      * @returns {{Promise<Session>}}

package/dist/mjs/RunnerClient.js

@@ -171,6 +171,16 @@ export class RunnerClient extends BaseClient {
                 .catch(this.error())
             : Promise.reject('sessionId is not defined');
     }
+    /**
+     * Restart a session
+     * @param sessionId the ID of the session to restart
+     * @returns {{Promise<void>}}
+     */
+    restartSession(sessionId) {
+        return (sessionId === null || sessionId === void 0 ? void 0 : sessionId.length) > 0
+            ? this.request('RestartSession', { SessionId: sessionId }).then(this.success()).catch(this.error())
+            : Promise.reject('sessionId is not defined');
+    }
     /**
      * Start a session
      * @returns {{Promise<Session>}}

package/dist/mjs/SessionClient.js

@@ -1,4 +1,4 @@
-import { BreakEvent, BreakPoints, CommonContext, CommonSettings, DataGridControl, GetTestPlanReferenceResponse, HeartbeatEvent, Image, Interaction, ListItemType, LogList, MetricsConfiguration, OnTestPlanRun, OnTestStepRun, Result, RunStatus, SaveTestPlanToRepositoryResponse, SessionEventName, SessionStateChangedEvent, SessionTimeoutEvent, Setting, SettingsChangedEvent, StartingEvent, StartedEvent, StoppedEvent, StoppingEvent, TestPlan, TestPlanChangedEvent, TestPlanReferenceChangedEvent, TestPlanSettingsChangedEvent, TestStepChangedEvent, TestStepType, TestStepValidationError, TypeCacheInvalidatedEvent, UserInputRequestCompletedEvent, UserInputRequestedEvent, WatchDog, EditStatus, } from './DTOs';
+import { BreakEvent, BreakPoints, CommonContext, CommonSettings, DataGridControl, GetTestPlanReferenceResponse, HeartbeatEvent, Image, Interaction, ListItemType, LogList, MetricsConfiguration, OnTestPlanRun, OnTestStepRun, Result, RunStatus, SaveTestPlanToRepositoryResponse, SessionEventName, SessionStateChangedEvent, SessionTimeoutEvent, Setting, SettingsChangedEvent, StartingEvent, StartedEvent, StoppedEvent, StoppingEvent, TestPlan, TestPlanChangedEvent, TestPlanReferenceChangedEvent, TestPlanSettingsChangedEvent, TestStepChangedEvent, TestStepType, TestStepValidationError, TypeCacheInvalidatedEvent, UserInputRequestCompletedEvent, UserInputRequestedEvent, RestartInitiatedEvent, RestartCompletedEvent, WatchDog, EditStatus, } from './DTOs';
 import { BaseClient } from './BaseClient';
 import { jsonCodec } from './encoders';
 import { NIL } from 'uuid';
@@ -808,4 +808,6 @@ SessionClient.sessionEventFactories = {
     [SessionEventName.Break]: payload => BreakEvent.fromJS(payload),
     [SessionEventName.UserInputRequested]: payload => UserInputRequestedEvent.fromJS(payload),
     [SessionEventName.UserInputRequestCompleted]: payload => UserInputRequestCompletedEvent.fromJS(payload),
+    [SessionEventName.RestartInitiated]: payload => RestartInitiatedEvent.fromJS(payload),
+    [SessionEventName.RestartCompleted]: payload => RestartCompletedEvent.fromJS(payload),
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@opentap/runner-client",
-  "version": "4.1.0",
+  "version": "4.3.0-beta.3.32232370336",
   "description": "This is the web client for the OpenTAP Runner.",
   "main": "./dist/cjs/index.js",
   "module": "./dist/mjs/index.js",