import { describe, it, expect } from 'vitest';
import { createEntityCollection, type Entity } from '../../src/utils/entity';

interface TestEntity extends Entity {
    name: string;
    value: number;
}

describe('createEntityCollection', () => {
    it('should create empty collection', () => {
        const collection = createEntityCollection<TestEntity>();
        expect(collection.collection.value).toEqual({});
    });

    it('should add single entity', () => {
        const collection = createEntityCollection<TestEntity>();
        const entity: TestEntity = { id: 'e1', name: 'Entity 1', value: 10 };
        
        collection.add(entity);
        
        expect(collection.collection.value).toHaveProperty('e1');
        expect(collection.get('e1').value).toEqual(entity);
    });

    it('should add multiple entities', () => {
        const collection = createEntityCollection<TestEntity>();
        const entity1: TestEntity = { id: 'e1', name: 'Entity 1', value: 10 };
        const entity2: TestEntity = { id: 'e2', name: 'Entity 2', value: 20 };
        const entity3: TestEntity = { id: 'e3', name: 'Entity 3', value: 30 };
        
        collection.add(entity1, entity2, entity3);
        
        expect(Object.keys(collection.collection.value)).toHaveLength(3);
        expect(collection.get('e1').value.name).toBe('Entity 1');
        expect(collection.get('e2').value.name).toBe('Entity 2');
        expect(collection.get('e3').value.name).toBe('Entity 3');
    });

    it('should remove single entity', () => {
        const collection = createEntityCollection<TestEntity>();
        const entity1: TestEntity = { id: 'e1', name: 'Entity 1', value: 10 };
        const entity2: TestEntity = { id: 'e2', name: 'Entity 2', value: 20 };
        
        collection.add(entity1, entity2);
        collection.remove('e1');
        
        expect(Object.keys(collection.collection.value)).toHaveLength(1);
        expect(collection.collection.value).not.toHaveProperty('e1');
        expect(collection.collection.value).toHaveProperty('e2');
    });

    it('should remove multiple entities', () => {
        const collection = createEntityCollection<TestEntity>();
        const entity1: TestEntity = { id: 'e1', name: 'Entity 1', value: 10 };
        const entity2: TestEntity = { id: 'e2', name: 'Entity 2', value: 20 };
        const entity3: TestEntity = { id: 'e3', name: 'Entity 3', value: 30 };
        
        collection.add(entity1, entity2, entity3);
        collection.remove('e1', 'e3');
        
        expect(Object.keys(collection.collection.value)).toHaveLength(1);
        expect(collection.collection.value).toHaveProperty('e2');
    });

    it('should update entity via accessor', () => {
        const collection = createEntityCollection<TestEntity>();
        const entity: TestEntity = { id: 'e1', name: 'Entity 1', value: 10 };
        
        collection.add(entity);
        
        const accessor = collection.get('e1');
        accessor.value = { ...entity, value: 100, name: 'Updated' };
        
        expect(collection.get('e1').value.value).toBe(100);
        expect(collection.get('e1').value.name).toBe('Updated');
    });

    it('should return undefined for non-existent entity', () => {
        const collection = createEntityCollection<TestEntity>();
        
        expect(collection.get('nonexistent').value).toBeUndefined();
    });

    it('should have correct accessor id', () => {
        const collection = createEntityCollection<TestEntity>();
        const entity: TestEntity = { id: 'e1', name: 'Entity 1', value: 10 };
        
        collection.add(entity);
        
        const accessor = collection.get('e1');
        expect(accessor.id).toBe('e1');
    });

    it('should handle removing non-existent entity', () => {
        const collection = createEntityCollection<TestEntity>();
        const entity: TestEntity = { id: 'e1', name: 'Entity 1', value: 10 };
        
        collection.add(entity);
        collection.remove('nonexistent');
        
        expect(Object.keys(collection.collection.value)).toHaveLength(1);
    });

    it('should work with reactive updates', () => {
        const collection = createEntityCollection<TestEntity>();
        const entity: TestEntity = { id: 'e1', name: 'Entity 1', value: 10 };
        
        collection.add(entity);
        
        // 验证 accessor 可以正确获取和设置值
        const accessor = collection.get('e1');
        expect(accessor.value.value).toBe(10);
        
        accessor.value = { ...entity, value: 50 };
        expect(accessor.value.value).toBe(50);
    });
});