ReadViewSDK/ReadViewDemo/ReadViewDemoUITests/ReaderUITests/HighlightsManagementTests.swift

import XCTest

final class HighlightsManagementTests: XCTestCase {
    private let app = XCUIApplication()

    override func setUpWithError() throws {
        continueAfterFailure = false
    }

    func testHighlightsPanelOpens() throws {
        app.launchAndOpenSampleBook()
        app.waitForReader(timeout: 12)

        let selectionText = app.otherElements[IDs.readerSelectionText].firstMatch
        guard selectionText.waitForExistence(timeout: 5) else {
            throw XCTSkip("文本选区元素不存在")
        }
        let start = selectionText.coordinate(withNormalizedOffset: CGVector(dx: 0.3, dy: 0.5))
        let end = selectionText.coordinate(withNormalizedOffset: CGVector(dx: 0.6, dy: 0.5))
        start.press(forDuration: 0.8, thenDragTo: end)

        let highlightMenuItem = app.menuItems["划线"]
        if highlightMenuItem.waitForExistence(timeout: 3) { highlightMenuItem.tap() }
        else if app.buttons["划线"].waitForExistence(timeout: 2) { app.buttons["划线"].tap() }

        app.waitForDemoReaderState(timeout: 5, description: "highlights exists") { $0.highlights != nil }

        app.showReaderChromeIfNeeded()
        XCTAssertTrue(app.buttons[IDs.readerHighlights].waitForExistence(timeout: 3), "高亮列表按钮不存在")
        app.buttons[IDs.readerHighlights].tap()

        XCTAssertTrue(app.tables[IDs.readerHighlightsTable].waitForExistence(timeout: 5), "高亮管理面板未出现")
    }

    func testHighlightsPanelShowsCreatedHighlight() throws {
        app.launchAndOpenSampleBook()
        app.waitForReader(timeout: 12)

        let selectionText = app.otherElements[IDs.readerSelectionText].firstMatch
        guard selectionText.waitForExistence(timeout: 5) else { throw XCTSkip("文本选区元素不存在") }
        let start = selectionText.coordinate(withNormalizedOffset: CGVector(dx: 0.3, dy: 0.5))
        let end = selectionText.coordinate(withNormalizedOffset: CGVector(dx: 0.6, dy: 0.5))
        start.press(forDuration: 0.8, thenDragTo: end)

        let highlightMenuItem = app.menuItems["划线"]
        if highlightMenuItem.waitForExistence(timeout: 3) { highlightMenuItem.tap() }

        app.showReaderChromeIfNeeded()
        app.buttons[IDs.readerHighlights].tap()

        let highlightsTable = app.tables[IDs.readerHighlightsTable]
        XCTAssertTrue(highlightsTable.waitForExistence(timeout: 5), "高亮表格未出现")
        XCTAssertTrue(waitForCellCount(in: highlightsTable, minimum: 1, timeout: 5), "创建高亮后面板应至少有 1 行数据")
    }

    func testHighlightsPanelFilterAll() throws {
        app.launchAndOpenSampleBook()
        app.waitForReader(timeout: 12)

        let selectionText = app.otherElements[IDs.readerSelectionText].firstMatch
        guard selectionText.waitForExistence(timeout: 5) else { throw XCTSkip("文本选区元素不存在") }
        let start = selectionText.coordinate(withNormalizedOffset: CGVector(dx: 0.3, dy: 0.5))
        let end = selectionText.coordinate(withNormalizedOffset: CGVector(dx: 0.6, dy: 0.5))
        start.press(forDuration: 0.8, thenDragTo: end)

        let highlightMenuItem = app.menuItems["划线"]
        if highlightMenuItem.waitForExistence(timeout: 3) { highlightMenuItem.tap() }

        app.showReaderChromeIfNeeded()
        app.buttons[IDs.readerHighlights].tap()

        let filterControl = app.segmentedControls[IDs.readerHighlightsFilter]
        XCTAssertTrue(filterControl.waitForExistence(timeout: 5), "筛选控件未出现")
        filterControl.buttons.element(boundBy: 0).tap()

        let highlightsTable = app.tables[IDs.readerHighlightsTable]
        XCTAssertTrue(highlightsTable.waitForExistence(timeout: 3))
        XCTAssertTrue(waitForCellCount(in: highlightsTable, minimum: 1, timeout: 5), "筛选全部后应有高亮数据")
    }

    func testHighlightsPanelEmptyState() throws {
        app.launchAndOpenSampleBook(resetsReaderState: true)
        app.waitForReader(timeout: 12)

        app.showReaderChromeIfNeeded()
        let highlightsButton = app.buttons[IDs.readerHighlights]
        XCTAssertTrue(highlightsButton.waitForExistence(timeout: 3), "高亮列表按钮不存在")
        highlightsButton.tap()

        XCTAssertTrue(app.tables[IDs.readerHighlightsTable].waitForExistence(timeout: 5), "高亮管理面板未出现")
        XCTAssertTrue(app.staticTexts[IDs.readerHighlightsEmptyLabel].waitForExistence(timeout: 5), "空状态 label 应存在")
        XCTAssertTrue(app.staticTexts[IDs.readerHighlightsEmptyLabel].label.contains("暂无"), "空状态应显示暂无标注")
    }

    func testDeleteHighlightFromPanel() throws {
        app.launchAndOpenSampleBook()
        app.waitForReader(timeout: 12)

        let selectionText = app.otherElements[IDs.readerSelectionText].firstMatch
        guard selectionText.waitForExistence(timeout: 5) else { throw XCTSkip("文本选区元素不存在") }
        let start = selectionText.coordinate(withNormalizedOffset: CGVector(dx: 0.3, dy: 0.5))
        let end = selectionText.coordinate(withNormalizedOffset: CGVector(dx: 0.6, dy: 0.5))
        start.press(forDuration: 0.8, thenDragTo: end)

        let highlightMenuItem = app.menuItems["划线"]
        if highlightMenuItem.waitForExistence(timeout: 3) { highlightMenuItem.tap() }

        app.showReaderChromeIfNeeded()
        app.buttons[IDs.readerHighlights].tap()

        let highlightsTable = app.tables[IDs.readerHighlightsTable]
        XCTAssertTrue(highlightsTable.waitForExistence(timeout: 5))
        XCTAssertTrue(waitForCellCount(in: highlightsTable, minimum: 1, timeout: 5))

        let editButton = app.navigationBars.buttons["编辑"].firstMatch
        XCTAssertTrue(editButton.waitForExistence(timeout: 3), "标注列表应提供编辑入口")
        editButton.tap()

        let deleteControl = highlightsTable.buttons["删除"].firstMatch
        XCTAssertTrue(deleteControl.waitForExistence(timeout: 3), "编辑状态应显示删除控件")
        deleteControl.tap()
        // 编辑态的行内“删除”与确认用的“删除”同名；确认按钮是最后出现的那个。
        let deleteButtons = app.buttons.matching(identifier: "删除")
        deleteButtons.element(boundBy: max(0, deleteButtons.count - 1)).tap()

        XCTAssertTrue(app.staticTexts[IDs.readerHighlightsEmptyLabel].waitForExistence(timeout: 5), "删除后应显示空状态")
    }

    private func waitForCellCount(
        in table: XCUIElement,
        minimum: Int,
        timeout: TimeInterval
    ) -> Bool {
        let deadline = Date().addingTimeInterval(timeout)
        while Date() < deadline {
            if table.cells.count >= minimum {
                return true
            }
            RunLoop.current.run(until: Date().addingTimeInterval(0.1))
        }
        return table.cells.count >= minimum
    }
}