import { test, expect } from '@playwright/test';

// API 接口测试
test.describe('Swarm API 接口测试', () => {
  const apiBase = 'http://localhost:8000/api';

  test('健康检查接口', async ({ request }) => {
    const response = await request.get(`${apiBase}/health`);
    expect(response.ok()).toBeTruthy();

    const body = await response.json();
    expect(body).toHaveProperty('status');
    expect(body).toHaveProperty('version');
    expect(body.status).toBe('healthy');
  });

  test('获取 Agent 列表', async ({ request }) => {
    const response = await request.get(`${apiBase}/agents`);
    expect(response.ok()).toBeTruthy();

    const body = await response.json();
    expect(body).toHaveProperty('agents');
    expect(Array.isArray(body.agents)).toBeTruthy();
  });

  test('获取文件锁列表', async ({ request }) => {
    const response = await request.get(`${apiBase}/locks`);
    expect(response.ok()).toBeTruthy();

    const body = await response.json();
    expect(body).toHaveProperty('locks');
    expect(Array.isArray(body.locks)).toBeTruthy();
  });

  test('获取心跳列表', async ({ request }) => {
    const response = await request.get(`${apiBase}/heartbeats`);
    expect(response.ok()).toBeTruthy();

    const body = await response.json();
    expect(body).toHaveProperty('heartbeats');
  });

  test('注册 Agent', async ({ request }) => {
    const testAgent = {
      agent_id: `test-agent-${Date.now()}`,
      name: 'Test Agent',
      role: 'developer',
      model: 'test-model',
      description: 'Playwright 测试用 Agent',
    };

    const response = await request.post(`${apiBase}/agents/register`, {
      data: testAgent,
    });

    expect(response.ok()).toBeTruthy();
    const body = await response.json();
    expect(body.agent_id).toBe(testAgent.agent_id);
    expect(body.name).toBe(testAgent.name);
  });

  test('解析任务文件', async ({ request }) => {
    const response = await request.post(`${apiBase}/parse-task`, {
      data: { task: '修复 src/auth/login.py 中的 bug' },
    });

    expect(response.ok()).toBeTruthy();
    const body = await response.json();
    expect(body).toHaveProperty('files');
    expect(Array.isArray(body.files)).toBeTruthy();
  });

  test('获取任务主要角色', async ({ request }) => {
    const response = await request.post(`${apiBase}/roles/primary`, {
      data: { task: '实现登录功能并编写测试用例' },
    });

    expect(response.ok()).toBeTruthy();
    const body = await response.json();
    expect(body).toHaveProperty('primary_role');
    expect(body).toHaveProperty('role_scores');
  });

  test('列出工作流文件', async ({ request }) => {
    const response = await request.get(`${apiBase}/workflows/files`);
    expect(response.ok()).toBeTruthy();

    const body = await response.json();
    expect(body).toHaveProperty('files');
    expect(Array.isArray(body.files)).toBeTruthy();
  });
});