370 lines
14 KiB
TypeScript
370 lines
14 KiB
TypeScript
/**
|
||
* SnapshotManager与Protobuf序列化集成测试
|
||
*/
|
||
|
||
import { Entity } from '../../../src/ECS/Entity';
|
||
import { Scene } from '../../../src/ECS/Scene';
|
||
import { Component } from '../../../src/ECS/Component';
|
||
import { SnapshotManager } from '../../../src/Utils/Snapshot/SnapshotManager';
|
||
import {
|
||
ProtoSerializable,
|
||
ProtoFloat,
|
||
ProtoInt32,
|
||
ProtoString,
|
||
ProtoBool
|
||
} from '../../../src/Utils/Serialization/ProtobufDecorators';
|
||
|
||
// 测试组件
|
||
@ProtoSerializable('TestPosition')
|
||
class TestPositionComponent extends Component {
|
||
@ProtoFloat(1)
|
||
public x: number = 0;
|
||
|
||
@ProtoFloat(2)
|
||
public y: number = 0;
|
||
|
||
constructor(x: number = 0, y: number = 0) {
|
||
super();
|
||
this.x = x;
|
||
this.y = y;
|
||
}
|
||
}
|
||
|
||
@ProtoSerializable('TestVelocity')
|
||
class TestVelocityComponent extends Component {
|
||
@ProtoFloat(1)
|
||
public vx: number = 0;
|
||
|
||
@ProtoFloat(2)
|
||
public vy: number = 0;
|
||
|
||
constructor(vx: number = 0, vy: number = 0) {
|
||
super();
|
||
this.vx = vx;
|
||
this.vy = vy;
|
||
}
|
||
}
|
||
|
||
@ProtoSerializable('TestHealth')
|
||
class TestHealthComponent extends Component {
|
||
@ProtoInt32(1)
|
||
public maxHealth: number = 100;
|
||
|
||
@ProtoInt32(2)
|
||
public currentHealth: number = 100;
|
||
|
||
@ProtoBool(3)
|
||
public isDead: boolean = false;
|
||
|
||
constructor(maxHealth: number = 100) {
|
||
super();
|
||
this.maxHealth = maxHealth;
|
||
this.currentHealth = maxHealth;
|
||
}
|
||
}
|
||
|
||
// 传统JSON序列化组件
|
||
class TraditionalComponent extends Component {
|
||
public customData = {
|
||
name: 'traditional',
|
||
values: [1, 2, 3],
|
||
settings: { enabled: true }
|
||
};
|
||
|
||
serialize(): any {
|
||
return {
|
||
customData: this.customData
|
||
};
|
||
}
|
||
|
||
deserialize(data: any): void {
|
||
if (data.customData) {
|
||
this.customData = data.customData;
|
||
}
|
||
}
|
||
}
|
||
|
||
// 简单组件（使用默认序列化）
|
||
class SimpleComponent extends Component {
|
||
public value: number = 42;
|
||
public text: string = 'simple';
|
||
public flag: boolean = true;
|
||
}
|
||
|
||
// Mock protobuf.js
|
||
const mockProtobuf = {
|
||
parse: jest.fn().mockReturnValue({
|
||
root: {
|
||
lookupType: jest.fn().mockImplementation((typeName: string) => {
|
||
const mockData = {
|
||
'ecs.TestPosition': { x: 10, y: 20 },
|
||
'ecs.TestVelocity': { vx: 5, vy: 3 },
|
||
'ecs.TestHealth': { maxHealth: 100, currentHealth: 80, isDead: false }
|
||
};
|
||
|
||
return {
|
||
verify: jest.fn().mockReturnValue(null),
|
||
create: jest.fn().mockImplementation((data) => data),
|
||
encode: jest.fn().mockReturnValue({
|
||
finish: jest.fn().mockReturnValue(new Uint8Array([1, 2, 3, 4]))
|
||
}),
|
||
decode: jest.fn().mockReturnValue(mockData[typeName] || {}),
|
||
toObject: jest.fn().mockImplementation((message) => message)
|
||
};
|
||
})
|
||
}
|
||
})
|
||
};
|
||
|
||
describe('SnapshotManager Protobuf集成', () => {
|
||
let snapshotManager: SnapshotManager;
|
||
let scene: Scene;
|
||
|
||
beforeEach(() => {
|
||
snapshotManager = new SnapshotManager();
|
||
snapshotManager.initializeProtobuf(mockProtobuf);
|
||
scene = new Scene();
|
||
jest.clearAllMocks();
|
||
});
|
||
|
||
describe('混合序列化快照', () => {
|
||
it('应该正确创建包含protobuf和JSON组件的快照', () => {
|
||
// 创建实体
|
||
const player = scene.createEntity('Player');
|
||
player.addComponent(new TestPositionComponent(100, 200));
|
||
player.addComponent(new TestVelocityComponent(5, 3));
|
||
player.addComponent(new TestHealthComponent(120));
|
||
player.addComponent(new TraditionalComponent());
|
||
player.addComponent(new SimpleComponent());
|
||
|
||
// 创建快照
|
||
const snapshot = snapshotManager.createSceneSnapshot([player]);
|
||
|
||
expect(snapshot).toBeDefined();
|
||
expect(snapshot.entities).toHaveLength(1);
|
||
expect(snapshot.entities[0].components).toHaveLength(5);
|
||
|
||
// 验证快照包含所有组件
|
||
const componentTypes = snapshot.entities[0].components.map(c => c.type);
|
||
expect(componentTypes).toContain('TestPositionComponent');
|
||
expect(componentTypes).toContain('TestVelocityComponent');
|
||
expect(componentTypes).toContain('TestHealthComponent');
|
||
expect(componentTypes).toContain('TraditionalComponent');
|
||
expect(componentTypes).toContain('SimpleComponent');
|
||
});
|
||
|
||
it('应该根据组件类型使用相应的序列化方式', () => {
|
||
const entity = scene.createEntity('TestEntity');
|
||
const position = new TestPositionComponent(50, 75);
|
||
const traditional = new TraditionalComponent();
|
||
|
||
entity.addComponent(position);
|
||
entity.addComponent(traditional);
|
||
|
||
const snapshot = snapshotManager.createSceneSnapshot([entity]);
|
||
const components = snapshot.entities[0].components;
|
||
|
||
// 检查序列化数据格式
|
||
const positionSnapshot = components.find(c => c.type === 'TestPositionComponent');
|
||
const traditionalSnapshot = components.find(c => c.type === 'TraditionalComponent');
|
||
|
||
expect(positionSnapshot).toBeDefined();
|
||
expect(traditionalSnapshot).toBeDefined();
|
||
|
||
// Protobuf组件应该有SerializedData格式
|
||
expect(positionSnapshot!.data).toHaveProperty('type');
|
||
expect(positionSnapshot!.data).toHaveProperty('componentType');
|
||
expect(positionSnapshot!.data).toHaveProperty('data');
|
||
expect(positionSnapshot!.data).toHaveProperty('size');
|
||
});
|
||
});
|
||
|
||
describe('快照恢复', () => {
|
||
it('应该正确恢复protobuf序列化的组件', () => {
|
||
// 创建原始实体
|
||
const originalEntity = scene.createEntity('Original');
|
||
const originalPosition = new TestPositionComponent(100, 200);
|
||
const originalHealth = new TestHealthComponent(150);
|
||
originalHealth.currentHealth = 120;
|
||
|
||
originalEntity.addComponent(originalPosition);
|
||
originalEntity.addComponent(originalHealth);
|
||
|
||
// 创建快照
|
||
const snapshot = snapshotManager.createSceneSnapshot([originalEntity]);
|
||
|
||
// 创建新实体进行恢复
|
||
const newEntity = scene.createEntity('New');
|
||
newEntity.addComponent(new TestPositionComponent());
|
||
newEntity.addComponent(new TestHealthComponent());
|
||
|
||
// 恢复快照
|
||
snapshotManager.restoreFromSnapshot(snapshot, [newEntity]);
|
||
|
||
// 验证数据被正确恢复（注意：由于使用mock，实际值来自mock数据）
|
||
const restoredPosition = newEntity.getComponent(TestPositionComponent);
|
||
const restoredHealth = newEntity.getComponent(TestHealthComponent);
|
||
|
||
expect(restoredPosition).toBeDefined();
|
||
expect(restoredHealth).toBeDefined();
|
||
|
||
// 验证protobuf的decode方法被调用
|
||
expect(mockProtobuf.parse().root.lookupType).toHaveBeenCalled();
|
||
});
|
||
|
||
it('应该正确恢复传统JSON序列化的组件', () => {
|
||
const originalEntity = scene.createEntity('Original');
|
||
const originalTraditional = new TraditionalComponent();
|
||
originalTraditional.customData.name = 'modified';
|
||
originalTraditional.customData.values = [4, 5, 6];
|
||
|
||
originalEntity.addComponent(originalTraditional);
|
||
|
||
const snapshot = snapshotManager.createSceneSnapshot([originalEntity]);
|
||
|
||
const newEntity = scene.createEntity('New');
|
||
const newTraditional = new TraditionalComponent();
|
||
newEntity.addComponent(newTraditional);
|
||
|
||
snapshotManager.restoreFromSnapshot(snapshot, [newEntity]);
|
||
|
||
// 验证JSON数据被正确恢复
|
||
expect(newTraditional.customData.name).toBe('modified');
|
||
expect(newTraditional.customData.values).toEqual([4, 5, 6]);
|
||
});
|
||
|
||
it('应该处理混合序列化的实体恢复', () => {
|
||
const originalEntity = scene.createEntity('Mixed');
|
||
const position = new TestPositionComponent(30, 40);
|
||
const traditional = new TraditionalComponent();
|
||
const simple = new SimpleComponent();
|
||
|
||
traditional.customData.name = 'mixed_test';
|
||
simple.value = 99;
|
||
simple.text = 'updated';
|
||
|
||
originalEntity.addComponent(position);
|
||
originalEntity.addComponent(traditional);
|
||
originalEntity.addComponent(simple);
|
||
|
||
const snapshot = snapshotManager.createSceneSnapshot([originalEntity]);
|
||
|
||
const newEntity = scene.createEntity('NewMixed');
|
||
newEntity.addComponent(new TestPositionComponent());
|
||
newEntity.addComponent(new TraditionalComponent());
|
||
newEntity.addComponent(new SimpleComponent());
|
||
|
||
snapshotManager.restoreFromSnapshot(snapshot, [newEntity]);
|
||
|
||
// 验证所有组件都被正确恢复
|
||
const restoredTraditional = newEntity.getComponent(TraditionalComponent);
|
||
const restoredSimple = newEntity.getComponent(SimpleComponent);
|
||
|
||
expect(restoredTraditional!.customData.name).toBe('mixed_test');
|
||
expect(restoredSimple!.value).toBe(99);
|
||
expect(restoredSimple!.text).toBe('updated');
|
||
});
|
||
});
|
||
|
||
describe('向后兼容性', () => {
|
||
it('应该能够处理旧格式的快照数据', () => {
|
||
// 模拟旧格式的快照数据
|
||
const legacySnapshot = {
|
||
entities: [{
|
||
id: 1,
|
||
name: 'LegacyEntity',
|
||
enabled: true,
|
||
active: true,
|
||
tag: 0,
|
||
updateOrder: 0,
|
||
components: [{
|
||
type: 'SimpleComponent',
|
||
id: 1,
|
||
data: { value: 123, text: 'legacy', flag: false }, // 直接的JSON数据
|
||
enabled: true,
|
||
config: { includeInSnapshot: true, compressionLevel: 0, syncPriority: 5, enableIncremental: true }
|
||
}],
|
||
children: [],
|
||
timestamp: Date.now()
|
||
}],
|
||
timestamp: Date.now(),
|
||
version: '1.0.0',
|
||
type: 'full' as const
|
||
};
|
||
|
||
const entity = scene.createEntity('TestEntity');
|
||
entity.addComponent(new SimpleComponent());
|
||
|
||
snapshotManager.restoreFromSnapshot(legacySnapshot, [entity]);
|
||
|
||
const component = entity.getComponent(SimpleComponent);
|
||
expect(component!.value).toBe(123);
|
||
expect(component!.text).toBe('legacy');
|
||
expect(component!.flag).toBe(false);
|
||
});
|
||
});
|
||
|
||
describe('错误处理', () => {
|
||
it('应该优雅地处理protobuf序列化失败', () => {
|
||
// 模拟protobuf验证失败
|
||
const mockType = mockProtobuf.parse().root.lookupType;
|
||
mockType.mockImplementation(() => ({
|
||
verify: jest.fn().mockReturnValue('验证失败'),
|
||
create: jest.fn(),
|
||
encode: jest.fn(),
|
||
decode: jest.fn(),
|
||
toObject: jest.fn()
|
||
}));
|
||
|
||
const entity = scene.createEntity('ErrorTest');
|
||
entity.addComponent(new TestPositionComponent(10, 20));
|
||
|
||
// 应该不抛出异常，而是回退到JSON序列化
|
||
expect(() => {
|
||
snapshotManager.createSceneSnapshot([entity]);
|
||
}).not.toThrow();
|
||
});
|
||
|
||
it('应该优雅地处理protobuf反序列化失败', () => {
|
||
const entity = scene.createEntity('Test');
|
||
const position = new TestPositionComponent(10, 20);
|
||
entity.addComponent(position);
|
||
|
||
const snapshot = snapshotManager.createSceneSnapshot([entity]);
|
||
|
||
// 模拟反序列化失败
|
||
const mockType = mockProtobuf.parse().root.lookupType;
|
||
mockType.mockImplementation(() => ({
|
||
verify: jest.fn().mockReturnValue(null),
|
||
create: jest.fn(),
|
||
encode: jest.fn().mockReturnValue({
|
||
finish: jest.fn().mockReturnValue(new Uint8Array([1, 2, 3, 4]))
|
||
}),
|
||
decode: jest.fn().mockImplementation(() => {
|
||
throw new Error('解码失败');
|
||
}),
|
||
toObject: jest.fn()
|
||
}));
|
||
|
||
const newEntity = scene.createEntity('NewTest');
|
||
newEntity.addComponent(new TestPositionComponent());
|
||
|
||
// 应该不抛出异常
|
||
expect(() => {
|
||
snapshotManager.restoreFromSnapshot(snapshot, [newEntity]);
|
||
}).not.toThrow();
|
||
});
|
||
});
|
||
|
||
describe('统计信息', () => {
|
||
it('应该包含protobuf统计信息', () => {
|
||
const stats = snapshotManager.getCacheStats();
|
||
|
||
expect(stats).toHaveProperty('snapshotCacheSize');
|
||
expect(stats).toHaveProperty('protobufStats');
|
||
expect(stats.protobufStats).toHaveProperty('registeredComponents');
|
||
expect(stats.protobufStats).toHaveProperty('protobufAvailable');
|
||
expect(stats.protobufStats!.protobufAvailable).toBe(true);
|
||
});
|
||
});
|
||
}); |