add
This commit is contained in:
zeus
@@ -1,332 +0,0 @@
|
||||
import XCTest
|
||||
import GRDB
|
||||
|
||||
class DatabaseCursorTests: GRDBTestCase {
|
||||
|
||||
// TODO: this test should be duplicated for all cursor types
|
||||
func testNextReturnsNilAfterExhaustion() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.inDatabase { db in
|
||||
do {
|
||||
let cursor = try Int.fetchCursor(db, sql: "SELECT 1 WHERE 0")
|
||||
XCTAssert(try cursor.next() == nil) // end
|
||||
XCTAssert(try cursor.next() == nil) // past the end
|
||||
}
|
||||
do {
|
||||
let cursor = try Int.fetchCursor(db, sql: "SELECT 1")
|
||||
XCTAssertEqual(try cursor.next()!, 1)
|
||||
XCTAssert(try cursor.next() == nil) // end
|
||||
XCTAssert(try cursor.next() == nil) // past the end
|
||||
}
|
||||
do {
|
||||
let cursor = try Int.fetchCursor(db, sql: "SELECT 1 UNION ALL SELECT 2")
|
||||
XCTAssertEqual(try cursor.next()!, 1)
|
||||
XCTAssertEqual(try cursor.next()!, 2)
|
||||
XCTAssert(try cursor.next() == nil) // end
|
||||
XCTAssert(try cursor.next() == nil) // past the end
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TODO: this test should be duplicated for all cursor types
|
||||
func testNextFollowedByForEach() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.inDatabase { db in
|
||||
do {
|
||||
let cursor = try Int.fetchCursor(db, sql: "SELECT 1")
|
||||
XCTAssertEqual(try cursor.next()!, 1)
|
||||
try cursor.forEach { _ in
|
||||
XCTFail("Cursor iteration should have ended")
|
||||
}
|
||||
XCTAssert(try cursor.next() == nil) // end
|
||||
}
|
||||
do {
|
||||
let cursor = try Int.fetchCursor(db, sql: "SELECT 1 UNION SELECT 2")
|
||||
XCTAssertEqual(try cursor.next()!, 1)
|
||||
try cursor.forEach { i in
|
||||
XCTAssertEqual(i, 2)
|
||||
}
|
||||
XCTAssert(try cursor.next() == nil) // end
|
||||
}
|
||||
do {
|
||||
let cursor = try Int.fetchCursor(db, sql: "SELECT 1 UNION SELECT 2 UNION SELECT 3")
|
||||
XCTAssertEqual(try cursor.next()!, 1)
|
||||
try XCTAssertEqual(Array(cursor), [2, 3])
|
||||
XCTAssert(try cursor.next() == nil) // end
|
||||
}
|
||||
do {
|
||||
struct Break: Error { }
|
||||
let cursor = try Int.fetchCursor(db, sql: "SELECT 1 UNION SELECT 2 UNION SELECT 3")
|
||||
try? cursor.forEach { i in
|
||||
XCTAssertEqual(i, 1)
|
||||
throw Break()
|
||||
}
|
||||
try? cursor.forEach { i in
|
||||
XCTAssertEqual(i, 2)
|
||||
throw Break()
|
||||
}
|
||||
try? cursor.forEach { i in
|
||||
XCTAssertEqual(i, 3)
|
||||
throw Break()
|
||||
}
|
||||
XCTAssert(try cursor.next() == nil) // end
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TODO: this test should be duplicated for all cursor types
|
||||
func testStepError() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
let customError = NSError(domain: "Custom", code: 0xDEAD)
|
||||
try dbQueue.inDatabase { db in
|
||||
db.add(function: DatabaseFunction("throw", argumentCount: 0, pure: true) { _ in throw customError })
|
||||
let cursor = try Int.fetchCursor(db, sql: "SELECT throw()")
|
||||
do {
|
||||
_ = try cursor.next()
|
||||
XCTFail()
|
||||
} catch let error as DatabaseError {
|
||||
XCTAssertEqual(error.resultCode, .SQLITE_ERROR)
|
||||
XCTAssertEqual(error.message, "\(customError)")
|
||||
XCTAssertEqual(error.sql!, "SELECT throw()")
|
||||
XCTAssertEqual(error.description, "SQLite error 1: \(customError) - while executing `SELECT throw()`")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TODO: this test should be duplicated for all cursor types
|
||||
func testStepDatabaseError() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
let customError = DatabaseError(resultCode: ResultCode(rawValue: 0xDEAD), message: "custom error")
|
||||
try dbQueue.inDatabase { db in
|
||||
db.add(function: DatabaseFunction("throw", argumentCount: 0, pure: true) { _ in throw customError })
|
||||
let cursor = try Int.fetchCursor(db, sql: "SELECT throw()")
|
||||
do {
|
||||
_ = try cursor.next()
|
||||
XCTFail()
|
||||
} catch let error as DatabaseError {
|
||||
XCTAssertEqual(error.resultCode.rawValue, 0xAD)
|
||||
XCTAssertEqual(error.extendedResultCode.rawValue, 0xDEAD)
|
||||
XCTAssertEqual(error.message, "custom error")
|
||||
XCTAssertEqual(error.sql!, "SELECT throw()")
|
||||
XCTAssertEqual(error.description, "SQLite error 173: custom error - while executing `SELECT throw()`")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Regression test for http://github.com/groue/GRDB.swift/issues/583
|
||||
func testIssue583() throws {
|
||||
struct User: Codable, TableRecord, FetchableRecord, MutablePersistableRecord {
|
||||
static let databaseTableName: String = "user"
|
||||
|
||||
var id: Int64?
|
||||
var username: String
|
||||
var isFlagged: Bool
|
||||
|
||||
init(id: Int64? = nil, username: String, isFlagged: Bool = false) {
|
||||
self.id = id
|
||||
self.username = username
|
||||
self.isFlagged = isFlagged
|
||||
}
|
||||
|
||||
mutating func didInsert(_ inserted: InsertionSuccess) {
|
||||
id = inserted.rowID
|
||||
}
|
||||
}
|
||||
|
||||
struct FlagUser: Codable, TableRecord, FetchableRecord, MutablePersistableRecord {
|
||||
static let databaseTableName: String = "flagUser"
|
||||
|
||||
var username: String
|
||||
}
|
||||
|
||||
let queue = try makeDatabaseQueue()
|
||||
try queue.write { database in
|
||||
try database.create(table: User.databaseTableName) { definition in
|
||||
definition.column("id", .integer).primaryKey(autoincrement: true)
|
||||
definition.column("username", .text).notNull()
|
||||
definition.column("isFlagged", .boolean).notNull().defaults(to: false)
|
||||
}
|
||||
|
||||
try database.create(table: FlagUser.databaseTableName) { definition in
|
||||
definition.column("username", .text).notNull()
|
||||
}
|
||||
|
||||
try [Int](0...50).forEach {
|
||||
var user = User(username: "User\($0)")
|
||||
try user.insert(database)
|
||||
}
|
||||
|
||||
try [Int](40...60).forEach {
|
||||
var flag = FlagUser(username: "User\($0)")
|
||||
try flag.insert(database)
|
||||
}
|
||||
}
|
||||
|
||||
let query = "SELECT * FROM flagUser WHERE (SELECT COUNT(id) FROM user WHERE username = flagUser.username AND isFlagged = 1) = 0"
|
||||
try queue.inDatabase { database in
|
||||
let cursor = try FlagUser.fetchCursor(database, sql: query)
|
||||
while let flagged = try cursor.next() {
|
||||
_ = try User.fetchOne(database, sql: "SELECT * FROM user WHERE username = '\(flagged.username)' LIMIT 1") ??
|
||||
User(username: flagged.username)
|
||||
}
|
||||
}
|
||||
|
||||
// For the record, the lines below show how this test used to fail,
|
||||
// with raw C SQLite3 apis. The faulty line is the call to
|
||||
// sqlite3_set_authorizer during the statement iteration.
|
||||
|
||||
// if #available(OSX 10.14, *) {
|
||||
// var connection: SQLiteConnection? = nil
|
||||
// sqlite3_open_v2(":memory:", &connection, SQLITE_OPEN_READWRITE | SQLITE_OPEN_NOMUTEX, nil)
|
||||
// sqlite3_extended_result_codes(connection, 1)
|
||||
//
|
||||
// sqlite3_exec(connection, """
|
||||
// CREATE TABLE user (username TEXT NOT NULL);
|
||||
// CREATE TABLE flagUser (username TEXT NOT NULL);
|
||||
// INSERT INTO flagUser (username) VALUES ('User1');
|
||||
// INSERT INTO flagUser (username) VALUES ('User2');
|
||||
// """, nil, nil, nil)
|
||||
//
|
||||
// var statement: SQLiteStatement? = nil
|
||||
// sqlite3_set_authorizer(connection, { (_, _, _, _, _, _) in SQLITE_OK }, nil)
|
||||
// sqlite3_prepare_v3(connection, """
|
||||
// SELECT * FROM flagUser WHERE (SELECT COUNT(*) FROM user WHERE username = flagUser.username) = 0
|
||||
// """, -1, 0, &statement, nil)
|
||||
// sqlite3_set_authorizer(connection, nil, nil)
|
||||
// while true {
|
||||
// let code = sqlite3_step(statement)
|
||||
// if code == SQLITE_DONE {
|
||||
// break
|
||||
// } else if code == SQLITE_ROW {
|
||||
// // part of the compilation of another statement, here
|
||||
// // reduced to the strict minimum that reproduces
|
||||
// // the error.
|
||||
// sqlite3_set_authorizer(connection, nil, nil)
|
||||
// } else {
|
||||
// print(String(cString: sqlite3_errmsg(connection)))
|
||||
// XCTFail("Error \(code)")
|
||||
// break
|
||||
// }
|
||||
// }
|
||||
// sqlite3_finalize(statement)
|
||||
// sqlite3_close_v2(connection)
|
||||
// }
|
||||
}
|
||||
|
||||
// For profiling tests
|
||||
let profilingSQL = """
|
||||
WITH RECURSIVE
|
||||
cnt(x) AS (VALUES(1) UNION ALL SELECT x+1 FROM cnt WHERE x<10000000)
|
||||
SELECT x FROM cnt
|
||||
"""
|
||||
|
||||
// Profiling test
|
||||
func testRowCursorStep() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.read { db in
|
||||
let cursor: RowCursor = try Row.fetchCursor(db, sql: profilingSQL)
|
||||
while let _ = try cursor.next() { }
|
||||
}
|
||||
}
|
||||
|
||||
// Profiling test
|
||||
func testRowCursorForEach() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.read { db in
|
||||
let cursor: RowCursor = try Row.fetchCursor(db, sql: profilingSQL)
|
||||
try cursor.forEach { _ in }
|
||||
}
|
||||
}
|
||||
|
||||
// Profiling test
|
||||
func testDatabaseValueCursorStep() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.read { db in
|
||||
let cursor: DatabaseValueCursor<Int> = try Int.fetchCursor(db, sql: profilingSQL)
|
||||
while let _ = try cursor.next() { }
|
||||
}
|
||||
}
|
||||
|
||||
// Profiling test
|
||||
func testDatabaseValueCursorForEach() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.read { db in
|
||||
let cursor: DatabaseValueCursor<Int> = try Int.fetchCursor(db, sql: profilingSQL)
|
||||
try cursor.forEach { _ in }
|
||||
}
|
||||
}
|
||||
|
||||
// Profiling test
|
||||
func testNullableDatabaseValueCursorStep() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.read { db in
|
||||
let cursor: DatabaseValueCursor<Int?> = try Optional<Int>.fetchCursor(db, sql: profilingSQL)
|
||||
while let _ = try cursor.next() { }
|
||||
}
|
||||
}
|
||||
|
||||
// Profiling test
|
||||
func testNullableDatabaseValueCursorForEach() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.read { db in
|
||||
let cursor: DatabaseValueCursor<Int?> = try Optional<Int>.fetchCursor(db, sql: profilingSQL)
|
||||
try cursor.forEach { _ in }
|
||||
}
|
||||
}
|
||||
|
||||
// Profiling test
|
||||
func testFastDatabaseValueCursorStep() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.read { db in
|
||||
let cursor: FastDatabaseValueCursor<Int> = try Int.fetchCursor(db, sql: profilingSQL)
|
||||
while let _ = try cursor.next() { }
|
||||
}
|
||||
}
|
||||
|
||||
// Profiling test
|
||||
func testFastDatabaseValueCursorForEach() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.read { db in
|
||||
let cursor: FastDatabaseValueCursor<Int> = try Int.fetchCursor(db, sql: profilingSQL)
|
||||
try cursor.forEach { _ in }
|
||||
}
|
||||
}
|
||||
|
||||
// Profiling test
|
||||
func testFastNullableDatabaseValueCursorStep() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.read { db in
|
||||
let cursor: FastDatabaseValueCursor<Int?> = try Optional<Int>.fetchCursor(db, sql: profilingSQL)
|
||||
while let _ = try cursor.next() { }
|
||||
}
|
||||
}
|
||||
|
||||
// Profiling test
|
||||
func testFastNullableDatabaseValueCursorForEach() throws {
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.read { db in
|
||||
let cursor: FastDatabaseValueCursor<Int?> = try Optional<Int>.fetchCursor(db, sql: profilingSQL)
|
||||
try cursor.forEach { _ in }
|
||||
}
|
||||
}
|
||||
|
||||
// Profiling test
|
||||
func testRecordCursorStep() throws {
|
||||
struct S: FetchableRecord { init(row: Row) { } }
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.read { db in
|
||||
let cursor: RecordCursor<S> = try S.fetchCursor(db, sql: profilingSQL)
|
||||
while let _ = try cursor.next() { }
|
||||
}
|
||||
}
|
||||
|
||||
// Profiling test
|
||||
func testRecordCursorForEach() throws {
|
||||
struct S: FetchableRecord { init(row: Row) { } }
|
||||
let dbQueue = try makeDatabaseQueue()
|
||||
try dbQueue.read { db in
|
||||
let cursor: RecordCursor<S> = try S.fetchCursor(db, sql: profilingSQL)
|
||||
try cursor.forEach { _ in }
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user