1

2026-09-17 14:13:31 +08:00
parent ced5db5473
commit 10697d3dfa
18 changed files with 2097 additions and 26 deletions
@@ -0,0 +1,62 @@
+import Foundation
+
+@main
+struct ExpensePreviewTests {
+    static func rejects(_ action: () throws -> Void) {
+        do {
+            try action()
+            preconditionFailure("Unexpected success")
+        } catch {
+            precondition(!error.localizedDescription.isEmpty)
+        }
+    }
+
+    static func main() throws {
+        let root = FileManager.default.temporaryDirectory.appendingPathComponent(UUID().uuidString)
+        try FileManager.default.createDirectory(at: root, withIntermediateDirectories: true)
+        defer { try? FileManager.default.removeItem(at: root) }
+        let source = root.appendingPathComponent("组 模板.XLSX")
+        let data = Data("test worksheet bytes".utf8)
+        try data.write(to: source)
+        let accepted = try ExpenseTemplateImportSource.validate([source])
+        precondition(accepted == source)
+        rejects { _ = try ExpenseTemplateImportSource.validate([]) }
+        rejects { _ = try ExpenseTemplateImportSource.validate([source, source]) }
+        rejects { _ = try ExpenseTemplateImportSource.validate([root]) }
+        rejects { _ = try ExpenseTemplateImportSource.validate([URL(string: "https://example.invalid/template.xlsx")!]) }
+        rejects { _ = try ExpenseTemplateImportSource.validate([root.appendingPathComponent("missing.xlsx")]) }
+        let empty = root.appendingPathComponent("empty.xlsx")
+        try Data().write(to: empty)
+        rejects { _ = try ExpenseTemplateImportSource.validate([empty]) }
+        let oldFormat = root.appendingPathComponent("old.xls")
+        try data.write(to: oldFormat)
+        rejects { _ = try ExpenseTemplateImportSource.validate([oldFormat]) }
+        let oversized = root.appendingPathComponent("large.xlsx")
+        try Data(count: 20 * 1024 * 1024 + 1).write(to: oversized)
+        rejects { _ = try ExpenseTemplateImportSource.validate([oversized]) }
+        let request: [String: Any] = ["purpose": "交通", "amount": "100.30", "payee": ["recipient": "测试"]]
+        let preview = PreparedExpensePreview(file: source, requestDigest: try PreparedExpensePreview.digest(request),
+                                             fileDigest: ExpenseTemplateStorage.fingerprint(data), templateName: "测试",
+                                             uncalculatedCells: [])
+        let verified = try preview.verifiedData(request: request)
+        precondition(verified == data)
+        var changed = request
+        changed["destination"] = "/different/output.xlsx"
+        let relocated = try preview.verifiedData(request: changed)
+        precondition(relocated == data)
+        changed["payee"] = ["recipient": "新收款人"]
+        rejects { _ = try preview.verifiedData(request: changed) }
+        changed = request
+        changed["amount"] = "444.00"
+        rejects { _ = try preview.verifiedData(request: changed) }
+        changed = request
+        changed["templateMapping"] = ["purpose": "D"]
+        rejects { _ = try preview.verifiedData(request: changed) }
+        try Data("changed file".utf8).write(to: source)
+        rejects { _ = try preview.verifiedData(request: request) }
+        preview.remove()
+        precondition(!FileManager.default.fileExists(atPath: source.path))
+        rejects { _ = try preview.verifiedData(request: request) }
+        print("Template drop validation, preview identity, stale input and temporary cleanup tests passed")
+    }
+}