tests/fixtures/test_data.py

"""
测试数据fixture
"""

from src.core import EntityInfo


def create_test_entity_a():
    """创建测试实体A"""
    return EntityInfo(
        id="test-entity-a-001",
        name="Test Entity A",
        redis_host="localhost",
        redis_port=6379,
        redis_db=0,
        channel="test-entity-a-001"
    )


def create_test_entity_b():
    """创建测试实体B"""
    return EntityInfo(
        id="test-entity-b-002", 
        name="Test Entity B",
        redis_host="localhost",
        redis_port=6379,
        redis_db=0,
        channel="test-entity-b-002"
    )


def get_test_messages():
    """获取测试消息列表"""
    return [
        "Hello from Test Entity A!",
        "This is a test message",
        "Human-Clone system working"
    ]