OSGKeyboard/OSGKeyboardTests/PolishOutputValidatorTests.swift

import XCTest
@testable import OSGKeyboardShared

final class PolishOutputValidatorTests: XCTestCase {
    func testMissingDictionaryCanonicalTermIsHardViolation() {
        let dictionary = PersonalDictionary(entries: [
            .init(
                term: "Kubernetes",
                aliases: ["k8s"],
                category: .productName,
                source: .manual
            ),
        ])
        let violations = PolishOutputValidator.validate(
            input: "部署 k8s 集群",
            output: "部署容器集群",
            dictionary: dictionary,
            lengthRatio: 0.5...2
        )
        XCTAssertTrue(violations.contains(.missingDictionaryTerms(["Kubernetes"])))
        XCTAssertTrue(violations.contains(where: \.isHard))
    }

    func testIdentifiersArePreservedExactly() {
        let violations = PolishOutputValidator.validate(
            input: "send https://example.com/a to dev@example.com using user_id",
            output: "send it to the team",
            dictionary: .empty,
            lengthRatio: 0.5...2
        )
        XCTAssertTrue(violations.contains { violation in
            if case .missingIdentifiers(let values) = violation {
                return values.contains("https://example.com/a")
                    && values.contains("dev@example.com")
                    && values.contains("user_id")
            }
            return false
        })
    }

    func testDatesFractionsAndSlashWordsAreNotProtectedPaths() {
        let cases = [
            ("在 2025/03/01 之前完成", "在2025年3月1日之前完成"),
            ("价格是 3/4 杯面粉", "价格是四分之三杯面粉"),
            ("我给 3/5 分", "我给五分之三"),
            ("读一下 and/or 的用法", "读一下 and or 的用法"),
        ]

        for (input, output) in cases {
            let violations = PolishOutputValidator.validate(
                input: input,
                output: output,
                dictionary: .empty,
                lengthRatio: 0.2...3
            )
            XCTAssertFalse(
                violations.contains {
                    if case .missingIdentifiers = $0 { return true }
                    return false
                },
                "Must not classify slash value as a protected path: \(input)"
            )
        }
    }

    func testStrongPathSignalsRemainHardProtectedIdentifiers() {
        let inputs = [
            "/usr/local/bin",
            "../Sources/App.swift",
            "Sources/Features/Auth",
            "src/user_id",
        ]
        for input in inputs {
            let violations = PolishOutputValidator.validate(
                input: "打开 \(input)",
                output: "打开对应文件",
                dictionary: .empty,
                lengthRatio: 0.2...3
            )
            XCTAssertTrue(
                violations.contains {
                    if case .missingIdentifiers(let values) = $0 {
                        return values.contains(input)
                    }
                    return false
                },
                "Expected hard path protection for \(input)"
            )
        }
    }

    func testOrdinalASRRepairDoesNotReportMissingZeroes() {
        let violations = PolishOutputValidator.validate(
            input: "第一点是A第2:00是B",
            output: "第一点是 A\n2. B",
            dictionary: .empty,
            lengthRatio: 0.2...3
        )
        XCTAssertFalse(violations.contains {
            if case .missingNumbers = $0 { return true }
            return false
        })
    }

    func testRealTimeStillReportsMissingZeroes() {
        let violations = PolishOutputValidator.validate(
            input: "第一点是坐第2:00班车",
            output: "第一点是坐第二班车",
            dictionary: .empty,
            lengthRatio: 0.2...3
        )
        XCTAssertTrue(violations.contains {
            if case .missingNumbers(let values) = $0 {
                return values.contains("00")
            }
            return false
        })
    }

    func testNumbersLengthAndLanguageAreObservationOnly() {
        let violations = PolishOutputValidator.validate(
            input: "项目 123 明天下午交付并通知全部相关成员",
            output: "Ship tomorrow.",
            dictionary: .empty,
            lengthRatio: 0.9...1.1
        )
        XCTAssertFalse(violations.isEmpty)
        XCTAssertTrue(violations.filter(\.isHard).isEmpty)
    }
}