114 lines
4.5 KiB
Swift
114 lines
4.5 KiB
Swift
//
|
|
// Copyright 2023 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import XCTest
|
|
|
|
@testable import SignalServiceKit
|
|
|
|
class UserProfileMergerTest: XCTestCase {
|
|
private var userProfileStore: MockUserProfileStore!
|
|
private var userProfileMerger: UserProfileMerger!
|
|
|
|
override func setUp() {
|
|
super.setUp()
|
|
|
|
userProfileStore = MockUserProfileStore()
|
|
userProfileMerger = UserProfileMerger(
|
|
userProfileStore: userProfileStore,
|
|
setProfileKeyShim: { [userProfileStore] userProfile, profileKey, tx in
|
|
userProfile.setValue(profileKey, forKey: "profileKey")
|
|
userProfileStore!.updateUserProfile(userProfile, tx: tx)
|
|
}
|
|
)
|
|
}
|
|
|
|
func testMergeLocal() {
|
|
let serviceId = ServiceId(uuidString: "00000000-0000-4000-8000-000000000000")!
|
|
let phoneNumber = E164("+16505550100")!
|
|
|
|
let localProfile = buildUserProfile(serviceId: nil, phoneNumber: kLocalProfileInvariantPhoneNumber, profileKey: nil)
|
|
let otherProfile = buildUserProfile(serviceId: ServiceId(UUID()), phoneNumber: nil, profileKey: nil)
|
|
|
|
userProfileStore.userProfiles = [
|
|
localProfile,
|
|
buildUserProfile(serviceId: serviceId, phoneNumber: nil, profileKey: Data(repeating: 1, count: 32)),
|
|
buildUserProfile(serviceId: nil, phoneNumber: phoneNumber.stringValue, profileKey: Data(repeating: 2, count: 32)),
|
|
otherProfile
|
|
]
|
|
|
|
MockDB().write { tx in
|
|
userProfileMerger.didLearnAssociation(
|
|
mergedRecipient: MergedRecipient(
|
|
serviceId: serviceId,
|
|
oldPhoneNumber: nil,
|
|
newPhoneNumber: phoneNumber,
|
|
isLocalRecipient: true,
|
|
signalRecipient: SignalRecipient(serviceId: serviceId, phoneNumber: phoneNumber)
|
|
),
|
|
transaction: tx
|
|
)
|
|
}
|
|
|
|
XCTAssertEqual(userProfileStore.userProfiles, [localProfile, otherProfile])
|
|
}
|
|
|
|
func testMergeOther() {
|
|
let serviceId = ServiceId(uuidString: "00000000-0000-4000-8000-000000000000")!
|
|
let phoneNumber = E164("+16505550100")!
|
|
|
|
let localProfile = buildUserProfile(serviceId: nil, phoneNumber: kLocalProfileInvariantPhoneNumber, profileKey: nil)
|
|
let otherProfile = buildUserProfile(serviceId: ServiceId(UUID()), phoneNumber: phoneNumber.stringValue, profileKey: nil)
|
|
let finalProfile = buildUserProfile(serviceId: serviceId, phoneNumber: nil, profileKey: nil)
|
|
|
|
userProfileStore.userProfiles = [
|
|
localProfile,
|
|
finalProfile,
|
|
buildUserProfile(serviceId: serviceId, phoneNumber: phoneNumber.stringValue, profileKey: Data(repeating: 2, count: 32)),
|
|
buildUserProfile(serviceId: nil, phoneNumber: phoneNumber.stringValue, profileKey: Data(repeating: 3, count: 32)),
|
|
otherProfile
|
|
]
|
|
|
|
MockDB().write { tx in
|
|
userProfileMerger.didLearnAssociation(
|
|
mergedRecipient: MergedRecipient(
|
|
serviceId: serviceId,
|
|
oldPhoneNumber: nil,
|
|
newPhoneNumber: phoneNumber,
|
|
isLocalRecipient: false,
|
|
signalRecipient: SignalRecipient(serviceId: serviceId, phoneNumber: phoneNumber)
|
|
),
|
|
transaction: tx
|
|
)
|
|
}
|
|
|
|
finalProfile.recipientPhoneNumber = phoneNumber.stringValue
|
|
finalProfile.setValue(OWSAES256Key(data: Data(repeating: 2, count: 32))!, forKey: "profileKey")
|
|
otherProfile.recipientPhoneNumber = nil
|
|
XCTAssertEqual(userProfileStore.userProfiles, [localProfile, finalProfile, otherProfile])
|
|
}
|
|
|
|
private func buildUserProfile(serviceId: ServiceId?, phoneNumber: String?, profileKey: Data?) -> OWSUserProfile {
|
|
OWSUserProfile(
|
|
grdbId: 0,
|
|
uniqueId: UUID().uuidString,
|
|
avatarFileName: nil,
|
|
avatarUrlPath: nil,
|
|
bio: nil,
|
|
bioEmoji: nil,
|
|
canReceiveGiftBadges: false,
|
|
familyName: nil,
|
|
isPniCapable: false,
|
|
isStoriesCapable: false,
|
|
lastFetchDate: nil,
|
|
lastMessagingDate: nil,
|
|
profileBadgeInfo: nil,
|
|
profileKey: profileKey.map { OWSAES256Key(data: $0)! },
|
|
profileName: nil,
|
|
recipientPhoneNumber: phoneNumber,
|
|
recipientUUID: serviceId?.uuidValue.uuidString
|
|
)
|
|
}
|
|
}
|