RouterOS/RouterOSAssistantTests/PortScannerTests.swift

import XCTest
@testable import RouterOSAssistant

final class PortScannerTests: XCTestCase {
    func testCommonPortsHaveNoDuplicates() {
        let ports = PortScanner.commonPorts.map(\.port)
        XCTAssertEqual(ports.count, Set(ports).count)
    }

    func testCommonPortsAreAllValidTCPPorts() {
        for info in PortScanner.commonPorts {
            XCTAssertTrue((1...65535).contains(info.port), "Port \(info.port) out of valid range")
        }
    }

    /// A high, essentially-never-bound port on loopback reliably gets an immediate TCP RST
    /// (connection refused) rather than a timeout, so this is fast and deterministic — not
    /// dependent on any real network/router.
    func testScanReportsClosedForRefusedLoopbackPort() async {
        let results = await PortScanner.scan(
            host: "127.0.0.1",
            ports: [PortScanner.PortInfo(port: 59_999, serviceName: nil)],
            timeout: 2
        )
        XCTAssertEqual(results.count, 1)
        XCTAssertEqual(results.first?.status, .closed)
    }

    func testScanSortsResultsByPortNumber() async {
        let ports = [
            PortScanner.PortInfo(port: 59_998, serviceName: nil),
            PortScanner.PortInfo(port: 59_997, serviceName: nil)
        ]
        let results = await PortScanner.scan(host: "127.0.0.1", ports: ports, timeout: 2)
        XCTAssertEqual(results.map(\.port.port), [59_997, 59_998])
    }
}