241 lines
10 KiB
Swift
241 lines
10 KiB
Swift
import XCTest
|
||
|
||
/// 图片型 PDF 的文字选择与标注回归。
|
||
///
|
||
/// `host-fixture` 的契约：第 1 页在文本层归一化坐标
|
||
/// `(0.18, 0.20)...(0.65, 0.20)` 上提供至少两条可选择文字；
|
||
/// `none` 则不提供文字/OCR 结果，使 `(0.15, 0.70)...(0.60, 0.84)` 可用于区域框选。
|
||
final class PDFAnnotationTests: XCTestCase {
|
||
private let app = XCUIApplication()
|
||
|
||
override func setUpWithError() throws {
|
||
continueAfterFailure = false
|
||
}
|
||
|
||
func testPDFHostFixtureLongPressSelectsTextWithoutTurningPage() {
|
||
let textLayer = openPDF(textSource: "host-fixture")
|
||
|
||
selectFixtureText(in: textLayer)
|
||
|
||
let fields = waitForTextLayer(textLayer, description: "host text selection") {
|
||
$0["source"] == "text" && ($0["rects"].flatMap(Int.init) ?? 0) > 0
|
||
}
|
||
XCTAssertFalse(fields["text", default: ""].isEmpty, "文字选区应公开已选文本")
|
||
assertReaderPageRemains(1, duration: 0.6)
|
||
}
|
||
|
||
func testPDFHostFixtureCopiesSelectedText() {
|
||
let textLayer = openPDF(textSource: "host-fixture")
|
||
selectFixtureText(in: textLayer)
|
||
|
||
let copyMenuItem = app.menuItems[IDs.pdfSelectionMenuCopy]
|
||
XCTAssertTrue(copyMenuItem.waitForExistence(timeout: 5), "选区菜单缺少复制选项")
|
||
copyMenuItem.tap()
|
||
|
||
let state = app.waitForDemoReaderState(timeout: 5, description: "copy=success and copiedText") {
|
||
$0["copy"] == "success" && !($0["copiedText"] ?? "").isEmpty
|
||
}
|
||
XCTAssertFalse((state["copiedText"] ?? "").isEmpty, "复制成功后应记录被复制的文本")
|
||
}
|
||
|
||
func testPDFHostFixtureCreatesTextHighlight() {
|
||
let textLayer = openPDF(textSource: "host-fixture")
|
||
selectFixtureText(in: textLayer)
|
||
|
||
let highlightMenuItem = app.menuItems[IDs.pdfSelectionMenuHighlight]
|
||
XCTAssertTrue(highlightMenuItem.waitForExistence(timeout: 5), "选区菜单缺少高亮选项")
|
||
highlightMenuItem.tap()
|
||
|
||
let fields = waitForTextLayer(textLayer, description: "one saved text highlight") {
|
||
($0["annotations"].flatMap(Int.init) ?? 0) == 1
|
||
}
|
||
XCTAssertEqual(fields["notes"], "0", "纯高亮不应额外创建笔记")
|
||
}
|
||
|
||
func testPDFAnnotationPersistsAfterClosingAndReopeningBook() {
|
||
let textLayer = openPDF(textSource: "host-fixture")
|
||
selectFixtureText(in: textLayer)
|
||
|
||
let highlightMenuItem = app.menuItems[IDs.pdfSelectionMenuHighlight]
|
||
XCTAssertTrue(highlightMenuItem.waitForExistence(timeout: 5))
|
||
highlightMenuItem.tap()
|
||
_ = waitForTextLayer(textLayer, description: "saved annotation before reopening") {
|
||
($0["annotations"].flatMap(Int.init) ?? 0) == 1
|
||
}
|
||
|
||
// 以不带 reset 参数的全新进程重开，验证标注不会只存在于当前控制器内存中。
|
||
app.terminate()
|
||
app.launchAndOpenSamplePDF(
|
||
bookTitleQuery: "英汉大词典",
|
||
resetsReaderState: false,
|
||
pdfTextSource: "host-fixture"
|
||
)
|
||
app.waitForReader()
|
||
|
||
let reopenedTextLayer = app.otherElements[IDs.pdfTextLayer].firstMatch
|
||
XCTAssertTrue(reopenedTextLayer.waitForExistence(timeout: 8))
|
||
_ = waitForTextLayer(reopenedTextLayer, description: "persisted annotation after reopening") {
|
||
($0["annotations"].flatMap(Int.init) ?? 0) == 1
|
||
}
|
||
}
|
||
|
||
func testPDFAnnotationsPanelListsSavedHighlight() {
|
||
let textLayer = openPDF(textSource: "host-fixture")
|
||
selectFixtureText(in: textLayer)
|
||
|
||
let highlightMenuItem = app.menuItems[IDs.pdfSelectionMenuHighlight]
|
||
XCTAssertTrue(highlightMenuItem.waitForExistence(timeout: 5))
|
||
highlightMenuItem.tap()
|
||
_ = waitForTextLayer(textLayer, description: "saved highlight before opening annotations") {
|
||
($0["annotations"].flatMap(Int.init) ?? 0) == 1
|
||
}
|
||
|
||
app.showReaderChromeIfNeeded(requiredButtonIDs: [IDs.pdfAnnotations])
|
||
let annotationsButton = app.buttons[IDs.pdfAnnotations]
|
||
XCTAssertTrue(annotationsButton.waitForExistence(timeout: 5), "底部工具栏缺少笔记入口")
|
||
annotationsButton.tap()
|
||
|
||
let table = app.tables[IDs.pdfAnnotationsTable]
|
||
XCTAssertTrue(table.waitForExistence(timeout: 5), "笔记列表页未打开")
|
||
XCTAssertGreaterThan(table.cells.count, 0, "笔记列表应显示刚创建的高亮")
|
||
}
|
||
|
||
func testPDFHostFixtureCreatesAnnotation() {
|
||
let textLayer = openPDF(textSource: "host-fixture")
|
||
selectFixtureText(in: textLayer)
|
||
|
||
let annotateMenuItem = app.menuItems[IDs.pdfSelectionMenuAnnotate]
|
||
XCTAssertTrue(annotateMenuItem.waitForExistence(timeout: 5), "选区菜单缺少注释选项")
|
||
annotateMenuItem.tap()
|
||
|
||
let editor = app.otherElements[IDs.pdfAnnotationEditor]
|
||
XCTAssertTrue(editor.waitForExistence(timeout: 5), "注释编辑页未出现")
|
||
let input = app.textViews[IDs.pdfAnnotationInput]
|
||
XCTAssertTrue(input.waitForExistence(timeout: 5), "注释输入框未出现")
|
||
input.tap()
|
||
input.typeText("fixture note")
|
||
|
||
let saveButton = app.buttons[IDs.pdfAnnotationSave]
|
||
XCTAssertTrue(saveButton.waitForExistence(timeout: 5), "注释编辑页缺少保存按钮")
|
||
saveButton.tap()
|
||
|
||
let fields = waitForTextLayer(textLayer, description: "one saved note annotation") {
|
||
($0["annotations"].flatMap(Int.init) ?? 0) == 1
|
||
&& ($0["notes"].flatMap(Int.init) ?? 0) == 1
|
||
}
|
||
XCTAssertEqual(fields["notes"], "1", "保存注释后页面应显示笔记标记")
|
||
}
|
||
|
||
func testPDFWithoutOCRCreatesRegionHighlightWithoutCopy() {
|
||
let textLayer = openPDF(textSource: "none")
|
||
_ = waitForTextLayer(textLayer, description: "no OCR text runs") {
|
||
($0["textRuns"].flatMap(Int.init) ?? -1) == 0
|
||
}
|
||
|
||
let start = textLayer.coordinate(withNormalizedOffset: CGVector(dx: 0.15, dy: 0.70))
|
||
let end = textLayer.coordinate(withNormalizedOffset: CGVector(dx: 0.60, dy: 0.84))
|
||
start.press(forDuration: 0.55, thenDragTo: end)
|
||
|
||
_ = waitForTextLayer(textLayer, description: "region selection") {
|
||
$0["source"] == "region" && ($0["rects"].flatMap(Int.init) ?? 0) > 0
|
||
}
|
||
let highlightMenuItem = app.menuItems[IDs.pdfSelectionMenuHighlight]
|
||
XCTAssertTrue(highlightMenuItem.waitForExistence(timeout: 5), "区域选区应弹出带高亮选项的菜单")
|
||
XCTAssertFalse(
|
||
app.menuItems[IDs.pdfSelectionMenuCopy].waitForExistence(timeout: 1),
|
||
"无 OCR 的区域选区不应提供复制选项"
|
||
)
|
||
highlightMenuItem.tap()
|
||
|
||
_ = waitForTextLayer(textLayer, description: "one saved region highlight") {
|
||
($0["annotations"].flatMap(Int.init) ?? 0) == 1
|
||
}
|
||
assertReaderPageRemains(1, duration: 0.4)
|
||
}
|
||
|
||
@discardableResult
|
||
private func openPDF(textSource: String) -> XCUIElement {
|
||
app.launchAndOpenSamplePDF(
|
||
bookTitleQuery: "英汉大词典",
|
||
resetsReaderState: true,
|
||
pdfTextSource: textSource
|
||
)
|
||
app.waitForReader()
|
||
app.waitForDemoReaderState(timeout: 8, description: "PDF page 1") {
|
||
$0.mode == "pdf" && $0.page == 1
|
||
}
|
||
|
||
let textLayer = app.otherElements[IDs.pdfTextLayer].firstMatch
|
||
XCTAssertTrue(textLayer.waitForExistence(timeout: 8), "PDF 图片文字层未出现")
|
||
return textLayer
|
||
}
|
||
|
||
private func selectFixtureText(in textLayer: XCUIElement) {
|
||
_ = waitForTextLayer(textLayer, description: "host fixture text runs") {
|
||
($0["textRuns"].flatMap(Int.init) ?? 0) >= 2
|
||
}
|
||
|
||
let start = textLayer.coordinate(withNormalizedOffset: CGVector(dx: 0.18, dy: 0.20))
|
||
let end = textLayer.coordinate(withNormalizedOffset: CGVector(dx: 0.65, dy: 0.20))
|
||
start.press(forDuration: 0.55, thenDragTo: end)
|
||
|
||
XCTAssertTrue(
|
||
app.menuItems[IDs.pdfSelectionMenuHighlight].waitForExistence(timeout: 5),
|
||
"长按拖选文字后未弹出选区菜单"
|
||
)
|
||
}
|
||
|
||
@discardableResult
|
||
private func waitForTextLayer(
|
||
_ textLayer: XCUIElement,
|
||
timeout: TimeInterval = 6,
|
||
description: String,
|
||
where predicate: ([String: String]) -> Bool
|
||
) -> [String: String] {
|
||
let deadline = Date().addingTimeInterval(timeout)
|
||
var lastFields: [String: String] = [:]
|
||
|
||
while Date() < deadline {
|
||
lastFields = textLayerFields(textLayer)
|
||
if predicate(lastFields) {
|
||
return lastFields
|
||
}
|
||
RunLoop.current.run(until: Date().addingTimeInterval(0.1))
|
||
}
|
||
|
||
XCTFail("PDF 文本层未满足 \(description)，当前：\(String(describing: textLayer.value))")
|
||
return lastFields
|
||
}
|
||
|
||
private func textLayerFields(_ textLayer: XCUIElement) -> [String: String] {
|
||
guard let rawValue = textLayer.value as? String else { return [:] }
|
||
return rawValue
|
||
.split(separator: ";", omittingEmptySubsequences: true)
|
||
.reduce(into: [:]) { fields, token in
|
||
let parts = token.split(separator: "=", maxSplits: 1, omittingEmptySubsequences: false)
|
||
guard parts.count == 2 else { return }
|
||
fields[String(parts[0])] = String(parts[1])
|
||
}
|
||
}
|
||
|
||
private func assertReaderPageRemains(
|
||
_ page: Int,
|
||
duration: TimeInterval,
|
||
file: StaticString = #filePath,
|
||
line: UInt = #line
|
||
) {
|
||
let deadline = Date().addingTimeInterval(duration)
|
||
repeat {
|
||
guard app.currentDemoReaderState()?.page == page else {
|
||
XCTFail(
|
||
"PDF 选择手势意外翻页，当前状态：\(app.currentDemoReaderState()?.rawValue ?? "<missing>")",
|
||
file: file,
|
||
line: line
|
||
)
|
||
return
|
||
}
|
||
RunLoop.current.run(until: Date().addingTimeInterval(0.1))
|
||
} while Date() < deadline
|
||
}
|
||
}
|