75f179f572
There are a lot of events that get dispatched onto the main queue (often /from/ the main queue), and processing them during the setup for the subsequent test can cause problems. In particular, the notification for setting up a local number in -[OWSMessageManagerTest test_GroupUpdate] was occasionally being processed in the subsequent -test_GroupUpdateWithAvatar, depending on how long GRDB took to set up in the second test.
94 lines
2.4 KiB
Swift
94 lines
2.4 KiB
Swift
//
|
|
// Copyright (c) 2021 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
import XCTest
|
|
@testable import SignalServiceKit
|
|
import CocoaLumberjack
|
|
|
|
@objc
|
|
public class SSKBaseTestSwift: XCTestCase {
|
|
|
|
// MARK: - Dependencies
|
|
|
|
var primaryStorage: OWSPrimaryStorage? {
|
|
return SSKEnvironment.shared.primaryStorage
|
|
}
|
|
|
|
// MARK: -
|
|
|
|
@objc
|
|
public override func setUp() {
|
|
super.setUp()
|
|
|
|
DDLog.add(DDTTYLogger.sharedInstance!)
|
|
|
|
ClearCurrentAppContextForTests()
|
|
SetCurrentAppContext(TestAppContext())
|
|
|
|
MockSSKEnvironment.activate()
|
|
|
|
GroupManager.forceV1Groups()
|
|
}
|
|
|
|
@objc
|
|
public override func tearDown() {
|
|
AssertIsOnMainThread()
|
|
|
|
// Spin the main run loop to flush any remaining async work.
|
|
var done = false
|
|
DispatchQueue.main.async { done = true }
|
|
while !done {
|
|
CFRunLoopRunInMode(.defaultMode, 0.0, true)
|
|
}
|
|
|
|
super.tearDown()
|
|
}
|
|
|
|
@objc
|
|
public var databaseStorage: SDSDatabaseStorage {
|
|
return SDSDatabaseStorage.shared
|
|
}
|
|
|
|
@objc
|
|
public func read(_ block: @escaping (SDSAnyReadTransaction) -> Void) {
|
|
return databaseStorage.read(block: block)
|
|
}
|
|
|
|
public func write<T>(_ block: @escaping (SDSAnyWriteTransaction) -> T) -> T {
|
|
return databaseStorage.write(block: block)
|
|
}
|
|
|
|
@objc
|
|
public func asyncWrite(_ block: @escaping (SDSAnyWriteTransaction) -> Void) {
|
|
return databaseStorage.asyncWrite(block: block)
|
|
}
|
|
|
|
@objc
|
|
public func yapRead(_ block: @escaping (YapDatabaseReadTransaction) -> Void) {
|
|
guard let primaryStorage = primaryStorage else {
|
|
XCTFail("Missing primaryStorage.")
|
|
return
|
|
}
|
|
return primaryStorage.dbReadConnection.read(block)
|
|
}
|
|
|
|
@objc
|
|
public func yapWrite(_ block: @escaping (YapDatabaseReadWriteTransaction) -> Void) {
|
|
guard let primaryStorage = primaryStorage else {
|
|
XCTFail("Missing primaryStorage.")
|
|
return
|
|
}
|
|
return primaryStorage.dbReadWriteConnection.readWrite(block)
|
|
}
|
|
|
|
@objc
|
|
public func yapAsyncWrite(_ block: @escaping (YapDatabaseReadWriteTransaction) -> Void) {
|
|
guard let primaryStorage = primaryStorage else {
|
|
XCTFail("Missing primaryStorage.")
|
|
return
|
|
}
|
|
return primaryStorage.dbReadWriteConnection.asyncReadWrite(block)
|
|
}
|
|
}
|