89 lines
3.2 KiB
Swift
89 lines
3.2 KiB
Swift
//
|
|
// Copyright 2020 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import XCTest
|
|
@testable import SignalServiceKit
|
|
|
|
final class UUIDBackfillTaskTest: SSKBaseTestSwift {
|
|
private class MockContactDiscoveryManager: ContactDiscoveryManager {
|
|
private(set) var requests = [Set<String>]()
|
|
|
|
func lookUp(phoneNumbers: Set<String>, mode: ContactDiscoveryMode) -> Promise<Set<SignalRecipient>> {
|
|
requests.append(phoneNumbers)
|
|
|
|
switch requests.count {
|
|
case 1:
|
|
return Promise(error: ContactDiscoveryError(
|
|
kind: .genericServerError, debugDescription: "", retryable: true, retryAfterDate: nil
|
|
))
|
|
case 2:
|
|
return Promise(error: ContactDiscoveryError(
|
|
kind: .rateLimit,
|
|
debugDescription: "",
|
|
retryable: true,
|
|
retryAfterDate: Date(timeIntervalSinceNow: -60)
|
|
))
|
|
case 3:
|
|
return Promise(error: ContactDiscoveryError(
|
|
kind: .rateLimit,
|
|
debugDescription: "",
|
|
retryable: true,
|
|
retryAfterDate: Date(timeIntervalSinceNow: 0.001)
|
|
))
|
|
default:
|
|
return .value([])
|
|
}
|
|
}
|
|
}
|
|
|
|
private lazy var localIdentifiers = LocalIdentifiers(
|
|
aci: ServiceId(UUID()),
|
|
pni: ServiceId(UUID()),
|
|
phoneNumber: "+16505550199"
|
|
)
|
|
|
|
override func setUp() {
|
|
super.setUp()
|
|
tsAccountManager.registerForTests(withLocalNumber: localIdentifiers.phoneNumber, uuid: localIdentifiers.aci.uuidValue)
|
|
}
|
|
|
|
func testRetryOnError() throws {
|
|
// Add a few phone number-only recipients.
|
|
let phoneNumbers = [
|
|
E164("+16505550100")!,
|
|
E164("+16505550101")!
|
|
]
|
|
databaseStorage.write { tx in
|
|
let recipientFetcher = DependenciesBridge.shared.recipientFetcher
|
|
for phoneNumber in phoneNumbers {
|
|
recipientFetcher.fetchOrCreate(phoneNumber: phoneNumber, tx: tx.asV2Write).markAsRegisteredAndSave(tx: tx)
|
|
}
|
|
let recipientMerger = DependenciesBridge.shared.recipientMerger
|
|
let mergedRecipient = recipientMerger.applyMergeFromLinkedDevice(
|
|
localIdentifiers: localIdentifiers,
|
|
serviceId: ServiceId(UUID()),
|
|
phoneNumber: E164("+16505550102")!,
|
|
tx: tx.asV2Write
|
|
)
|
|
mergedRecipient.markAsRegisteredAndSave(tx: tx)
|
|
}
|
|
|
|
let manager = MockContactDiscoveryManager()
|
|
let task = UUIDBackfillTask(contactDiscoveryManager: manager, databaseStorage: databaseStorage)
|
|
|
|
let completed = expectation(description: "Waiting for task.")
|
|
task.perform().done {
|
|
completed.fulfill()
|
|
}
|
|
waitForExpectations(timeout: 10)
|
|
|
|
// We expect four requests.
|
|
XCTAssertEqual(manager.requests.count, 4)
|
|
// Which should match the addresses missing UUIDs.
|
|
let phoneNumbersMissingUUIDs = phoneNumbers.map { $0.stringValue }
|
|
XCTAssertEqual(Set(manager.requests), [Set(phoneNumbersMissingUUIDs)])
|
|
}
|
|
}
|