61 lines
1.7 KiB
Swift
61 lines
1.7 KiB
Swift
//
|
|
// Copyright 2023 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
@testable import SignalServiceKit
|
|
import XCTest
|
|
|
|
class OWSDeviceManagerTest: XCTestCase {
|
|
private let db: DB = MockDB()
|
|
private var deviceManager: OWSDeviceManager!
|
|
|
|
override func setUp() {
|
|
deviceManager = OWSDeviceManagerImpl(
|
|
keyValueStoreFactory: InMemoryKeyValueStoreFactory()
|
|
)
|
|
}
|
|
|
|
func testHasReceivedSyncMessage() {
|
|
db.read { tx in
|
|
XCTAssertFalse(deviceManager.hasReceivedSyncMessage(
|
|
inLastSeconds: 60,
|
|
transaction: tx
|
|
))
|
|
}
|
|
|
|
db.write { transaction in
|
|
deviceManager.setHasReceivedSyncMessage(
|
|
lastReceivedAt: Date().addingTimeInterval(-5),
|
|
transaction: transaction
|
|
)
|
|
}
|
|
|
|
db.read { tx in
|
|
XCTAssertFalse(deviceManager.hasReceivedSyncMessage(
|
|
inLastSeconds: 4,
|
|
transaction: tx
|
|
))
|
|
}
|
|
|
|
db.read { tx in
|
|
XCTAssertTrue(deviceManager.hasReceivedSyncMessage(
|
|
inLastSeconds: 6,
|
|
transaction: tx
|
|
))
|
|
}
|
|
}
|
|
|
|
func testMayHaveLinkedDevices() {
|
|
db.write { transaction in
|
|
XCTAssertTrue(deviceManager.mightHaveUnknownLinkedDevice(transaction: transaction))
|
|
|
|
deviceManager.setMightHaveUnknownLinkedDevice(false, transaction: transaction)
|
|
XCTAssertFalse(deviceManager.mightHaveUnknownLinkedDevice(transaction: transaction))
|
|
|
|
deviceManager.setMightHaveUnknownLinkedDevice(true, transaction: transaction)
|
|
XCTAssertTrue(deviceManager.mightHaveUnknownLinkedDevice(transaction: transaction))
|
|
}
|
|
}
|
|
}
|