4886049d57
Updates `UserProfileFinder` to have `userProfiles(for:,transaction:)` that looks up many profiles by either UUID or phone number. Refactors the existing `userProfile(for:,transaction:)` to use the batch implementation. Also adds Refinery.refineNonnilKeys to simplify a repeating pattern where we need to fetch values from a database where keys are nonnil. This provides better type safety versus using `refine(condition:,then:,otherwise:)` because the closure can get non-optional keys as input. Adds tests for the new user profile lookup code.
87 lines
3.4 KiB
Swift
87 lines
3.4 KiB
Swift
//
|
|
// Copyright (c) 2022 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
import XCTest
|
|
@testable import SignalServiceKit
|
|
|
|
class UserProfileTest: SignalBaseTest {
|
|
private lazy var localAddress = CommonGenerator.address()
|
|
|
|
override func setUp() {
|
|
super.setUp()
|
|
// Create local account.
|
|
tsAccountManager.registerForTests(withLocalNumber: localAddress.phoneNumber!,
|
|
uuid: localAddress.uuid!)
|
|
}
|
|
|
|
func testUserProfileForUUID() {
|
|
let uuid = UUID()
|
|
let address = SignalServiceAddress(uuid: uuid)
|
|
write { transaction in
|
|
OWSUserProfile(address: address).anyInsert(transaction: transaction)
|
|
}
|
|
read { transaction in
|
|
let actual = OWSUserProfile.getFor(address, transaction: transaction)
|
|
XCTAssertEqual(actual?.recipientUUID, uuid.uuidString)
|
|
}
|
|
read { transaction in
|
|
let actual = OWSUserProfile.getFor(SignalServiceAddress(uuid: UUID()), transaction: transaction)
|
|
XCTAssertNil(actual)
|
|
}
|
|
}
|
|
|
|
func testUserProfilesForUUIDs() {
|
|
let addresses = [SignalServiceAddress(uuid: UUID()),
|
|
SignalServiceAddress(uuid: UUID())]
|
|
let profiles = addresses.map { OWSUserProfile(address: $0) }
|
|
write { transaction in
|
|
for profile in profiles {
|
|
profile.anyInsert(transaction: transaction)
|
|
}
|
|
}
|
|
read { transaction in
|
|
let bogusAddresses = [SignalServiceAddress(uuid: UUID())]
|
|
let actual = SignalServiceKit.OWSUserProfile.getFor(keys: addresses + bogusAddresses,
|
|
transaction: transaction)
|
|
let expected = profiles + [nil]
|
|
XCTAssertEqual(actual.map { $0?.recipientUUID },
|
|
expected.map { $0?.recipientUUID })
|
|
}
|
|
}
|
|
|
|
func testUserProfileForPhoneNumber() {
|
|
let address = SignalServiceAddress(phoneNumber: "+17035550000")
|
|
write { transaction in
|
|
OWSUserProfile(address: address).anyInsert(transaction: transaction)
|
|
}
|
|
read { transaction in
|
|
let actual = OWSUserProfile.getFor(address, transaction: transaction)
|
|
XCTAssertEqual(actual?.recipientPhoneNumber, "+17035550000")
|
|
}
|
|
read { transaction in
|
|
let actual = OWSUserProfile.getFor(SignalServiceAddress(phoneNumber: "+17035550001"), transaction: transaction)
|
|
XCTAssertNil(actual)
|
|
}
|
|
}
|
|
|
|
func testUserProfilesForPhoneNumbers() {
|
|
let addresses = [SignalServiceAddress(phoneNumber: "+17035550000"),
|
|
SignalServiceAddress(phoneNumber: "+17035550001")]
|
|
let profiles = addresses.map { OWSUserProfile(address: $0) }
|
|
write { transaction in
|
|
for profile in profiles {
|
|
profile.anyInsert(transaction: transaction)
|
|
}
|
|
}
|
|
read { transaction in
|
|
let bogusAddresses = [SignalServiceAddress(phoneNumber: "+17035550002")]
|
|
let actual = SignalServiceKit.OWSUserProfile.getFor(keys: addresses + bogusAddresses,
|
|
transaction: transaction)
|
|
let expected = profiles + [nil]
|
|
XCTAssertEqual(actual.map { $0?.recipientPhoneNumber },
|
|
expected.map { $0?.recipientPhoneNumber })
|
|
}
|
|
}
|
|
}
|