49 lines
1.5 KiB
Swift
49 lines
1.5 KiB
Swift
//
|
|
// Copyright 2023 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import XCTest
|
|
|
|
@testable import SignalServiceKit
|
|
|
|
class DBTimeBatchingTest: XCTestCase {
|
|
func testAllElements() {
|
|
let elements = [1, 2, 3, 4]
|
|
var seenElements = [Int]()
|
|
MockDB().enumerateWithTimeBatchedWriteTx(elements) { item, tx in
|
|
seenElements.append(item)
|
|
}
|
|
XCTAssertEqual(seenElements, elements)
|
|
}
|
|
|
|
func testSeparateTransactions() {
|
|
var transactions = [DBWriteTransaction]()
|
|
MockDB().enumerateWithTimeBatchedWriteTx(1...100, yieldTxAfter: -1) { _, tx in
|
|
transactions.append(tx)
|
|
}
|
|
let uniqueTransactions = Set(transactions.map { ObjectIdentifier($0) })
|
|
XCTAssertEqual(uniqueTransactions.count, 100)
|
|
}
|
|
|
|
func testSingleTransaction() {
|
|
var transactions = [DBWriteTransaction]()
|
|
MockDB().enumerateWithTimeBatchedWriteTx(1...100, yieldTxAfter: .infinity) { _, tx in
|
|
transactions.append(tx)
|
|
}
|
|
let uniqueTransactions = Set(transactions.map { ObjectIdentifier($0) })
|
|
XCTAssertEqual(uniqueTransactions.count, 1)
|
|
}
|
|
|
|
func testThrownError() {
|
|
var seenElements = [Int]()
|
|
XCTAssertThrowsError(try MockDB().enumerateWithTimeBatchedWriteTx(1...100, block: { item, tx in
|
|
seenElements.append(item)
|
|
if item == 5 {
|
|
throw OWSGenericError("")
|
|
}
|
|
}))
|
|
XCTAssertEqual(seenElements, [1, 2, 3, 4, 5])
|
|
}
|
|
}
|