kitos
f8824291a2
Aegis CI / lint-and-test (push) Has been cancelled
Snyk Security Scan / Python vulnerabilities (backend) (push) Has been cancelled
Snyk Security Scan / npm vulnerabilities (frontend) (push) Has been cancelled
Snyk Security Scan / Docker image vulnerabilities (backend) (push) Has been cancelled
49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
"""Professional reports router tests (FASE-2.4)."""
|
|
|
|
import os
|
|
import tempfile
|
|
from unittest.mock import patch
|
|
|
|
from app.models.campaign import Campaign
|
|
from app.config import settings
|
|
|
|
|
|
@patch("app.services.report_generation_service.generate_purple_campaign_report")
|
|
def test_purple_campaign_pdf_download(mock_gen, client, auth_headers, db):
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
fake_pdf = os.path.join(tmpdir, "report.pdf")
|
|
with open(fake_pdf, "wb") as f:
|
|
f.write(b"%PDF-1.4 fake")
|
|
mock_gen.return_value = fake_pdf
|
|
|
|
with patch.object(settings, "REPORT_OUTPUT_DIR", tmpdir):
|
|
campaign = Campaign(name="Export Camp", status="active")
|
|
db.add(campaign)
|
|
db.commit()
|
|
|
|
r = client.get(
|
|
f"/api/v1/reports/generate/purple-campaign/{campaign.id}",
|
|
params={"format": "pdf"},
|
|
headers=auth_headers,
|
|
)
|
|
assert r.status_code == 200
|
|
assert r.headers["content-type"] == "application/pdf"
|
|
|
|
|
|
@patch("app.services.report_generation_service.generate_coverage_report")
|
|
def test_coverage_summary_html(mock_gen, client, auth_headers):
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
fake_html = os.path.join(tmpdir, "report.html")
|
|
with open(fake_html, "w") as f:
|
|
f.write("<html><body>ok</body></html>")
|
|
mock_gen.return_value = fake_html
|
|
|
|
with patch.object(settings, "REPORT_OUTPUT_DIR", tmpdir):
|
|
r = client.get(
|
|
"/api/v1/reports/generate/coverage-summary",
|
|
params={"format": "html"},
|
|
headers=auth_headers,
|
|
)
|
|
assert r.status_code == 200
|
|
assert "text/html" in r.headers["content-type"]
|