新增syncvar高级特性,使用protobuf定义
This commit is contained in:
174
packages/network/tests/NetworkCore.test.ts
Normal file
174
packages/network/tests/NetworkCore.test.ts
Normal file
@@ -0,0 +1,174 @@
|
||||
import { NetworkManager } from '../src/Core/NetworkManager';
|
||||
import { MessageHandler } from '../src/Messaging/MessageHandler';
|
||||
import { JsonMessage } from '../src/Messaging/NetworkMessage';
|
||||
|
||||
// 测试消息
|
||||
class TestMessage extends JsonMessage<{ text: string }> {
|
||||
public override readonly messageType: number = 1000;
|
||||
|
||||
constructor(text: string = 'test') {
|
||||
super({ text });
|
||||
}
|
||||
}
|
||||
|
||||
describe('网络核心功能测试', () => {
|
||||
let serverPort: number;
|
||||
|
||||
beforeAll(() => {
|
||||
// 使用随机端口避免冲突
|
||||
serverPort = 8000 + Math.floor(Math.random() * 1000);
|
||||
});
|
||||
|
||||
afterEach(async () => {
|
||||
// 每个测试后清理
|
||||
await NetworkManager.Stop();
|
||||
MessageHandler.Instance.clear();
|
||||
});
|
||||
|
||||
describe('NetworkManager', () => {
|
||||
test('应该能启动和停止服务端', async () => {
|
||||
// 启动服务端
|
||||
const startResult = await NetworkManager.StartServer(serverPort);
|
||||
expect(startResult).toBe(true);
|
||||
expect(NetworkManager.isServer).toBe(true);
|
||||
expect(NetworkManager.connectionCount).toBe(0);
|
||||
|
||||
// 停止服务端
|
||||
await NetworkManager.StopServer();
|
||||
expect(NetworkManager.isServer).toBe(false);
|
||||
}, 10000);
|
||||
|
||||
test('应该能启动和停止客户端', async () => {
|
||||
// 先启动服务端
|
||||
await NetworkManager.StartServer(serverPort);
|
||||
|
||||
// 启动客户端
|
||||
const connectResult = await NetworkManager.StartClient(`ws://localhost:${serverPort}`);
|
||||
expect(connectResult).toBe(true);
|
||||
expect(NetworkManager.isClient).toBe(true);
|
||||
|
||||
// 等待连接建立
|
||||
await new Promise(resolve => setTimeout(resolve, 100));
|
||||
|
||||
// 检查连接数
|
||||
expect(NetworkManager.connectionCount).toBe(1);
|
||||
|
||||
// 停止客户端
|
||||
await NetworkManager.StopClient();
|
||||
expect(NetworkManager.isClient).toBe(false);
|
||||
}, 10000);
|
||||
});
|
||||
|
||||
describe('消息系统', () => {
|
||||
test('应该能注册和处理消息', async () => {
|
||||
let receivedMessage: TestMessage | null = null;
|
||||
|
||||
// 注册消息处理器
|
||||
MessageHandler.Instance.registerHandler(
|
||||
1000,
|
||||
TestMessage,
|
||||
{
|
||||
handle: (message: TestMessage) => {
|
||||
receivedMessage = message;
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// 创建测试消息
|
||||
const testMessage = new TestMessage('Hello World');
|
||||
|
||||
// 序列化和反序列化测试
|
||||
const serialized = testMessage.serialize();
|
||||
expect(serialized.length).toBeGreaterThan(0);
|
||||
|
||||
// 模拟消息处理
|
||||
await MessageHandler.Instance.handleRawMessage(serialized);
|
||||
|
||||
// 验证消息被正确处理
|
||||
expect(receivedMessage).not.toBeNull();
|
||||
expect(receivedMessage!.payload!.text).toBe('Hello World');
|
||||
});
|
||||
|
||||
test('应该能处理多个处理器', async () => {
|
||||
let handler1Called = false;
|
||||
let handler2Called = false;
|
||||
|
||||
// 注册多个处理器
|
||||
MessageHandler.Instance.registerHandler(1000, TestMessage, {
|
||||
handle: () => { handler1Called = true; }
|
||||
}, 0);
|
||||
|
||||
MessageHandler.Instance.registerHandler(1000, TestMessage, {
|
||||
handle: () => { handler2Called = true; }
|
||||
}, 1);
|
||||
|
||||
// 发送消息
|
||||
const testMessage = new TestMessage('Test');
|
||||
await MessageHandler.Instance.handleMessage(testMessage);
|
||||
|
||||
// 验证两个处理器都被调用
|
||||
expect(handler1Called).toBe(true);
|
||||
expect(handler2Called).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
describe('端到端通信', () => {
|
||||
test('客户端和服务端应该能相互通信', async () => {
|
||||
let serverReceivedMessage: TestMessage | null = null;
|
||||
let clientReceivedMessage: TestMessage | null = null;
|
||||
|
||||
// 注册服务端消息处理器
|
||||
MessageHandler.Instance.registerHandler(1000, TestMessage, {
|
||||
handle: (message: TestMessage, connection) => {
|
||||
serverReceivedMessage = message;
|
||||
|
||||
// 服务端回复消息
|
||||
if (connection && NetworkManager.server) {
|
||||
const reply = new TestMessage('Server Reply');
|
||||
const replyData = reply.serialize();
|
||||
connection.send(replyData);
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
// 启动服务端
|
||||
await NetworkManager.StartServer(serverPort);
|
||||
|
||||
// 启动客户端
|
||||
await NetworkManager.StartClient(`ws://localhost:${serverPort}`);
|
||||
|
||||
// 等待连接建立
|
||||
await new Promise(resolve => setTimeout(resolve, 200));
|
||||
|
||||
// 设置客户端消息处理
|
||||
if (NetworkManager.client) {
|
||||
NetworkManager.client.on('message', async (data) => {
|
||||
const handled = await MessageHandler.Instance.handleRawMessage(data);
|
||||
if (handled) {
|
||||
// 从消息数据中重建消息
|
||||
const message = new TestMessage();
|
||||
message.deserialize(data);
|
||||
clientReceivedMessage = message;
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
// 客户端发送消息
|
||||
if (NetworkManager.client) {
|
||||
const clientMessage = new TestMessage('Client Hello');
|
||||
const messageData = clientMessage.serialize();
|
||||
NetworkManager.client.send(messageData);
|
||||
}
|
||||
|
||||
// 等待消息传输
|
||||
await new Promise(resolve => setTimeout(resolve, 200));
|
||||
|
||||
// 验证通信成功
|
||||
expect(serverReceivedMessage).not.toBeNull();
|
||||
expect(serverReceivedMessage!.payload!.text).toBe('Client Hello');
|
||||
|
||||
expect(clientReceivedMessage).not.toBeNull();
|
||||
expect(clientReceivedMessage!.payload!.text).toBe('Server Reply');
|
||||
}, 15000);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user