testServerConnection.js 7.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218
  1. "use strict";
  2. var __create = Object.create;
  3. var __defProp = Object.defineProperty;
  4. var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
  5. var __getOwnPropNames = Object.getOwnPropertyNames;
  6. var __getProtoOf = Object.getPrototypeOf;
  7. var __hasOwnProp = Object.prototype.hasOwnProperty;
  8. var __export = (target, all) => {
  9. for (var name in all)
  10. __defProp(target, name, { get: all[name], enumerable: true });
  11. };
  12. var __copyProps = (to, from, except, desc) => {
  13. if (from && typeof from === "object" || typeof from === "function") {
  14. for (let key of __getOwnPropNames(from))
  15. if (!__hasOwnProp.call(to, key) && key !== except)
  16. __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
  17. }
  18. return to;
  19. };
  20. var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
  21. // If the importer is in node compatibility mode or this is not an ESM
  22. // file that has been converted to a CommonJS file using a Babel-
  23. // compatible transform (i.e. "__esModule" has not been set), then set
  24. // "default" to the CommonJS "module.exports" for node compatibility.
  25. isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
  26. mod
  27. ));
  28. var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
  29. var testServerConnection_exports = {};
  30. __export(testServerConnection_exports, {
  31. TestServerConnection: () => TestServerConnection,
  32. WebSocketTestServerTransport: () => WebSocketTestServerTransport
  33. });
  34. module.exports = __toCommonJS(testServerConnection_exports);
  35. var events = __toESM(require("./events"));
  36. class WebSocketTestServerTransport {
  37. constructor(url) {
  38. this._ws = new WebSocket(url);
  39. }
  40. onmessage(listener) {
  41. this._ws.addEventListener("message", (event) => listener(event.data.toString()));
  42. }
  43. onopen(listener) {
  44. this._ws.addEventListener("open", listener);
  45. }
  46. onerror(listener) {
  47. this._ws.addEventListener("error", listener);
  48. }
  49. onclose(listener) {
  50. this._ws.addEventListener("close", listener);
  51. }
  52. send(data) {
  53. this._ws.send(data);
  54. }
  55. close() {
  56. this._ws.close();
  57. }
  58. }
  59. class TestServerConnection {
  60. constructor(transport) {
  61. this._onCloseEmitter = new events.EventEmitter();
  62. this._onReportEmitter = new events.EventEmitter();
  63. this._onStdioEmitter = new events.EventEmitter();
  64. this._onTestFilesChangedEmitter = new events.EventEmitter();
  65. this._onLoadTraceRequestedEmitter = new events.EventEmitter();
  66. this._onRecoverFromStepErrorEmitter = new events.EventEmitter();
  67. this._lastId = 0;
  68. this._callbacks = /* @__PURE__ */ new Map();
  69. this._isClosed = false;
  70. this.onClose = this._onCloseEmitter.event;
  71. this.onReport = this._onReportEmitter.event;
  72. this.onStdio = this._onStdioEmitter.event;
  73. this.onTestFilesChanged = this._onTestFilesChangedEmitter.event;
  74. this.onLoadTraceRequested = this._onLoadTraceRequestedEmitter.event;
  75. this.onRecoverFromStepError = this._onRecoverFromStepErrorEmitter.event;
  76. this._transport = transport;
  77. this._transport.onmessage((data) => {
  78. const message = JSON.parse(data);
  79. const { id, result, error, method, params } = message;
  80. if (id) {
  81. const callback = this._callbacks.get(id);
  82. if (!callback)
  83. return;
  84. this._callbacks.delete(id);
  85. if (error)
  86. callback.reject(new Error(error));
  87. else
  88. callback.resolve(result);
  89. } else {
  90. this._dispatchEvent(method, params);
  91. }
  92. });
  93. const pingInterval = setInterval(() => this._sendMessage("ping").catch(() => {
  94. }), 3e4);
  95. this._connectedPromise = new Promise((f, r) => {
  96. this._transport.onopen(f);
  97. this._transport.onerror(r);
  98. });
  99. this._transport.onclose(() => {
  100. this._isClosed = true;
  101. this._onCloseEmitter.fire();
  102. clearInterval(pingInterval);
  103. });
  104. }
  105. isClosed() {
  106. return this._isClosed;
  107. }
  108. async _sendMessage(method, params) {
  109. const logForTest = globalThis.__logForTest;
  110. logForTest?.({ method, params });
  111. await this._connectedPromise;
  112. const id = ++this._lastId;
  113. const message = { id, method, params };
  114. this._transport.send(JSON.stringify(message));
  115. return new Promise((resolve, reject) => {
  116. this._callbacks.set(id, { resolve, reject });
  117. });
  118. }
  119. _sendMessageNoReply(method, params) {
  120. this._sendMessage(method, params).catch(() => {
  121. });
  122. }
  123. _dispatchEvent(method, params) {
  124. if (method === "report")
  125. this._onReportEmitter.fire(params);
  126. else if (method === "stdio")
  127. this._onStdioEmitter.fire(params);
  128. else if (method === "testFilesChanged")
  129. this._onTestFilesChangedEmitter.fire(params);
  130. else if (method === "loadTraceRequested")
  131. this._onLoadTraceRequestedEmitter.fire(params);
  132. else if (method === "recoverFromStepError")
  133. this._onRecoverFromStepErrorEmitter.fire(params);
  134. }
  135. async initialize(params) {
  136. await this._sendMessage("initialize", params);
  137. }
  138. async ping(params) {
  139. await this._sendMessage("ping", params);
  140. }
  141. async pingNoReply(params) {
  142. this._sendMessageNoReply("ping", params);
  143. }
  144. async watch(params) {
  145. await this._sendMessage("watch", params);
  146. }
  147. watchNoReply(params) {
  148. this._sendMessageNoReply("watch", params);
  149. }
  150. async open(params) {
  151. await this._sendMessage("open", params);
  152. }
  153. openNoReply(params) {
  154. this._sendMessageNoReply("open", params);
  155. }
  156. async resizeTerminal(params) {
  157. await this._sendMessage("resizeTerminal", params);
  158. }
  159. resizeTerminalNoReply(params) {
  160. this._sendMessageNoReply("resizeTerminal", params);
  161. }
  162. async checkBrowsers(params) {
  163. return await this._sendMessage("checkBrowsers", params);
  164. }
  165. async installBrowsers(params) {
  166. await this._sendMessage("installBrowsers", params);
  167. }
  168. async runGlobalSetup(params) {
  169. return await this._sendMessage("runGlobalSetup", params);
  170. }
  171. async runGlobalTeardown(params) {
  172. return await this._sendMessage("runGlobalTeardown", params);
  173. }
  174. async startDevServer(params) {
  175. return await this._sendMessage("startDevServer", params);
  176. }
  177. async stopDevServer(params) {
  178. return await this._sendMessage("stopDevServer", params);
  179. }
  180. async clearCache(params) {
  181. return await this._sendMessage("clearCache", params);
  182. }
  183. async listFiles(params) {
  184. return await this._sendMessage("listFiles", params);
  185. }
  186. async listTests(params) {
  187. return await this._sendMessage("listTests", params);
  188. }
  189. async runTests(params) {
  190. return await this._sendMessage("runTests", params);
  191. }
  192. async findRelatedTestFiles(params) {
  193. return await this._sendMessage("findRelatedTestFiles", params);
  194. }
  195. async stopTests(params) {
  196. await this._sendMessage("stopTests", params);
  197. }
  198. stopTestsNoReply(params) {
  199. this._sendMessageNoReply("stopTests", params);
  200. }
  201. async closeGracefully(params) {
  202. await this._sendMessage("closeGracefully", params);
  203. }
  204. async resumeAfterStepError(params) {
  205. await this._sendMessage("resumeAfterStepError", params);
  206. }
  207. close() {
  208. try {
  209. this._transport.close();
  210. } catch {
  211. }
  212. }
  213. }
  214. // Annotate the CommonJS export names for ESM import in node:
  215. 0 && (module.exports = {
  216. TestServerConnection,
  217. WebSocketTestServerTransport
  218. });