test: make sure getSelectedText will not crash

tisfeng · Nov 9, 2024 · e55735f · e55735f
1 parent 27551af
commit e55735f
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/EasydictSwiftTests/Test.swift b/EasydictSwiftTests/Test.swift
@@ -6,6 +6,7 @@
 //  Copyright © 2024 izual. All rights reserved.
 //
 
+import SelectedTextKit
 import Testing
 import Translation
 
@@ -62,3 +63,24 @@ import Translation
 @Test func setAlertVolume() async throws {
     try await AppleScriptTask.setAlertVolume(50)
 }
+
+@Test func testGetSelectedText() async {
+    // Run thousands of times to test crash.
+    for i in 0..<2000 {
+        print("test index: \(i)")
+        let selectedText = await (try? getSelectedText()) ?? ""
+        print("\(i) selectedText: \(selectedText)")
+    }
+}
+
+@Test func testConcurrentGetSelectedText() async throws {
+    await withTaskGroup(of: Void.self) { group in
+        for i in 0..<2000 {
+            group.addTask {
+                print("test index: \(i)")
+                let selectedText = (try? await getSelectedText()) ?? ""
+                print("\(i) selectedText: \(selectedText)")
+            }
+        }
+    }
+}