from jawbreaker.schema import ScamAnalysis
from jawbreaker.analyzers import prediction_to_analysis
from jawbreaker.render import render_analysis_html, render_scanning_html


def test_family_impersonation_is_dangerous() -> None:
    analysis = ScamAnalysis.from_heuristics(
        "Hi Grandma, I lost my phone. This is my new number. Can you send $800 today?"
    )

    assert analysis.risk_level == "dangerous"
    assert analysis.scam_type == "family_impersonation"
    assert "Trust a new number" == analysis.scam_dna["Ask"]


def test_legitimate_fraud_alert_needs_check_not_dangerous() -> None:
    analysis = ScamAnalysis.from_heuristics(
        "Chase fraud alert: Did you attempt a $249.00 purchase at TARGET? Reply YES or NO."
    )

    assert analysis.risk_level == "needs_check"
    assert analysis.scam_type == "possible_legitimate_alert"


def test_device_protection_subscription_alert_is_tech_support() -> None:
    analysis = ScamAnalysis.from_heuristics(
        "Device protection team: subscription expired. Call the number in this alert to avoid data loss."
    )

    assert analysis.risk_level == "dangerous"
    assert analysis.scam_type == "tech_support"


def test_billing_link_with_account_closure_is_dangerous() -> None:
    analysis = ScamAnalysis.from_heuristics(
        "Streaming billing: payment failed and your account closes tonight. Update through this message link: http://billing-review.example"
    )

    assert analysis.risk_level == "dangerous"
    assert analysis.scam_type == "credential_theft"


def test_prediction_to_analysis_normalizes_model_json() -> None:
    analysis = prediction_to_analysis(
        {
            "risk_level": "not_valid",
            "scam_type": "package_phishing",
            "summary": "Fake delivery notice.",
            "tactics": ["fake authority"],
            "safest_action": "Open the official carrier website yourself.",
            "trusted_person_message": "Can you check this?",
            "scam_dna": {
                "impersonates": "USPS",
                "pressure": "Held package",
                "ask": "Open link",
                "risk": "credential theft",
            },
        },
        similar_memory="This resembles a saved pattern.",
    )

    assert analysis.risk_level == "needs_check"
    assert analysis.scam_dna["Impersonates"] == "USPS"
    assert analysis.similar_memory == "This resembles a saved pattern."


def test_render_humanizes_model_style_labels() -> None:
    analysis = ScamAnalysis(
        risk_level="dangerous",
        scam_type="credential_theft",
        summary="This message mentions credential_theft.",
        tactics=["unknown_urgent_action", "unknown_urgent_payment", "credential_theft"],
        safest_action="Do not click the link.",
        scam_dna={
            "Impersonates": "legitimate_company",
            "Pressure": "sense_of_urgency",
            "Ask": "click_link_and_verify",
            "Risk": "credential_theft",
        },
    )

    html = render_analysis_html("USPS: verify now", analysis)

    assert "legitimate company" in html
    assert "click a link and verify" in html
    assert "unknown_urgent_action" not in html
    assert "urgent payment" in html
    assert "data-copy=" in html


def test_render_scanning_html_tracks_progress() -> None:
    html = render_scanning_html(active_step=2, progress=54)

    assert "54% COMPLETE" in html
    assert "✓ Preparing the safety model" in html
    assert "> Checking scam pressure signals" in html