kawa
d3590017e9
- Create tests/test_script_download.py with 5 tests for .ps1 routes (install/uninstall/detect, 404, 422) - Add test_detail_page_shows_script_links to TestCommandPreview in tests/test_packages.py - All 6 new tests go RED (routes and template links do not exist yet)
121 lines
4.4 KiB
Python
121 lines
4.4 KiB
Python
"""Integration tests for the .ps1-suffixed script download routes."""
|
|
import io
|
|
import json
|
|
import zipfile
|
|
|
|
import pytest
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Fixtures (same pattern as tests/test_packages.py)
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
@pytest.fixture
|
|
def driver_zip_bytes():
|
|
"""Create a minimal valid driver ZIP with a fake INF and CAT file."""
|
|
buf = io.BytesIO()
|
|
with zipfile.ZipFile(buf, "w", compression=zipfile.ZIP_DEFLATED) as zf:
|
|
zf.writestr("printer.inf", "[Version]\nSignature=$WINDOWS NT$\n")
|
|
zf.writestr("printer.cat", "FAKE_CAT")
|
|
return buf.getvalue()
|
|
|
|
|
|
@pytest.fixture
|
|
def setup_printer_with_driver(tmp_data_dir, driver_zip_bytes):
|
|
"""Create Driver record (with ZIP on disk) and Printer record linked to it."""
|
|
import hashlib
|
|
import os
|
|
|
|
import imptune.config as cfg
|
|
from imptune.db.models import Driver, Printer
|
|
|
|
sha = hashlib.sha256(driver_zip_bytes).hexdigest()
|
|
drivers_dir = cfg.DRIVERS_DIR
|
|
os.makedirs(drivers_dir, exist_ok=True)
|
|
zip_path = os.path.join(drivers_dir, f"{sha}.zip")
|
|
with open(zip_path, "wb") as f:
|
|
f.write(driver_zip_bytes)
|
|
|
|
driver = Driver.create(
|
|
sha256=sha,
|
|
original_filename="printer_driver.zip",
|
|
size_bytes=len(driver_zip_bytes),
|
|
driver_desc=json.dumps(["HP LaserJet Pro"]),
|
|
inf_filename="printer.inf",
|
|
)
|
|
printer = Printer.create(
|
|
name="Test Printer",
|
|
ip_address="192.168.1.100",
|
|
port_name="IP_192.168.1.100",
|
|
driver=driver,
|
|
duplex_mode="OneSided",
|
|
color_mode=True,
|
|
paper_size="A4",
|
|
collate=True,
|
|
)
|
|
return printer, driver
|
|
|
|
|
|
@pytest.fixture
|
|
def printer_no_driver(tmp_data_dir):
|
|
"""Create Printer record with no driver assigned."""
|
|
from imptune.db.models import Printer
|
|
|
|
return Printer.create(
|
|
name="No Driver Printer",
|
|
ip_address="10.0.0.1",
|
|
port_name="IP_10.0.0.1",
|
|
driver=None,
|
|
)
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# .ps1 route tests
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
class TestPs1Routes:
|
|
def test_install_ps1_route(self, client, setup_printer_with_driver):
|
|
"""GET /printers/{id}/scripts/install.ps1 returns 200 with attachment and PowerShell content."""
|
|
printer, _ = setup_printer_with_driver
|
|
resp = client.get(f"/printers/{printer.id}/scripts/install.ps1")
|
|
assert resp.status_code == 200
|
|
assert 'attachment' in resp.headers["content-disposition"]
|
|
assert 'filename="install.ps1"' in resp.headers["content-disposition"]
|
|
body = resp.text
|
|
assert len(body) > 0
|
|
assert "Add-Printer" in body or "$PSScriptRoot" in body
|
|
|
|
def test_uninstall_ps1_route(self, client, setup_printer_with_driver):
|
|
"""GET /printers/{id}/scripts/uninstall.ps1 returns 200 with attachment and uninstall content."""
|
|
printer, _ = setup_printer_with_driver
|
|
resp = client.get(f"/printers/{printer.id}/scripts/uninstall.ps1")
|
|
assert resp.status_code == 200
|
|
assert 'attachment' in resp.headers["content-disposition"]
|
|
assert 'filename="uninstall.ps1"' in resp.headers["content-disposition"]
|
|
body = resp.text
|
|
assert len(body) > 0
|
|
assert "Remove-Printer" in body
|
|
|
|
def test_detect_ps1_route(self, client, setup_printer_with_driver):
|
|
"""GET /printers/{id}/scripts/detect.ps1 returns 200 with attachment and detect content."""
|
|
printer, _ = setup_printer_with_driver
|
|
resp = client.get(f"/printers/{printer.id}/scripts/detect.ps1")
|
|
assert resp.status_code == 200
|
|
assert 'attachment' in resp.headers["content-disposition"]
|
|
assert 'filename="detect.ps1"' in resp.headers["content-disposition"]
|
|
body = resp.text
|
|
assert len(body) > 0
|
|
assert "Get-Printer" in body
|
|
|
|
def test_ps1_routes_missing_printer(self, client, tmp_data_dir):
|
|
"""GET for non-existent printer returns 404."""
|
|
resp = client.get("/printers/99999/scripts/install.ps1")
|
|
assert resp.status_code == 404
|
|
|
|
def test_ps1_routes_no_driver(self, client, printer_no_driver):
|
|
"""GET for printer without driver returns 422."""
|
|
resp = client.get(f"/printers/{printer_no_driver.id}/scripts/install.ps1")
|
|
assert resp.status_code == 422
|