zeus
64 lines
2.0 KiB
Swift
64 lines
2.0 KiB
Swift
import XCTest
|
|
import GRDB
|
|
|
|
private struct TestError : Error { }
|
|
|
|
class PrefixCursorTests: GRDBTestCase {
|
|
|
|
func testPrefixCursorFromCursor() throws {
|
|
do {
|
|
let base = AnyCursor([1, 2, 3, 4, 5])
|
|
let cursor = base.prefix(0)
|
|
XCTAssertTrue(try cursor.next() == nil) // end
|
|
XCTAssertTrue(try cursor.next() == nil) // past the end
|
|
}
|
|
do {
|
|
let base = AnyCursor([1, 2, 3, 4, 5])
|
|
let cursor = base.prefix(2)
|
|
try XCTAssertEqual(cursor.next()!, 1)
|
|
try XCTAssertEqual(cursor.next()!, 2)
|
|
XCTAssertTrue(try cursor.next() == nil) // end
|
|
XCTAssertTrue(try cursor.next() == nil) // past the end
|
|
}
|
|
do {
|
|
let base = AnyCursor([1, 2, 3, 4, 5])
|
|
let cursor = base.prefix(10)
|
|
try XCTAssertEqual(cursor.next()!, 1)
|
|
try XCTAssertEqual(cursor.next()!, 2)
|
|
try XCTAssertEqual(cursor.next()!, 3)
|
|
try XCTAssertEqual(cursor.next()!, 4)
|
|
try XCTAssertEqual(cursor.next()!, 5)
|
|
XCTAssertTrue(try cursor.next() == nil) // end
|
|
XCTAssertTrue(try cursor.next() == nil) // past the end
|
|
}
|
|
}
|
|
|
|
func testPrefixCursorFromThrowingCursor() throws {
|
|
var i = 1
|
|
let base: AnyCursor<Int> = AnyCursor {
|
|
guard i < 3 else { throw TestError() }
|
|
defer { i += 1 }
|
|
return i
|
|
}
|
|
|
|
let cursor = base.prefix(3)
|
|
try XCTAssertEqual(cursor.next()!, 1)
|
|
try XCTAssertEqual(cursor.next()!, 2)
|
|
do {
|
|
_ = try cursor.next()
|
|
XCTFail()
|
|
} catch is TestError {
|
|
} catch {
|
|
XCTFail()
|
|
}
|
|
}
|
|
|
|
func testPrefixCursorChain() throws {
|
|
let base = AnyCursor([1, 2, 3, 4, 5])
|
|
let cursor = base.prefix(2).prefix(1)
|
|
try XCTAssertEqual(cursor.next()!, 1)
|
|
XCTAssertTrue(try cursor.next() == nil) // end
|
|
XCTAssertTrue(try cursor.next() == nil) // past the end
|
|
}
|
|
}
|