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

import pytest
from datetime import datetime

from src.models.task import Task


def test_task_creation():
    """测试Task创建"""
    task = Task(
        id="task_123",
        status=Task.STATUS_PENDING,
        query="查询技术部员工",
        complexity_score=0.45,
    )

    assert task.id == "task_123"
    assert task.status == "pending"
    assert task.query == "查询技术部员工"
    assert task.complexity_score == 0.45


def test_task_status_transitions():
    """测试任务状态"""
    assert Task.STATUS_PENDING == "pending"
    assert Task.STATUS_RUNNING == "running"
    assert Task.STATUS_COMPLETED == "completed"
    assert Task.STATUS_FAILED == "failed"
    assert Task.STATUS_CANCELLED == "cancelled"