import json
from datetime import UTC, datetime

import pytest

from whitehat_security.engagement import AuthorizationError, Engagement


def record():
    return {
        "engagement_id": "WH-ENG-1", "authorization_id": "auth-1",
        "authorizer": "operator", "approved_at": "2026-08-21T08:00:00Z",
        "expires_at": "2026-08-21T12:00:00Z", "target": "https://fixture.invalid",
        "target_owner": "target-repo", "environment": "build", "source": "runner",
        "routes": ["GET /objects/{id}"], "fixture_ids": ["object-a", "object-b"],
        "credential_lane": "openbao", "credential_role": "runtime",
        "credential_max_ttl_seconds": 900, "techniques": ["e2-differential"],
        "prohibited_techniques": ["saturation"], "rate_limit_per_minute": 10,
        "max_concurrency": 1, "window_start": "2026-08-21T08:00:00Z",
        "window_end": "2026-08-21T10:00:00Z", "operator_contact": "operator",
        "abort_contact": "target-owner", "posture_claim": "E2",
        "attacker_model": "E2-authenticated-tenant-a", "finding_destination": "risk-nexus",
        "target_owner_acknowledged_at": "2026-08-21T08:01:00Z",
    }


def load(tmp_path, data):
    path = tmp_path / "engagement.json"
    path.write_text(json.dumps(data), encoding="utf-8")
    return Engagement.load(path, now=datetime(2026, 8, 21, 9, tzinfo=UTC))


def test_complete_current_record_is_accepted(tmp_path):
    engagement = load(tmp_path, record())
    engagement.permits(technique="e2-differential", route="GET /objects/{id}")


@pytest.mark.parametrize("field", ["target_owner_acknowledged_at", "abort_contact", "fixture_ids"])
def test_incomplete_record_fails_closed(tmp_path, field):
    data = record()
    del data[field]
    with pytest.raises(AuthorizationError):
        load(tmp_path, data)


def test_expired_record_fails_closed(tmp_path):
    with pytest.raises(AuthorizationError, match="outside"):
        Engagement.load(
            _write(tmp_path, record()), now=datetime(2026, 8, 21, 13, tzinfo=UTC)
        )


def test_unauthorized_route_fails_closed(tmp_path):
    engagement = load(tmp_path, record())
    with pytest.raises(AuthorizationError, match="route"):
        engagement.permits(technique="e2-differential", route="DELETE /objects/{id}")


def _write(tmp_path, data):
    path = tmp_path / "engagement.json"
    path.write_text(json.dumps(data), encoding="utf-8")
    return path