add swiftUI code

2025-01-22 14:09:10 +08:00
parent 68e7b7347c
commit 8a99853829
2531 changed files with 486215 additions and 0 deletions
@@ -0,0 +1,73 @@
+import XCTest
+import GRDB
+
+private struct FTS3TokenizeModule : VirtualTableModule {
+    let moduleName = "fts3tokenize"
+    
+    func makeTableDefinition(configuration: VirtualTableConfiguration) -> FTS3TokenizeTableDefinition {
+        FTS3TokenizeTableDefinition()
+    }
+    
+    func moduleArguments(for definition: FTS3TokenizeTableDefinition, in db: Database) -> [String] {
+        guard let tokenizer = definition.tokenizer else {
+            return []
+        }
+        return [tokenizer]
+    }
+    
+    func database(_ db: Database, didCreate tableName: String, using definition: FTS3TokenizeTableDefinition) throws {
+    }
+}
+
+private struct ThrowingFTS3TokenizeModule : VirtualTableModule {
+    let moduleName = "fts3tokenize"
+    
+    func makeTableDefinition(configuration: VirtualTableConfiguration) -> FTS3TokenizeTableDefinition {
+        FTS3TokenizeTableDefinition()
+    }
+    
+    func moduleArguments(for definition: FTS3TokenizeTableDefinition, in db: Database) -> [String] {
+        guard let tokenizer = definition.tokenizer else {
+            return []
+        }
+        return [tokenizer]
+    }
+    
+    func database(_ db: Database, didCreate tableName: String, using definition: FTS3TokenizeTableDefinition) throws {
+        throw DatabaseError(resultCode: ResultCode(rawValue: 123))
+    }
+}
+
+private final class FTS3TokenizeTableDefinition {
+    var tokenizer: String?
+}
+
+class VirtualTableModuleTests: GRDBTestCase {
+    
+    func testCustomVirtualTableModule() throws {
+        let dbQueue = try makeDatabaseQueue()
+        try dbQueue.inDatabase { db in
+            try db.create(virtualTable: "test", using: FTS3TokenizeModule()) { t in
+                t.tokenizer = "simple"
+            }
+            assertDidExecute(sql: "CREATE VIRTUAL TABLE \"test\" USING fts3tokenize(simple)")
+            XCTAssertTrue(try db.tableExists("test"))
+        }
+    }
+
+    func testThrowingCustomVirtualTableModule() throws {
+        let dbQueue = try makeDatabaseQueue()
+        try dbQueue.inDatabase { db in
+            do {
+                try db.create(virtualTable: "test", using: ThrowingFTS3TokenizeModule()) { t in
+                    t.tokenizer = "simple"
+                }
+                XCTFail("Expected DatabaseError")
+            } catch let error as DatabaseError {
+                XCTAssertEqual(error.resultCode.rawValue, 123)
+            }
+            assertDidExecute(sql: "CREATE VIRTUAL TABLE \"test\" USING fts3tokenize(simple)")
+            XCTAssertFalse(try db.tableExists("test"))
+        }
+    }
+}