Signal-iOS/Signal/test/PerformanceTests/PerformanceBaseTest.swift
2020-08-13 08:59:56 -03:00

52 lines
1.2 KiB
Swift

//
// Copyright (c) 2020 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: - Dependencies
public var databaseStorage: SDSDatabaseStorage {
return SDSDatabaseStorage.shared
}
var storageCoordinator: StorageCoordinator {
return SSKEnvironment.shared.storageCoordinator
}
// 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)
}
}