Signal-iOS/Signal/test/PerformanceTests/PerformanceBaseTest.swift
2021-03-25 09:24:27 -03:00

42 lines
1022 B
Swift

//
// Copyright (c) 2021 Open Whisper Systems. All rights reserved.
//
import Foundation
import XCTest
import SignalServiceKit
public class PerformanceBaseTest: XCTestCase {
// MARK: Hooks
public override func setUp() {
super.setUp()
ClearCurrentAppContextForTests()
SetCurrentAppContext(TestAppContext())
SDSDatabaseStorage.shouldLogDBQueries = false
MockSSKEnvironment.activate()
}
public override func tearDown() {
SDSDatabaseStorage.shouldLogDBQueries = DebugFlags.logSQLQueries
super.tearDown()
}
// MARK: Helpers
public func read(_ block: @escaping (SDSAnyReadTransaction) -> Void) {
return databaseStorage.read(block: block)
}
public func write(_ block: @escaping (SDSAnyWriteTransaction) -> Void) {
return databaseStorage.write(block: block)
}
public func asyncWrite(_ block: @escaping (SDSAnyWriteTransaction) -> Void) {
return databaseStorage.asyncWrite(block: block)
}
}