# tests/unit/test_message_model.py
import sys
sys.path.insert(0, '/home/jang/Projects/ai-researcher')

import pytest
from datetime import datetime

from src.models.message import Message


def test_message_creation():
    """测试Message创建"""
    msg = Message(
        id="msg_123",
        conversation_id="conv_456",
        role="assistant",
        content={"text": "技术部有15名员工"},
        trace_id="trace_789",
    )

    assert msg.id == "msg_123"
    assert msg.conversation_id == "conv_456"
    assert msg.role == "assistant"
    assert msg.content == {"text": "技术部有15名员工"}
    assert msg.trace_id == "trace_789"


def test_message_roles():
    """测试消息角色枚举"""
    assert Message.ROLE_USER == "user"
    assert Message.ROLE_ASSISTANT == "assistant"
    assert Message.ROLE_SYSTEM == "system"