import pytest
import sys
sys.path.insert(0, '/home/jang/Projects/ai-researcher')

from src.agent.skills.data_query.handler import DataQuerySkill
from src.agent.executor import SkillContext


@pytest.fixture
def skill():
    return DataQuerySkill()


@pytest.mark.asyncio
async def test_query_employee_info(skill):
    """测试查询员工信息"""
    context = SkillContext(
        user_id="user_1",
        tenant_id="tenant_1",
        trace_id="trace_123"
    )

    result = await skill.execute(
        capability="query_employee_info",
        params={"department": "技术部"},
        context=context
    )

    assert result is not None
    assert hasattr(result, 'data')
    assert result.data is not None