143 lines
6.4 KiB
Swift
143 lines
6.4 KiB
Swift
//
|
|
// Copyright (c) 2022 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
import XCTest
|
|
import GRDB
|
|
|
|
@testable import SignalServiceKit
|
|
|
|
class OWSRecipientIdentityTest: SSKBaseTestSwift {
|
|
private lazy var localAddress = CommonGenerator.address()
|
|
private lazy var aliceAddress = CommonGenerator.address()
|
|
private lazy var bobAddress = CommonGenerator.address()
|
|
private lazy var charlieAddress = CommonGenerator.address()
|
|
private var recipients: [SignalServiceAddress] {
|
|
[aliceAddress, bobAddress, charlieAddress, localAddress]
|
|
}
|
|
private var groupThread: TSGroupThread!
|
|
private var identityKeys = [SignalServiceAddress: Data]()
|
|
|
|
private func identityKey(_ address: SignalServiceAddress) -> Data {
|
|
if let value = identityKeys[address] {
|
|
return value
|
|
}
|
|
let data = Randomness.generateRandomBytes(Int32(kStoredIdentityKeyLength))
|
|
identityKeys[address] = data
|
|
return data
|
|
}
|
|
|
|
private func createFakeGroup() throws {
|
|
// Create local account.
|
|
tsAccountManager.registerForTests(withLocalNumber: localAddress.phoneNumber!,
|
|
uuid: localAddress.uuid!)
|
|
// Create recipients.
|
|
write { transaction in
|
|
for recipient in self.recipients {
|
|
SignalRecipient.mark(asRegisteredAndGet: recipient,
|
|
trustLevel: .high,
|
|
transaction: transaction)
|
|
}
|
|
}
|
|
// Create identities for our recipients.
|
|
for recipient in recipients {
|
|
OWSIdentityManager.shared.saveRemoteIdentity(identityKey(recipient),
|
|
address: recipient)
|
|
}
|
|
|
|
// Create a group with our recipients plus us.
|
|
self.groupThread = try! GroupManager.createGroupForTests(members: recipients,
|
|
name: "Test Group")
|
|
}
|
|
|
|
override func setUp() {
|
|
super.setUp()
|
|
try! createFakeGroup()
|
|
}
|
|
|
|
func testNoneVerified() throws {
|
|
XCTAssertTrue(Self.identityManager.groupContainsUnverifiedMember(groupThread.uniqueId))
|
|
}
|
|
|
|
func testAllVerified() throws {
|
|
for recipient in recipients {
|
|
OWSIdentityManager.shared.setVerificationState(.verified,
|
|
identityKey: identityKey(recipient),
|
|
address: recipient,
|
|
isUserInitiatedChange: true)
|
|
}
|
|
XCTAssertFalse(Self.identityManager.groupContainsUnverifiedMember(groupThread.uniqueId))
|
|
}
|
|
|
|
func testSomeVerified() throws {
|
|
let recipient = recipients[0]
|
|
OWSIdentityManager.shared.setVerificationState(.verified,
|
|
identityKey: identityKey(recipient),
|
|
address: recipient,
|
|
isUserInitiatedChange: true)
|
|
XCTAssertTrue(Self.identityManager.groupContainsUnverifiedMember(groupThread.uniqueId))
|
|
}
|
|
|
|
func testSomeNoLongerVerified() throws {
|
|
// Verify everyone
|
|
for recipient in recipients {
|
|
OWSIdentityManager.shared.setVerificationState(.verified,
|
|
identityKey: identityKey(recipient),
|
|
address: recipient,
|
|
isUserInitiatedChange: true)
|
|
}
|
|
// Make Alice and Bob no-longer-verified.
|
|
let deverifiedAddresses = [aliceAddress, bobAddress]
|
|
for recipient in deverifiedAddresses {
|
|
OWSIdentityManager.shared.setVerificationState(.noLongerVerified,
|
|
identityKey: identityKey(recipient),
|
|
address: recipient,
|
|
isUserInitiatedChange: false)
|
|
}
|
|
XCTAssertTrue(Self.identityManager.groupContainsUnverifiedMember(groupThread.uniqueId))
|
|
|
|
// Check that the list of no-longer-verified addresses is just Alice and Bob.
|
|
read { transaction in
|
|
let noLongerVerifiedAddresses =
|
|
OWSRecipientIdentity.noLongerVerifiedAddresses(inGroup: self.groupThread.uniqueId,
|
|
limit: 2,
|
|
transaction: transaction)
|
|
XCTAssertEqual(Set(noLongerVerifiedAddresses), Set(deverifiedAddresses))
|
|
}
|
|
}
|
|
|
|
func testNoLongerVerifiedLimit() throws {
|
|
for recipient in recipients {
|
|
OWSIdentityManager.shared.setVerificationState(.noLongerVerified,
|
|
identityKey: identityKey(recipient),
|
|
address: recipient,
|
|
isUserInitiatedChange: false)
|
|
}
|
|
// All recipients are no longer verified. Check that the limit is respected.
|
|
for limit in 1..<recipients.count {
|
|
read { transaction in
|
|
let noLongerVerifiedAddresses =
|
|
OWSRecipientIdentity.noLongerVerifiedAddresses(inGroup: self.groupThread.uniqueId,
|
|
limit: limit,
|
|
transaction: transaction)
|
|
XCTAssertEqual(noLongerVerifiedAddresses.count, limit)
|
|
}
|
|
}
|
|
}
|
|
|
|
func testLocalAddressIgnoredForVerifiedCheck() {
|
|
// Verify everyone except me.
|
|
for recipient in recipients {
|
|
if recipient == localAddress {
|
|
continue
|
|
}
|
|
OWSIdentityManager.shared.setVerificationState(.verified,
|
|
identityKey: identityKey(recipient),
|
|
address: recipient,
|
|
isUserInitiatedChange: true)
|
|
}
|
|
XCTAssertFalse(Self.identityManager.groupContainsUnverifiedMember(groupThread.uniqueId))
|
|
}
|
|
}
|