from __future__ import annotations
from typing import Dict, List, Literal, Optional
from pydantic import BaseModel, Field
Priority = Literal["P0", "P1", "P2"]
Category = Literal["happy_path", "boundary", "negative", "auth", "idempotency", "concurrency"]
class APIInfo(BaseModel):
name: str = Field(..., min_length=1)
method: Literal["GET", "POST", "PUT", "DELETE"]
path: str = Field(..., pattern=r"^/.*")
class Request(BaseModel):
headers: Dict[str, str] = Field(default_factory=dict)
query: Dict[str, object] = Field(default_factory=dict)
body: Dict[str, object] = Field(default_factory=dict)
class Expected(BaseModel):
http_status: int = Field(..., ge=100, le=599)
body_contains: List[str] = Field(default_factory=list)
error_code: Optional[str] = Field(default=None, pattern=r"^[A-Z_]+$")
class TestCase(BaseModel):
id: str = Field(..., pattern=r"^TC-\d{4}$")
title: str = Field(..., min_length=4)
priority: Priority
category: Category
precondition: str = ""
steps: List[str] = Field(..., min_length=2, max_length=30)
request: Request
expected: Expected
class CaseGenOutput(BaseModel):
api: APIInfo
testcases: List[TestCase] = Field(..., min_length=6)