feat(framework): net Transport 接口 + 测试用 FakeTransport
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,27 @@
|
||||
import type { Transport } from '../../YouleNexus/assets/framework/net/transport.ts';
|
||||
|
||||
/** 内存 Transport:模拟服务器。用 microtask 队列异步触发回调,贴近真实 WS 时序。 */
|
||||
export class FakeTransport implements Transport {
|
||||
sent: string[] = [];
|
||||
url = '';
|
||||
private openCb?: () => void;
|
||||
private msgCb?: (f: string) => void;
|
||||
private closeCb?: () => void;
|
||||
private pending: Array<() => void> = [];
|
||||
private closed = false;
|
||||
|
||||
connect(url: string): void { this.url = url; this.pending.push(() => this.openCb?.()); }
|
||||
send(frame: string): void { if (!this.closed) this.sent.push(frame); }
|
||||
close(): void { if (this.closed) return; this.closed = true; this.pending.push(() => this.closeCb?.()); }
|
||||
onOpen(cb: () => void): void { this.openCb = cb; }
|
||||
onMessage(cb: (f: string) => void): void { this.msgCb = cb; }
|
||||
onClose(cb: () => void): void { this.closeCb = cb; }
|
||||
|
||||
/** 测试侧:模拟服务器推一帧给客户端。 */
|
||||
serverPush(frame: string): void { if (!this.closed) this.pending.push(() => this.msgCb?.(frame)); }
|
||||
|
||||
/** 冲刷所有挂起回调(解析微任务)。 */
|
||||
async flush(): Promise<void> {
|
||||
while (this.pending.length) { const fn = this.pending.shift()!; fn(); await Promise.resolve(); }
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,36 @@
|
||||
import { test } from 'node:test';
|
||||
import assert from 'node:assert/strict';
|
||||
import { FakeTransport } from '../helpers/fake-transport.ts';
|
||||
|
||||
test('FakeTransport:connect 后异步触发 onOpen', async () => {
|
||||
const t = new FakeTransport();
|
||||
let opened = false;
|
||||
t.onOpen(() => { opened = true; });
|
||||
t.connect('ws://x');
|
||||
await t.flush();
|
||||
assert.equal(opened, true);
|
||||
});
|
||||
|
||||
test('FakeTransport:客户端 send 进入 sent 队列;serverPush 触发 onMessage', async () => {
|
||||
const t = new FakeTransport();
|
||||
const got: string[] = [];
|
||||
t.onMessage((f) => got.push(f));
|
||||
t.connect('ws://x');
|
||||
await t.flush();
|
||||
t.send('hello');
|
||||
assert.deepEqual(t.sent, ['hello']);
|
||||
t.serverPush('world');
|
||||
await t.flush();
|
||||
assert.deepEqual(got, ['world']);
|
||||
});
|
||||
|
||||
test('FakeTransport:close 触发 onClose', async () => {
|
||||
const t = new FakeTransport();
|
||||
let closed = false;
|
||||
t.onClose(() => { closed = true; });
|
||||
t.connect('ws://x');
|
||||
await t.flush();
|
||||
t.close();
|
||||
await t.flush();
|
||||
assert.equal(closed, true);
|
||||
});
|
||||
Reference in New Issue
Block a user