tests/test_models.py

from pydantic import ValidationError

from auditenv.models import AuditAction, AuditFinding


def test_finding_confidence_validation() -> None:
    try:
        AuditFinding(
            document_id="E-DOC-001",
            violation_type="duplicate_receipt",
            evidence=["E-DOC-001"],
            confidence=1.2,
        )
        assert False, "Expected ValidationError for confidence > 1.0"
    except ValidationError:
        assert True


def test_action_submit_finding_payload() -> None:
    action = AuditAction(
        action_type="submit_finding",
        task_id="easy",
        finding=AuditFinding(
            document_id="E-DOC-001",
            violation_type="duplicate_receipt",
            evidence=["E-DOC-001"],
            confidence=0.8,
        ),
    )
    assert action.finding is not None