Signal-iOS/SignalServiceKit/tests/Storage/SDSDatabaseStorageTest.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

187 lines
7.5 KiB
Swift

//
// Copyright 2019 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
//
import Foundation
import XCTest
@testable import SignalServiceKit
extension TSThread {
@objc
public class func anyFetchAll(databaseStorage: SDSDatabaseStorage) -> [TSThread] {
var result = [TSThread]()
databaseStorage.read { transaction in
result += anyFetchAll(transaction: transaction)
}
return result
}
}
// MARK: -
extension TSInteraction {
@objc
public class func anyFetchAll(databaseStorage: SDSDatabaseStorage) -> [TSInteraction] {
var result = [TSInteraction]()
databaseStorage.read { transaction in
result += anyFetchAll(transaction: transaction)
}
return result
}
}
// MARK: -
class SDSDatabaseStorageTest: SSKBaseTestSwift {
// MARK: - Test Life Cycle
override func setUp() {
super.setUp()
// ensure local client has necessary "registered" state
let localE164Identifier = "+13235551234"
let localUUID = UUID()
tsAccountManager.registerForTests(withLocalNumber: localE164Identifier, uuid: localUUID)
}
// MARK: -
func test_threads() {
let storage = SDSDatabaseStorage.shared
XCTAssertEqual(0, TSThread.anyFetchAll(databaseStorage: storage).count)
let contactAddress = SignalServiceAddress(phoneNumber: "+13213214321")
let contactThread = TSContactThread(contactAddress: contactAddress)
storage.write { transaction in
XCTAssertEqual(0, TSThread.anyFetchAll(transaction: transaction).count)
contactThread.anyInsert(transaction: transaction)
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
}
XCTAssertEqual(1, TSThread.anyFetchAll(databaseStorage: storage).count)
var groupThread: TSGroupThread!
storage.write { transaction in
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
groupThread = try! GroupManager.createGroupForTests(members: [contactAddress],
name: "Test Group",
transaction: transaction)
XCTAssertEqual(2, TSThread.anyFetchAll(transaction: transaction).count)
}
XCTAssertEqual(2, TSThread.anyFetchAll(databaseStorage: storage).count)
storage.write { transaction in
XCTAssertEqual(2, TSThread.anyFetchAll(transaction: transaction).count)
contactThread.anyRemove(transaction: transaction)
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
}
XCTAssertEqual(1, TSThread.anyFetchAll(databaseStorage: storage).count)
// Update
storage.write { transaction in
let threads = TSThread.anyFetchAll(transaction: transaction)
guard let firstThread = threads.first else {
XCTFail("Missing model.")
return
}
XCTAssertNil(firstThread.messageDraft)
firstThread.update(withDraft: MessageBody(text: "Some draft",
ranges: .empty),
replyInfo: nil,
transaction: transaction)
}
storage.read { transaction in
let threads = TSThread.anyFetchAll(transaction: transaction)
guard let firstThread = threads.first else {
XCTFail("Missing model.")
return
}
XCTAssertEqual(firstThread.messageDraft, "Some draft")
}
XCTAssertEqual(1, TSThread.anyFetchAll(databaseStorage: storage).count)
storage.write { transaction in
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
groupThread.anyRemove(transaction: transaction)
XCTAssertEqual(0, TSThread.anyFetchAll(transaction: transaction).count)
}
XCTAssertEqual(0, TSThread.anyFetchAll(databaseStorage: storage).count)
}
func test_interactions() {
let storage = SDSDatabaseStorage.shared
XCTAssertEqual(0, TSInteraction.anyFetchAll(databaseStorage: storage).count)
let contactAddress = SignalServiceAddress(phoneNumber: "+13213214321")
let contactThread = TSContactThread(contactAddress: contactAddress)
storage.write { transaction in
XCTAssertEqual(0, TSThread.anyFetchAll(transaction: transaction).count)
contactThread.anyInsert(transaction: transaction)
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
}
XCTAssertEqual(1, TSThread.anyFetchAll(databaseStorage: storage).count)
XCTAssertEqual(0, TSInteraction.anyFetchAll(databaseStorage: storage).count)
let message1 = TSOutgoingMessage(in: contactThread, messageBody: "message1", attachmentId: nil)
storage.write { transaction in
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
XCTAssertEqual(0, TSInteraction.anyFetchAll(transaction: transaction).count)
message1.anyInsert(transaction: transaction)
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
XCTAssertEqual(1, TSInteraction.anyFetchAll(transaction: transaction).count)
}
XCTAssertEqual(1, TSThread.anyFetchAll(databaseStorage: storage).count)
XCTAssertEqual(1, TSInteraction.anyFetchAll(databaseStorage: storage).count)
let message2 = TSOutgoingMessage(in: contactThread, messageBody: "message2", attachmentId: nil)
storage.write { transaction in
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
XCTAssertEqual(1, TSInteraction.anyFetchAll(transaction: transaction).count)
message2.anyInsert(transaction: transaction)
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
XCTAssertEqual(2, TSInteraction.anyFetchAll(transaction: transaction).count)
}
XCTAssertEqual(1, TSThread.anyFetchAll(databaseStorage: storage).count)
XCTAssertEqual(2, TSInteraction.anyFetchAll(databaseStorage: storage).count)
storage.write { transaction in
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
XCTAssertEqual(2, TSInteraction.anyFetchAll(transaction: transaction).count)
message1.anyRemove(transaction: transaction)
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
XCTAssertEqual(1, TSInteraction.anyFetchAll(transaction: transaction).count)
}
XCTAssertEqual(1, TSThread.anyFetchAll(databaseStorage: storage).count)
XCTAssertEqual(1, TSInteraction.anyFetchAll(databaseStorage: storage).count)
storage.write { transaction in
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
XCTAssertEqual(1, TSInteraction.anyFetchAll(transaction: transaction).count)
message2.anyRemove(transaction: transaction)
XCTAssertEqual(1, TSThread.anyFetchAll(transaction: transaction).count)
XCTAssertEqual(0, TSInteraction.anyFetchAll(transaction: transaction).count)
}
XCTAssertEqual(1, TSThread.anyFetchAll(databaseStorage: storage).count)
XCTAssertEqual(0, TSInteraction.anyFetchAll(databaseStorage: storage).count)
}
}