models.py

from pydantic import BaseModel
from typing import Optional


class TestFailure(BaseModel):
    name: str
    status: str  # "failed" or "broken"
    uid: str
    duration_seconds: float
    flaky: bool
    mftf_test_name: str  # extracted test name used for Jira search


class JiraTicket(BaseModel):
    key: str
    summary: str
    status: str
    url: str
    description: Optional[str] = None


class FailureWithJira(BaseModel):
    test: TestFailure
    jira_tickets: list[JiraTicket] = []
    jira_search_performed: bool = False
    jira_error: Optional[str] = None


class AnalyzeRequest(BaseModel):
    allure_url: str
    allure_auth_token: Optional[str] = None  # Bearer token if report is behind auth
    jira_base_url: Optional[str] = None
    jira_token: Optional[str] = None


class ParseRequest(BaseModel):
    allure_data: dict  # raw Allure JSON (suites.json or behaviors.json content)
    jira_base_url: Optional[str] = None
    jira_token: Optional[str] = None


class AnalyzeSummary(BaseModel):
    total_tests_scanned: int
    total_failed: int
    total_broken: int
    total_with_jira_tickets: int
    allure_url: str


class AnalyzeResponse(BaseModel):
    summary: AnalyzeSummary
    results: list[FailureWithJira]