test: fix

Author: alexander-akait

client-src/index.js

@@ -721,3 +721,5 @@ const createSocketURL = (parsedURL) => {
 const socketURL = createSocketURL(parsedResourceQuery);
 
 socket(socketURL, onSocketMessage, options.reconnect);
+
+export { getCurrentScriptSource, parseURL, createSocketURL };

lib/Server.js

@@ -404,6 +404,7 @@ class Server {
       let host;
 
       const networks = Object.values(os.networkInterfaces())
+        // eslint-disable-next-line no-shadow
         .flatMap((networks) => networks ?? [])
         .filter((network) => {
           if (!network || !network.address) {

test/client/utils/createSocketURL.test.js

@@ -5,8 +5,10 @@
 "use strict";
 
 describe("'createSocketURL' function ", () => {
+  global.__webpack_hash__ = "hash";
+
   const samples = [
-    // __resourceQuery, location and socket URL
+    // // __resourceQuery, location and socket URL
     [
       "?hostname=example.com&pathname=/ws",
       "http://example.com",
@@ -103,16 +105,20 @@ describe("'createSocketURL' function ", () => {
   ];
 
   samples.forEach(([__resourceQuery, location, expected]) => {
-    jest.doMock(
-      "../../../client-src/utils/getCurrentScriptSource",
-      () => () => new URL("./entry.js", location).toString(),
-    );
+    test(`should return '${expected}' socket URL when '__resourceQuery' is '${__resourceQuery}' and 'self.location' is '${location}'`, () => {
+      global.__resourceQuery = __resourceQuery;
 
-    const createSocketURL =
-      require("../../../client-src/utils/createSocketURL").default;
-    const parseURL = require("../../../client-src/utils/parseURL").default;
+      if (__resourceQuery === null) {
+        Object.defineProperty(document, "currentScript", {
+          value: document.createElement("script"),
+          configurable: true,
+        });
+      }
+
+      const client = require("../../../client-src/index");
+      const createSocketURL = client.createSocketURL;
+      const parseURL = client.parseURL;
 
-    test(`should return '${expected}' socket URL when '__resourceQuery' is '${__resourceQuery}' and 'self.location' is '${location}'`, () => {
       const selfLocation = new URL(location);
 
       delete window.location;
@@ -121,6 +127,13 @@ describe("'createSocketURL' function ", () => {
 
       const parsedURL = parseURL(__resourceQuery);
 
+      if (__resourceQuery === null) {
+        Object.defineProperty(document, "currentScript", {
+          value: null,
+          configurable: true,
+        });
+      }
+
       expect(createSocketURL(parsedURL)).toBe(expected);
     });
 

test/client/utils/getCurrentScriptSource.test.js

@@ -4,10 +4,17 @@
 
 "use strict";
 
-const getCurrentScriptSource =
-  require("../../../client-src/utils/getCurrentScriptSource").default;
-
 describe("'getCurrentScriptSource' function", () => {
+  let getCurrentScriptSource;
+
+  beforeEach(() => {
+    global.__webpack_hash__ = "mock-hash";
+    global.__resourceQuery = "?protocol=ws&hostname=0.0.0.0";
+
+    getCurrentScriptSource =
+      require("../../../client-src/index").getCurrentScriptSource;
+  });
+
   afterEach(() => {
     Object.defineProperty(document, "currentScript", {
       // eslint-disable-next-line no-undefined