zeus
74 lines
2.5 KiB
Swift
74 lines
2.5 KiB
Swift
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"))
|
|
}
|
|
}
|
|
}
|