OSGKeyboard/OSGKeyboardTests/AIClipboardPromptTests.swift

// AIClipboardPromptTests.swift
// OSGKeyboardTests

@testable import OSGKeyboardShared
import XCTest

final class AIClipboardPromptTests: XCTestCase {
    func testInternalClipboardEnvelopeIsDetected() {
        let prompt = AIClipboardPrompt.compose(
            instruction: "请翻译剪贴板",
            material: "hello"
        )
        XCTAssertTrue(AIClipboardPrompt.isInternalPrompt(prompt))
        XCTAssertFalse(AIClipboardPrompt.isInternalPrompt("请翻译剪贴板"))
    }

    func testComposeKeepsInstructionAndMaterialInSeparateBlocks() {
        let prompt = AIClipboardPrompt.compose(
            instruction: "请翻译剪贴板",
            material: "会议改到 <A 栋>"
        )
        XCTAssertTrue(prompt.contains("<instruction>"))
        XCTAssertTrue(prompt.contains("<clipboard_text>"))
        XCTAssertTrue(prompt.contains("请翻译剪贴板"))
        // Untrusted material can never open its own tags.
        XCTAssertTrue(prompt.contains("&lt;A 栋&gt;"))
        XCTAssertFalse(prompt.contains("<A 栋>"))
    }

    func testConversationContextIsAnUntrustedBlockSeparateFromInstruction() {
        let prompt = AIClipboardPrompt.compose(
            instruction: "Reply to the clipboard message.",
            material: "周六一起吃饭？",
            conversationContext: "Them: 在吗\nYou: 在的 <随时>"
        )
        XCTAssertTrue(prompt.contains("<conversation_context>"))
        // Context is untrusted data too: its angle brackets are escaped so it
        // cannot break out into new tags.
        XCTAssertTrue(prompt.contains("&lt;随时&gt;"))
        XCTAssertFalse(prompt.contains("<随时>"))
    }

    func testComposeOmitsContextBlockWhenAbsentOrBlank() {
        XCTAssertFalse(
            AIClipboardPrompt.compose(instruction: "i", material: "m")
                .contains("conversation_context")
        )
        XCTAssertFalse(
            AIClipboardPrompt.compose(
                instruction: "i",
                material: "m",
                conversationContext: "   "
            ).contains("conversation_context")
        )
    }

    func testResolveThreadsConversationContextIntoEnvelope() throws {
        let resolution = AIClipboardPrompt.resolve(
            instruction: "Reply.",
            material: "周六一起吃饭？",
            conversationContext: "Them: 在吗\nYou: 在的"
        )
        guard case .ready(let prompt) = resolution else {
            return XCTFail("expected a composed prompt")
        }
        XCTAssertTrue(prompt.contains("<conversation_context>"))
        XCTAssertTrue(prompt.contains("在吗"))
    }

    func testResolveFailsClosedWithoutMaterial() {
        XCTAssertEqual(
            AIClipboardPrompt.resolve(instruction: "请翻译剪贴板", material: nil),
            .materialUnavailable
        )
        XCTAssertEqual(
            AIClipboardPrompt.resolve(instruction: "请翻译剪贴板", material: "   "),
            .materialUnavailable
        )
    }

    func testSpokenQuestionWithoutClipboardIntentPassesThrough() {
        XCTAssertEqual(
            AIClipboardPrompt.resolveSpoken(question: "明天上海天气如何", material: "机密内容"),
            .ready("明天上海天气如何")
        )
    }

    func testSpokenClipboardIntentAttachesMaterial() throws {
        let resolution = AIClipboardPrompt.resolveSpoken(
            question: "帮我回复剪贴板里的这条消息",
            material: "周五下午两点可以吗？"
        )
        guard case .ready(let prompt) = resolution else {
            return XCTFail("expected a composed prompt")
        }
        XCTAssertTrue(prompt.contains("周五下午两点可以吗？"))
        XCTAssertTrue(prompt.contains("clipboard_request"))
    }

    func testSpokenClipboardIntentFailsClosedWhenHistoryIsEmpty() {
        XCTAssertEqual(
            AIClipboardPrompt.resolveSpoken(question: "translate my clipboard", material: nil),
            .materialUnavailable
        )
    }

    func testClipboardCardFailsClosedAfterEligibilityWindow() {
        let card = AIHintCard(
            id: "remote-clipboard-reply",
            displayText: "回复剪贴板",
            prompt: "请回复剪贴板内容",
            category: "clipboard",
            source: "remote",
            locale: "zh",
            conditions: ["clipboard_30s"]
        )
        XCTAssertEqual(
            AIHintPool.resolvePrompt(for: card, clipboardText: nil),
            .materialUnavailable
        )
    }

    func testNonClipboardCardDropsLegacyPlaceholder() throws {
        let card = AIHintCard(
            id: "remote-1",
            displayText: "聊聊热点",
            prompt: "请概括今日热点 \(AIClipboardPrompt.materialPlaceholder)",
            category: "society"
        )
        guard case .ready(let prompt) = AIHintPool.resolvePrompt(
            for: card,
            clipboardText: "无关内容"
        ) else {
            return XCTFail("expected a ready prompt")
        }
        XCTAssertEqual(prompt, "请概括今日热点")
        XCTAssertFalse(prompt.contains("无关内容"))
    }
}