198 lines
7.2 KiB
Python
198 lines
7.2 KiB
Python
"""Unit tests for script_generator.py — SCRPT-01, SCRPT-02, SCRPT-03, SCRPT-04, SCRPT-05."""
|
|
import pytest
|
|
from imptune.generators.script_generator import render_detect, render_install, render_uninstall
|
|
|
|
|
|
def _create_test_driver_and_printer():
|
|
"""Helper: create a Driver + Printer for integration tests."""
|
|
from imptune.db.models import Driver, Printer
|
|
|
|
driver = Driver.create(
|
|
sha256="abc123",
|
|
original_filename="test.zip",
|
|
size_bytes=100,
|
|
driver_desc='["Test Driver"]',
|
|
inf_filename="test.inf",
|
|
)
|
|
printer = Printer.create(
|
|
name="Test Printer",
|
|
ip_address="10.0.0.1",
|
|
port_name="IP_10.0.0.1",
|
|
driver=driver,
|
|
duplex_mode="LongEdge",
|
|
color_mode=True,
|
|
paper_size="A4",
|
|
collate=True,
|
|
)
|
|
return driver, printer
|
|
|
|
|
|
SAMPLE_ARGS = dict(
|
|
printer_name="HP LaserJet 4000",
|
|
ip_address="192.168.1.10",
|
|
port_name="IP_192.168.1.10",
|
|
driver_name="HP Universal Printing PCL 6",
|
|
inf_filename="hpcu270u.inf",
|
|
duplex_mode="LongEdge",
|
|
color_mode=True,
|
|
paper_size="A4",
|
|
collate=True,
|
|
)
|
|
|
|
|
|
def test_render_install_returns_string():
|
|
"""render_install returns a non-empty string."""
|
|
result = render_install(**SAMPLE_ARGS)
|
|
assert isinstance(result, str)
|
|
assert len(result) > 0
|
|
|
|
|
|
def test_render_install_contains_pnputil():
|
|
"""SCRPT-01: Output contains pnputil /add-driver and Add-PrinterDriver."""
|
|
result = render_install(**SAMPLE_ARGS)
|
|
assert "pnputil.exe /add-driver" in result
|
|
assert "Add-PrinterDriver" in result
|
|
assert "hpcu270u.inf" in result
|
|
assert "HP Universal Printing PCL 6" in result
|
|
|
|
|
|
def test_render_install_print_config():
|
|
"""SCRPT-01: Set-PrintConfiguration with translated duplex mode."""
|
|
# LongEdge -> TwoSidedLongEdge
|
|
result = render_install(**SAMPLE_ARGS)
|
|
assert "Set-PrintConfiguration" in result
|
|
assert "TwoSidedLongEdge" in result
|
|
assert "A4" in result
|
|
|
|
# ShortEdge -> TwoSidedShortEdge
|
|
result_short = render_install(**{**SAMPLE_ARGS, "duplex_mode": "ShortEdge"})
|
|
assert "TwoSidedShortEdge" in result_short
|
|
|
|
# OneSided -> OneSided (no change)
|
|
result_one = render_install(**{**SAMPLE_ARGS, "duplex_mode": "OneSided"})
|
|
assert "OneSided" in result_one
|
|
# Must NOT contain the two-sided variants when OneSided
|
|
assert "TwoSided" not in result_one
|
|
|
|
|
|
def test_render_install_wow64_guard():
|
|
"""SCRPT-05: WOW64 relaunch guard is present in the output."""
|
|
result = render_install(**SAMPLE_ARGS)
|
|
assert "PROCESSOR_ARCHITECTURE" in result
|
|
assert "SysNative" in result
|
|
# Guard must appear near the top (before pnputil)
|
|
wow64_pos = result.find("PROCESSOR_ARCHITECTURE")
|
|
pnputil_pos = result.find("pnputil.exe")
|
|
assert wow64_pos < pnputil_pos, "WOW64 guard must appear before pnputil"
|
|
|
|
|
|
def test_render_install_uac_guard():
|
|
"""SCRPT-04: SYSTEM vs user detection with UAC self-elevation."""
|
|
result = render_install(**SAMPLE_ARGS)
|
|
assert "IsSystem" in result
|
|
assert "Start-Process" in result
|
|
assert "-Verb Runas" in result or "Verb Runas" in result
|
|
|
|
|
|
def test_render_install_idempotency():
|
|
"""SCRPT-01: All add operations wrapped in idempotency checks."""
|
|
result = render_install(**SAMPLE_ARGS)
|
|
# Port idempotency
|
|
assert "Get-PrinterPort" in result
|
|
assert "Add-PrinterPort" in result
|
|
# Printer idempotency
|
|
assert "Get-Printer" in result
|
|
assert "Add-Printer" in result
|
|
# Port check must come before Add-PrinterPort
|
|
get_port_pos = result.find("Get-PrinterPort")
|
|
add_port_pos = result.find("Add-PrinterPort")
|
|
assert get_port_pos < add_port_pos, "Get-PrinterPort check must precede Add-PrinterPort"
|
|
|
|
|
|
def test_render_install_booleans():
|
|
"""Color and collate are rendered as $true / $false in the script."""
|
|
result_true = render_install(**SAMPLE_ARGS) # color_mode=True, collate=True
|
|
assert "$true" in result_true
|
|
|
|
result_false = render_install(**{**SAMPLE_ARGS, "color_mode": False, "collate": False})
|
|
assert "$false" in result_false
|
|
|
|
|
|
def test_render_uninstall():
|
|
"""SCRPT-02: render_uninstall produces correct removal order with SilentlyContinue."""
|
|
result = render_uninstall("Test Printer", "HP Driver", "IP_10.0.0.1")
|
|
# All three Remove-* commands must be present
|
|
assert 'Remove-Printer -Name "Test Printer"' in result
|
|
assert 'Remove-PrinterDriver -Name "HP Driver"' in result
|
|
assert 'Remove-PrinterPort -Name "IP_10.0.0.1"' in result
|
|
# All must have -ErrorAction SilentlyContinue
|
|
assert result.count("-ErrorAction SilentlyContinue") >= 3
|
|
# Order: Remove-Printer BEFORE Remove-PrinterDriver (driver removal fails if printer still active)
|
|
printer_pos = result.find("Remove-Printer -Name")
|
|
driver_pos = result.find("Remove-PrinterDriver -Name")
|
|
port_pos = result.find("Remove-PrinterPort -Name")
|
|
assert printer_pos < driver_pos, "Remove-Printer must appear before Remove-PrinterDriver"
|
|
assert driver_pos < port_pos, "Remove-PrinterDriver must appear before Remove-PrinterPort"
|
|
|
|
|
|
def test_render_detect():
|
|
"""SCRPT-03: render_detect follows Intune detection contract (Write-Output + exit 0/1)."""
|
|
result = render_detect("Test Printer")
|
|
assert 'Get-Printer -Name "Test Printer"' in result
|
|
assert "Write-Output" in result
|
|
assert "exit 0" in result
|
|
assert "exit 1" in result
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Integration tests — script API endpoints
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_install_endpoint(client):
|
|
"""GET /printers/{id}/scripts/install returns 200 with pnputil in content."""
|
|
_driver, printer = _create_test_driver_and_printer()
|
|
response = client.get(f"/printers/{printer.id}/scripts/install")
|
|
assert response.status_code == 200
|
|
assert "pnputil" in response.text
|
|
|
|
|
|
def test_uninstall_endpoint(client):
|
|
"""GET /printers/{id}/scripts/uninstall returns 200 with Remove-Printer in content."""
|
|
_driver, printer = _create_test_driver_and_printer()
|
|
response = client.get(f"/printers/{printer.id}/scripts/uninstall")
|
|
assert response.status_code == 200
|
|
assert "Remove-Printer" in response.text
|
|
|
|
|
|
def test_detect_endpoint(client):
|
|
"""GET /printers/{id}/scripts/detect returns 200 with Write-Output in content."""
|
|
_driver, printer = _create_test_driver_and_printer()
|
|
response = client.get(f"/printers/{printer.id}/scripts/detect")
|
|
assert response.status_code == 200
|
|
assert "Write-Output" in response.text
|
|
|
|
|
|
def test_script_endpoint_missing_printer(client):
|
|
"""GET /printers/{id}/scripts/install returns 404 for nonexistent printer."""
|
|
response = client.get("/printers/9999/scripts/install")
|
|
assert response.status_code == 404
|
|
|
|
|
|
def test_script_endpoint_no_driver(client):
|
|
"""GET /printers/{id}/scripts/install returns 422 when no driver assigned."""
|
|
from imptune.db.models import Printer
|
|
|
|
printer = Printer.create(
|
|
name="No Driver Printer",
|
|
ip_address="10.0.0.2",
|
|
port_name="IP_10.0.0.2",
|
|
driver=None,
|
|
duplex_mode="OneSided",
|
|
color_mode=True,
|
|
paper_size="A4",
|
|
collate=True,
|
|
)
|
|
response = client.get(f"/printers/{printer.id}/scripts/install")
|
|
assert response.status_code == 422
|