3d6225060d
* Add ability to compute registration recovery password * Add RegistrationCoordinator * remove repeat nav controller that evan added * Scaffolding for RegistrationCoodinator test * Add first round of RegistrationCoordinator tests, some stuff I needed to get them working, and some fixes to RegistrationCoordinator itself I discovered thanks to the tests * switch to owsFailBeta * Add auth credential candidate flow tests, related fixes and additions to coordinator and friends * Add tests for session based registration path and associated fixes * PR comment nits
38 lines
1.1 KiB
Swift
38 lines
1.1 KiB
Swift
//
|
|
// Copyright 2023 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import Foundation
|
|
|
|
#if TESTABLE_BUILD
|
|
|
|
public class KBSAuthCredentialStorageMock: KBSAuthCredentialStorage {
|
|
|
|
public var currentUsername: String?
|
|
public var dict = [String: KBSAuthCredential]()
|
|
|
|
public init() {}
|
|
|
|
public func storeAuthCredentialForCurrentUsername(_ auth: KBSAuthCredential, _ transaction: DBWriteTransaction) {
|
|
dict[auth.username] = auth
|
|
currentUsername = auth.username
|
|
}
|
|
|
|
public func getAuthCredentials(_ transaction: DBReadTransaction) -> [KBSAuthCredential] {
|
|
return Array(dict.values)
|
|
}
|
|
|
|
public func getAuthCredentialForCurrentUser(_ transaction: DBReadTransaction) -> KBSAuthCredential? {
|
|
guard let currentUsername = currentUsername else {
|
|
return nil
|
|
}
|
|
return dict[currentUsername]
|
|
}
|
|
|
|
public func deleteInvalidCredentials(_ invalidCredentials: [KBSAuthCredential], _ transaction: DBWriteTransaction) {
|
|
invalidCredentials.lazy.map(\.username).forEach { dict[$0] = nil }
|
|
}
|
|
}
|
|
#endif
|