Added test cases with custom exports/imports.

Napalys · Napalys · commit c7fad0966478 · 2025-04-04T08:33:26.000+02:00
diff --git a/javascript/ql/test/library-tests/frameworks/WebSocket/browser-custom.js b/javascript/ql/test/library-tests/frameworks/WebSocket/browser-custom.js
@@ -0,0 +1,34 @@
+import { MyWebSocket, MySockJS } from './browser.js';
+
+(function () {
+	const socket = new MyWebSocket('ws://localhost:9080'); // $ MISSING: clientSocket
+
+	socket.addEventListener('open', function (event) {
+		socket.send('Hi from browser!'); // $ MISSING: clientSend
+	});
+
+	socket.addEventListener('message', function (event) {
+		console.log('Message from server ', event.data);
+	}); // $ MISSING: clientReceive
+
+	socket.onmessage = function (event) {
+		console.log("Message from server 2", event.data)
+	}; // $ MISSING: clientReceive
+})();
+
+
+(function () {
+	var sock = new MySockJS('http://0.0.0.0:9999/echo'); // $ MISSING: clientSocket
+	sock.onopen = function () {
+		sock.send('test'); // $ MISSING: clientSend
+	};
+	
+	sock.onmessage = function (e) {
+		console.log('message', e.data);
+		sock.close();
+	}; // $ MISSING: clientReceive
+	
+	sock.addEventListener('message', function (event) {
+		console.log('Using addEventListener ', event.data);
+	}); // $ MISSING: clientReceive
+})();
diff --git a/javascript/ql/test/library-tests/frameworks/WebSocket/browser.js b/javascript/ql/test/library-tests/frameworks/WebSocket/browser.js
@@ -30,3 +30,6 @@
 		console.log('Using addEventListener ', event.data);
 	}); // $clientReceive
 })();
+
+export const MyWebSocket = WebSocket;
+export const MySockJS = SockJS;
diff --git a/javascript/ql/test/library-tests/frameworks/WebSocket/client-custom.js b/javascript/ql/test/library-tests/frameworks/WebSocket/client-custom.js
@@ -0,0 +1,13 @@
+const { MyWebSocketWS } = require('./client.js');
+
+(function () {
+	const ws = new MyWebSocketWS('ws://example.org'); // $ MISSING: clientSocket
+
+	ws.on('open', function open() {
+		ws.send('Hi from client!'); // $ MISSING: clientSend
+	});
+
+	ws.on('message', function incoming(data) {
+		console.log(data);
+	}); // $ MISSING: clientReceive
+})();
diff --git a/javascript/ql/test/library-tests/frameworks/WebSocket/client.js b/javascript/ql/test/library-tests/frameworks/WebSocket/client.js
@@ -11,3 +11,5 @@
 		console.log(data);
 	}); // $clientReceive
 })();
+
+module.exports.MyWebSocketWS = require('ws');
diff --git a/javascript/ql/test/library-tests/frameworks/WebSocket/server-custom.js b/javascript/ql/test/library-tests/frameworks/WebSocket/server-custom.js
@@ -0,0 +1,13 @@
+const { MyWebSocketServer } = require('./client.js');
+
+(function () {
+	const wss = new MyWebSocketServer({ port: 8080 });
+
+	wss.on('connection', function connection(ws) { // $ MISSING: serverSocket
+		ws.on('message', function incoming(message) { // $ MISSING: remoteFlow
+			console.log('received: %s', message);
+		}); // $ MISSING: serverReceive
+
+		ws.send('Hi from server!'); // $ MISSING: serverSend
+	});
+})();
diff --git a/javascript/ql/test/library-tests/frameworks/WebSocket/server.js b/javascript/ql/test/library-tests/frameworks/WebSocket/server.js
@@ -11,3 +11,5 @@
 		ws.send('Hi from server!'); // $serverSend
 	});
 })();
+
+module.exports.MyWebSocketServer = require('ws').Server;
diff --git a/javascript/ql/test/library-tests/frameworks/WebSocket/test.expected b/javascript/ql/test/library-tests/frameworks/WebSocket/test.expected
@@ -13,9 +13,12 @@ clientSocket
 | browser.js:19:13:19:50 | new Soc ... /echo') |
 | client.js:4:13:4:45 | new Web ... e.org') |
 flowSteps
+| browser-custom.js:1:10:1:20 | MyWebSocket | browser-custom.js:1:10:1:20 | MyWebSocket |
+| browser-custom.js:1:23:1:30 | MySockJS | browser-custom.js:1:23:1:30 | MySockJS |
 | browser.js:5:15:5:32 | 'Hi from browser!' | server.js:7:38:7:44 | message |
 | browser.js:21:13:21:18 | 'test' | sockjs.js:9:31:9:37 | message |
 | client.js:7:11:7:27 | 'Hi from client!' | server.js:7:38:7:44 | message |
+| client.js:15:32:15:44 | require('ws') | client-custom.js:1:9:1:21 | MyWebSocketWS |
 | server.js:11:11:11:27 | 'Hi from server!' | browser.js:9:39:9:48 | event.data |
 | server.js:11:11:11:27 | 'Hi from server!' | browser.js:13:40:13:49 | event.data |
 | server.js:11:11:11:27 | 'Hi from server!' | client.js:10:37:10:40 | data |
