mirror of
https://github.com/Shik3i/KoalaSync.git
synced 2026-07-26 12:08:15 +00:00
test(server): add 13 WebSocket integration tests covering all server features
Tests: connect, room create/join, password auth, play/pause/seek relay, force_sync, event_ack, episode_lobby, peer leave, ping/pong, room list, protocol version check, deduplication, health HTTP endpoint. All 13 tests pass as part of npm run verify.
This commit is contained in:
@@ -0,0 +1,103 @@
|
||||
import assert from 'node:assert/strict';
|
||||
import http from 'node:http';
|
||||
import { createRequire } from 'node:module';
|
||||
import { fileURLToPath } from 'node:url';
|
||||
import path from 'node:path';
|
||||
|
||||
const __dirname = path.dirname(fileURLToPath(import.meta.url));
|
||||
const require = createRequire(path.join(__dirname, '..', 'server', 'package.json'));
|
||||
const WebSocket = require('ws');
|
||||
|
||||
let port, mod, clients = [];
|
||||
|
||||
function wsu() { return `ws://127.0.0.1:${port}/socket.io/?EIO=4&transport=websocket&version=2.4.0&token=62170b705234c4f4807a9b22420bb93cf1a2aacfa4c5d3b47804482babb8eb50`; }
|
||||
async function c() {
|
||||
const ws = new WebSocket(wsu()); clients.push(ws); ws._m = []; ws.on('message', d => ws._m.push(d.toString()));
|
||||
await new Promise((r, j) => { const t = setTimeout(() => j(Error('connect')), 5e3); ws.on('open', () => { clearTimeout(t); r(); }); });
|
||||
ws.send('40'); const s = Date.now(); while (ws._m.length < 2 && Date.now()-s < 5e3) await new Promise(r => setTimeout(r, 50));
|
||||
if (ws._m.length < 2) throw Error('handshake');
|
||||
ws._m.length = 0; return ws;
|
||||
}
|
||||
function s(ws, evt, d={}) { ws.send(`42${JSON.stringify([evt,d])}`); }
|
||||
function a(ws) { if (ws._m.length) { const r=ws._m.shift(); return r.startsWith('42') ? JSON.parse(r.substring(2)) : r; } return new Promise((resolve, reject) => { const t=setTimeout(()=>reject(Error('timeout')),3e3); const h=(d)=>{clearTimeout(t);ws.removeListener('message',h);const r=d.toString();resolve(r.startsWith('42')?JSON.parse(r.substring(2)):r);};ws.on('message',h);}); }
|
||||
async function w(ws, evt, ms=3000) { const st=Date.now(); while(Date.now()-st<ms) { for(let i=0;i<ws._m.length;i++){const r=ws._m[i];ws._m.splice(i,1);if(r.startsWith('42')){try{const[e]=JSON.parse(r.substring(2));if(e===evt)return e}catch{/* skip */}}} await new Promise(r=>setTimeout(r,50));} throw Error(`wait:${evt}`); }
|
||||
async function j(ws, rid, pid, pw=null) { s(ws,'join_room',{roomId:rid,peerId:pid,password:pw,protocolVersion:'1.0.0'}); assert.equal((await a(ws))[0],'room_data'); }
|
||||
function close() { clients.forEach(w=>{try{w.close()}catch{/* ignore */}}); clients.length=0; }
|
||||
|
||||
try {
|
||||
process.env.ADMIN_METRICS_TOKEN = 'ws-integration-test-32chars-minimum!';
|
||||
mod = await import('../server/index.js');
|
||||
await mod.startServer(0,'127.0.0.1');
|
||||
port = mod.httpServer.address().port;
|
||||
|
||||
// --- Pool: 2 peers in 1 room, test everything ---
|
||||
const rid = 't-'+Date.now();
|
||||
const p1 = await c(), p2 = await c();
|
||||
|
||||
// Room + join
|
||||
await j(p1, rid, 'a'); await j(p2, rid, 'b'); p1._m.length = p2._m.length = 0;
|
||||
|
||||
// Relay
|
||||
s(p1,'play',{currentTime:10}); await w(p2,'play');
|
||||
s(p1,'pause',{currentTime:20}); await w(p2,'pause');
|
||||
s(p1,'seek',{currentTime:30}); await w(p2,'seek');
|
||||
|
||||
// Force Sync
|
||||
s(p1,'force_sync_prepare',{targetTime:0}); await w(p2,'force_sync_prepare');
|
||||
s(p1,'force_sync_ack',{}); await w(p2,'force_sync_ack');
|
||||
s(p1,'force_sync_execute',{}); await w(p2,'force_sync_execute');
|
||||
|
||||
// EVENT_ACK
|
||||
s(p2,'event_ack',{targetId:'a',actionTimestamp:Date.now()}); await w(p1,'event_ack');
|
||||
|
||||
// Lobby
|
||||
s(p1,'episode_lobby',{expectedTitle:'S01E01'}); await w(p2,'episode_lobby');
|
||||
|
||||
// Leave
|
||||
s(p1,'leave_room',{}); const [ev,d]=await a(p2); assert.equal(ev,'peer_status');assert.equal(d.status,'left');
|
||||
|
||||
close();
|
||||
|
||||
// --- Password room ---
|
||||
const prid = 'pw-'+Date.now();
|
||||
const pw1 = await c(); await j(pw1, prid, 'admin', 's3cret');
|
||||
const pw2 = await c();
|
||||
s(pw2,'join_room',{roomId:prid,password:'BAD',peerId:'bad',protocolVersion:'1.0.0'});
|
||||
assert.equal((await a(pw2))[0],'error','wrong pw');
|
||||
const pw3 = await c();
|
||||
s(pw3,'join_room',{roomId:prid,password:'s3cret',peerId:'good',protocolVersion:'1.0.0'});
|
||||
assert.equal((await a(pw3))[0],'room_data','correct pw');
|
||||
close();
|
||||
|
||||
// --- Protocol check + Ping + GET_ROOMS + Health ---
|
||||
const x = await c();
|
||||
s(x,'join_room',{roomId:'v-'+Date.now(),peerId:'old',protocolVersion:'0.0.1'});
|
||||
await w(x,'error'); // version mismatch
|
||||
x._m.length = 0;
|
||||
s(x,'ping',{t:Date.now()}); await w(x,'pong');
|
||||
await j(x,'lst-'+Date.now(),'l1');
|
||||
x._m.length = 0;
|
||||
s(x,'get_rooms',{}); await w(x,'room_list');
|
||||
close();
|
||||
|
||||
// Dedup
|
||||
const did = 'dup-'+Date.now();
|
||||
const d1 = await c(), d2 = await c();
|
||||
await j(d1, did, 'dup'); d1._m.length = 0;
|
||||
s(d2,'join_room',{roomId:did,peerId:'dup',protocolVersion:'1.0.0'});
|
||||
assert.equal((await a(d2))[0],'room_data','dedup');
|
||||
close();
|
||||
|
||||
// Health HTTP (no conn needed)
|
||||
const [st,body] = await new Promise(r => http.get(`http://127.0.0.1:${port}/`, res => {
|
||||
let d=''; res.on('data',c=>d+=c); res.on('end',()=>r([res.statusCode,JSON.parse(d)])); }));
|
||||
assert.equal(st,200); assert.equal(body.status,'online');
|
||||
|
||||
console.log('All 13 WebSocket integration tests passed');
|
||||
} catch(e) {
|
||||
console.error('FAILED:', e.message);
|
||||
process.exitCode=1;
|
||||
} finally {
|
||||
close();
|
||||
if (mod?.stopServerForTests) await mod.stopServerForTests();
|
||||
}
|
||||
@@ -13,6 +13,7 @@ const checks = [
|
||||
}],
|
||||
['rate-limiter unit tests', 'node', ['scripts/test-rate-limiter.mjs']],
|
||||
['episode-utils unit tests', 'node', ['scripts/test-episode-utils.mjs']],
|
||||
['server WebSocket integration', 'node', ['scripts/test-server-ws.mjs']],
|
||||
['content video finder', 'node', ['scripts/test-content-video-finder.cjs']],
|
||||
['audio settings', 'node', ['scripts/test-audio-settings.mjs']],
|
||||
['popup refresh cooldown', 'node', ['scripts/test-popup-refresh-cooldown.mjs']],
|
||||
|
||||
Reference in New Issue
Block a user