iccery-v2-mac/Tests/ICCeryCoreTests/PrintPanelTests.swift

import Foundation
import XCTest
@testable import ICCeryCore
@testable import ICCery

/// Issue 13 — panel outcome mapping (cancel → nil, ok → result).
/// The real `NSPrintPanel` is never run in tests; these exercise the
/// `UITestHooks` seam the UI tests rely on.
final class PrintPanelStubTests: XCTestCase {

    private func withEnv(
        _ vars: [String: String?],
        _ body: () throws -> Void
    ) rethrows {
        var saved: [String: String?] = [:]
        for key in vars.keys {
            saved[key] = ProcessInfo.processInfo.environment[key]
        }
        for (key, value) in vars {
            if let value { setenv(key, value, 1) } else { unsetenv(key) }
        }
        defer {
            for (key, value) in saved {
                if let value { setenv(key, value, 1) } else { unsetenv(key) }
            }
        }
        try body()
    }

    func testCancelIsNil() throws {
        try withEnv([
            "ICCERY_UI_TESTING": "1",
            "ICCERY_TEST_PRINT_PANEL": "cancel",
        ]) {
            XCTAssertTrue(UITestHooks.printPanelStubbed)
            XCTAssertNil(UITestHooks.printPanelResult(forQueue: "q"))
        }
    }

    func testOkResult() throws {
        try withEnv([
            "ICCERY_UI_TESTING": "1",
            "ICCERY_TEST_PRINT_PANEL": "ok",
            "ICCERY_TEST_PANEL_OPTIONS": "MediaType=Photo InputSlot=Rear",
            "ICCERY_TEST_PANEL_PRINTER": "Other_Queue",
        ]) {
            let result = UITestHooks.printPanelResult(forQueue: "q")
            XCTAssertEqual(result?.selectedPrinter, "Other_Queue")
            XCTAssertEqual(result?.options.cupsOptions, "MediaType=Photo InputSlot=Rear")
            XCTAssertEqual(result?.options.mediaType, "Photo")
            XCTAssertEqual(result?.options.ppdUncorrectedPassthrough, true)
        }
    }

    func testOkDefaultsPrinter() throws {
        try withEnv([
            "ICCERY_UI_TESTING": "1",
            "ICCERY_TEST_PRINT_PANEL": "ok",
            "ICCERY_TEST_PANEL_OPTIONS": nil,
            "ICCERY_TEST_PANEL_PRINTER": nil,
        ]) {
            let result = UITestHooks.printPanelResult(forQueue: "My_Queue")
            XCTAssertEqual(result?.selectedPrinter, "My_Queue")
            XCTAssertNil(result?.options.cupsOptions)
        }
    }
}