Signal-iOS/SignalServiceKit/tests/Storage/OWSIdentityManagerTests.swift
Evan Hahn 370ff654e7
Change license to AGPL
Change license to AGPL

This commit:

- Updates the `LICENSE` file

- Start every file with something like:

      // Copyright YEAR_FIRST_PUBLISHED Signal Messenger, LLC
      // SPDX-License-Identifier: AGPL-3.0-only

---

First, I removed existing license headers with this Ruby 3.1.2 script:

    require 'set'

    EXTENSIONS_TO_CHECK = Set['.h', '.hpp', '.cpp', '.m', '.mm', '.pch', '.swift']

    same = 0
    different = 0

    all_files = `git ls-files`.lines.map { |line| line.strip }
    all_files.each do |relative_path|
      if relative_path == 'Pods'
        next
      end

      unless EXTENSIONS_TO_CHECK.include? File.extname(relative_path)
        next
      end

      path = File.expand_path(relative_path)

      contents = File.read(path)
      new_contents = contents.sub(/\/\/\n\/\/  Copyright .*\n\/\/\n\n/, '')

      if contents == new_contents
        same += 1
      else
        different += 1
      end

      File.write(path, new_contents)
    end

    puts "updated #{different} file(s), left #{same} untouched"

I'm sure this script could be improved, but it worked well enough.

Then, I created `Scripts/lint/lint-license-headers` and ran it to auto-
fix a lot of files. This changed the mode of some files, but I think
that's actually desirable. For example,
`SignalServiceKit/src/Util/AppContext.m` previously had a mode of
`0755/-rwxr-xr-x`, and it's now `0644/-rw-r--r--`.

Then I fixed some stragglers and updated the precommit script.

See [a similar change in the Desktop app][0].

[0]: 8bfaf598af
2022-10-13 08:25:37 -05:00

133 lines
7.1 KiB
Swift

//
// Copyright 2022 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
//
@testable import SignalServiceKit
import XCTest
class OWSIdentityManagerTests: SSKBaseTestSwift {
override func setUp() {
super.setUp()
tsAccountManager.registerForTests(withLocalNumber: "+13235551234", uuid: UUID())
}
func testNewEmptyKey() {
let newKey = Randomness.generateRandomBytes(32)
let address = SignalServiceAddress(phoneNumber: "+12223334444")
write { transaction in
_ = OWSAccountIdFinder.ensureAccountId(forAddress: address, transaction: transaction)
XCTAssert(identityManager.isTrustedIdentityKey(newKey,
address: address,
direction: .outgoing,
transaction: transaction))
XCTAssert(identityManager.isTrustedIdentityKey(newKey,
address: address,
direction: .incoming,
transaction: transaction))
}
}
func testAlreadyRegisteredKey() {
let newKey = Randomness.generateRandomBytes(32)
let address = SignalServiceAddress(phoneNumber: "+12223334444")
write { transaction in
identityManager.saveRemoteIdentity(newKey, address: address, transaction: transaction)
XCTAssert(identityManager.isTrustedIdentityKey(newKey,
address: address,
direction: .outgoing,
transaction: transaction))
XCTAssert(identityManager.isTrustedIdentityKey(newKey,
address: address,
direction: .incoming,
transaction: transaction))
}
}
func testChangedKey() {
let originalKey = Randomness.generateRandomBytes(32)
let address = SignalServiceAddress(phoneNumber: "+12223334444")
write { transaction in
identityManager.saveRemoteIdentity(originalKey, address: address, transaction: transaction)
XCTAssert(identityManager.isTrustedIdentityKey(originalKey,
address: address,
direction: .outgoing,
transaction: transaction))
XCTAssert(identityManager.isTrustedIdentityKey(originalKey,
address: address,
direction: .incoming,
transaction: transaction))
let otherKey = Randomness.generateRandomBytes(32)
XCTAssertFalse(identityManager.isTrustedIdentityKey(otherKey,
address: address,
direction: .outgoing,
transaction: transaction))
XCTAssert(identityManager.isTrustedIdentityKey(otherKey,
address: address,
direction: .incoming,
transaction: transaction))
}
}
func testIdentityKey() {
let newKey = identityManager.generateNewIdentityKey(for: .aci)
XCTAssertEqual(newKey.publicKey.count, 32)
let pniKey = identityManager.generateNewIdentityKey(for: .pni)
XCTAssertEqual(pniKey.publicKey.count, 32)
XCTAssertNotEqual(pniKey.privateKey, newKey.privateKey)
let fetchedKey = identityManager.identityKeyPair(for: .aci)!
XCTAssertEqual(newKey.privateKey, fetchedKey.privateKey)
let fetchedPniKey = identityManager.identityKeyPair(for: .pni)!
XCTAssertEqual(pniKey.privateKey, fetchedPniKey.privateKey)
}
func testShouldSharePhoneNumber() {
let aliceAddress = SignalServiceAddress(uuid: UUID(), phoneNumber: "+12223334444")
let bobAddress = SignalServiceAddress(uuid: UUID(), phoneNumber: "+17775556666")
write { transaction in
// {}
XCTAssertFalse(identityManager.shouldSharePhoneNumber(with: aliceAddress, transaction: transaction))
XCTAssertFalse(identityManager.shouldSharePhoneNumber(with: bobAddress, transaction: transaction))
// {Alice}
identityManager.setShouldSharePhoneNumber(with: aliceAddress, transaction: transaction)
XCTAssertTrue(identityManager.shouldSharePhoneNumber(with: aliceAddress, transaction: transaction))
XCTAssertFalse(identityManager.shouldSharePhoneNumber(with: bobAddress, transaction: transaction))
// {Alice}; redundant set shouldn't change anything.
identityManager.setShouldSharePhoneNumber(with: aliceAddress, transaction: transaction)
XCTAssertTrue(identityManager.shouldSharePhoneNumber(with: aliceAddress, transaction: transaction))
XCTAssertFalse(identityManager.shouldSharePhoneNumber(with: bobAddress, transaction: transaction))
// {Alice, Bob}
identityManager.setShouldSharePhoneNumber(with: bobAddress, transaction: transaction)
XCTAssertTrue(identityManager.shouldSharePhoneNumber(with: aliceAddress, transaction: transaction))
XCTAssertTrue(identityManager.shouldSharePhoneNumber(with: bobAddress, transaction: transaction))
// {Bob}
identityManager.clearShouldSharePhoneNumber(with: aliceAddress, transaction: transaction)
XCTAssertFalse(identityManager.shouldSharePhoneNumber(with: aliceAddress, transaction: transaction))
XCTAssertTrue(identityManager.shouldSharePhoneNumber(with: bobAddress, transaction: transaction))
// {Bob}; redundant clear shouldn't change anything.
identityManager.clearShouldSharePhoneNumber(with: aliceAddress, transaction: transaction)
XCTAssertFalse(identityManager.shouldSharePhoneNumber(with: aliceAddress, transaction: transaction))
XCTAssertTrue(identityManager.shouldSharePhoneNumber(with: bobAddress, transaction: transaction))
// {Alice, Bob}
identityManager.setShouldSharePhoneNumber(with: aliceAddress, transaction: transaction)
// {}
identityManager.clearShouldSharePhoneNumberForEveryone(transaction: transaction)
XCTAssertFalse(identityManager.shouldSharePhoneNumber(with: aliceAddress, transaction: transaction))
XCTAssertFalse(identityManager.shouldSharePhoneNumber(with: bobAddress, transaction: transaction))
}
}
}