Aegis/backend/tests/test_campaign_approval_router.py

"""Router-level tests for the campaign manager-approval workflow."""

from app.models.campaign import Campaign, CampaignTest
from app.models.technique import Technique
from app.models.test import Test
from app.models.enums import TestState


def _make_draft_campaign(db, owner_id):
    tech = Technique(mitre_id="T1059", name="Command Line", tactic="execution", platforms=["windows"])
    db.add(tech)
    db.flush()
    campaign = Campaign(name="Router Test Campaign", type="custom", status="draft", created_by=owner_id)
    db.add(campaign)
    db.flush()
    test = Test(technique_id=tech.id, name="T1059 test", state=TestState.draft, created_by=owner_id)
    db.add(test)
    db.flush()
    db.add(CampaignTest(campaign_id=campaign.id, test_id=test.id, order_index=0))
    db.commit()
    db.refresh(campaign)
    return campaign


def test_lead_can_submit_own_campaign(api, db, red_lead_user, red_lead_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    resp = api("post", f"/api/v1/campaigns/{campaign.id}/submit", red_lead_headers)
    assert resp.status_code == 200
    assert resp.json()["status"] == "pending_approval"


def test_red_tech_cannot_submit(api, db, red_lead_user, red_tech_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    resp = api("post", f"/api/v1/campaigns/{campaign.id}/submit", red_tech_headers)
    assert resp.status_code == 403


def test_manager_can_approve_and_sets_start_date(api, db, red_lead_user, red_lead_headers, manager_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    api("post", f"/api/v1/campaigns/{campaign.id}/submit", red_lead_headers)

    resp = api(
        "post",
        f"/api/v1/campaigns/{campaign.id}/approve",
        manager_headers,
        json={"start_date": "2026-09-01T00:00:00"},
    )
    assert resp.status_code == 200
    body = resp.json()
    assert body["status"] == "active"
    assert body["start_date"] is not None


def test_lead_cannot_approve(api, db, red_lead_user, red_lead_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    api("post", f"/api/v1/campaigns/{campaign.id}/submit", red_lead_headers)

    resp = api(
        "post",
        f"/api/v1/campaigns/{campaign.id}/approve",
        red_lead_headers,
        json={"start_date": "2026-09-01T00:00:00"},
    )
    assert resp.status_code == 403


def test_admin_can_approve_as_manager_backup(api, db, red_lead_user, red_lead_headers, auth_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    api("post", f"/api/v1/campaigns/{campaign.id}/submit", red_lead_headers)

    resp = api(
        "post",
        f"/api/v1/campaigns/{campaign.id}/approve",
        auth_headers,
        json={"start_date": "2026-09-01T00:00:00"},
    )
    assert resp.status_code == 200


def test_manager_can_reject_with_reason(api, db, red_lead_user, red_lead_headers, manager_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    api("post", f"/api/v1/campaigns/{campaign.id}/submit", red_lead_headers)

    resp = api(
        "post",
        f"/api/v1/campaigns/{campaign.id}/reject",
        manager_headers,
        json={"reason": "Needs more detail"},
    )
    assert resp.status_code == 200
    assert resp.json()["status"] == "draft"


def test_manager_reject_without_reason_rejected_by_validation(api, db, red_lead_user, red_lead_headers, manager_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    api("post", f"/api/v1/campaigns/{campaign.id}/submit", red_lead_headers)

    resp = api(
        "post",
        f"/api/v1/campaigns/{campaign.id}/reject",
        manager_headers,
        json={"reason": ""},
    )
    assert resp.status_code == 400


def _activate_campaign_directly(db, campaign):
    """Test helper — bypass the approval flow to get straight to 'active' for setup."""
    campaign.status = "active"
    db.commit()
    db.refresh(campaign)
    return campaign


def test_lead_can_create_modification_request(api, db, red_lead_user, red_lead_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    _activate_campaign_directly(db, campaign)

    tech = Technique(mitre_id="T1547", name="Boot Autostart", tactic="persistence", platforms=["windows"])
    db.add(tech)
    db.flush()
    new_test = Test(technique_id=tech.id, name="Autostart test", state=TestState.draft, created_by=red_lead_user.id)
    db.add(new_test)
    db.commit()

    resp = api(
        "post",
        f"/api/v1/campaigns/{campaign.id}/modification-requests",
        red_lead_headers,
        json={"action": "add_test", "test_id": str(new_test.id), "justification": "Coverage gap"},
    )
    assert resp.status_code == 201
    assert resp.json()["status"] == "pending"


def test_modification_request_without_justification_rejected(api, db, red_lead_user, red_lead_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    _activate_campaign_directly(db, campaign)
    ct = db.query(CampaignTest).filter(CampaignTest.campaign_id == campaign.id).first()

    resp = api(
        "post",
        f"/api/v1/campaigns/{campaign.id}/modification-requests",
        red_lead_headers,
        json={"action": "remove_test", "test_id": str(ct.test_id), "justification": ""},
    )
    assert resp.status_code == 400


def test_manager_can_list_pending_modification_requests(api, db, red_lead_user, red_lead_headers, manager_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    _activate_campaign_directly(db, campaign)
    ct = db.query(CampaignTest).filter(CampaignTest.campaign_id == campaign.id).first()

    api(
        "post",
        f"/api/v1/campaigns/{campaign.id}/modification-requests",
        red_lead_headers,
        json={"action": "remove_test", "test_id": str(ct.test_id), "justification": "Superseded"},
    )

    resp = api("get", "/api/v1/campaigns/modification-requests/pending", manager_headers)
    assert resp.status_code == 200
    assert len(resp.json()) == 1


def test_manager_can_approve_modification_request(api, db, red_lead_user, red_lead_headers, manager_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    _activate_campaign_directly(db, campaign)
    ct = db.query(CampaignTest).filter(CampaignTest.campaign_id == campaign.id).first()

    create_resp = api(
        "post",
        f"/api/v1/campaigns/{campaign.id}/modification-requests",
        red_lead_headers,
        json={"action": "remove_test", "test_id": str(ct.test_id), "justification": "Superseded"},
    )
    request_id = create_resp.json()["id"]

    resp = api(
        "post",
        f"/api/v1/campaigns/modification-requests/{request_id}/approve",
        manager_headers,
    )
    assert resp.status_code == 200
    assert resp.json()["status"] == "approved"

    remaining = db.query(CampaignTest).filter(CampaignTest.campaign_id == campaign.id).all()
    assert len(remaining) == 0


def test_manager_can_reject_modification_request(api, db, red_lead_user, red_lead_headers, manager_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    _activate_campaign_directly(db, campaign)
    ct = db.query(CampaignTest).filter(CampaignTest.campaign_id == campaign.id).first()

    create_resp = api(
        "post",
        f"/api/v1/campaigns/{campaign.id}/modification-requests",
        red_lead_headers,
        json={"action": "remove_test", "test_id": str(ct.test_id), "justification": "Superseded"},
    )
    request_id = create_resp.json()["id"]

    resp = api(
        "post",
        f"/api/v1/campaigns/modification-requests/{request_id}/reject",
        manager_headers,
        json={"review_notes": "Still needed"},
    )
    assert resp.status_code == 200
    assert resp.json()["status"] == "rejected"

    remaining = db.query(CampaignTest).filter(CampaignTest.campaign_id == campaign.id).all()
    assert len(remaining) == 1


def test_lead_cannot_approve_modification_request(api, db, red_lead_user, red_lead_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    _activate_campaign_directly(db, campaign)
    ct = db.query(CampaignTest).filter(CampaignTest.campaign_id == campaign.id).first()

    create_resp = api(
        "post",
        f"/api/v1/campaigns/{campaign.id}/modification-requests",
        red_lead_headers,
        json={"action": "remove_test", "test_id": str(ct.test_id), "justification": "Superseded"},
    )
    request_id = create_resp.json()["id"]

    resp = api(
        "post",
        f"/api/v1/campaigns/modification-requests/{request_id}/approve",
        red_lead_headers,
    )
    assert resp.status_code == 403


def test_direct_add_test_blocked_on_active_campaign_via_router(api, db, red_lead_user, red_lead_headers):
    campaign = _make_draft_campaign(db, red_lead_user.id)
    _activate_campaign_directly(db, campaign)

    tech = Technique(mitre_id="T1547", name="Boot Autostart", tactic="persistence", platforms=["windows"])
    db.add(tech)
    db.flush()
    new_test = Test(technique_id=tech.id, name="Autostart test", state=TestState.draft, created_by=red_lead_user.id)
    db.add(new_test)
    db.commit()

    resp = api(
        "post",
        f"/api/v1/campaigns/{campaign.id}/tests",
        red_lead_headers,
        json={"test_id": str(new_test.id)},
    )
    assert resp.status_code == 400