mirror of
https://github.com/permissionlesstech/bitchat.git
synced 2026-07-25 11:05:19 +00:00
* Extract BLE service policy helpers * Stabilize image media transfer test --------- Co-authored-by: jack <jackjackbits@users.noreply.github.com>
112 lines
3.9 KiB
Swift
112 lines
3.9 KiB
Swift
import BitFoundation
|
|
import Foundation
|
|
import Testing
|
|
@testable import bitchat
|
|
|
|
@Suite("BLE peer registry tests")
|
|
struct BLEPeerRegistryTests {
|
|
@Test("upserted announces track new, reconnect, and rename transitions")
|
|
func upsertVerifiedAnnounceTracksTransitions() {
|
|
var registry = BLEPeerRegistry()
|
|
let peerID = PeerID(str: "1122334455667788")
|
|
let firstSeen = Date(timeIntervalSince1970: 100)
|
|
|
|
let first = registry.upsertVerifiedAnnounce(
|
|
peerID: peerID,
|
|
nickname: "alice",
|
|
noisePublicKey: Data([1, 2, 3]),
|
|
signingPublicKey: Data([4, 5, 6]),
|
|
isConnected: true,
|
|
now: firstSeen
|
|
)
|
|
|
|
#expect(first.isNewPeer)
|
|
#expect(!first.wasDisconnected)
|
|
#expect(first.previousNickname == nil)
|
|
#expect(registry.connectedPeerIDs == [peerID])
|
|
#expect(registry.nickname(for: peerID, connectedOnly: true) == "alice")
|
|
|
|
registry.markDisconnected(peerID)
|
|
let reconnect = registry.upsertVerifiedAnnounce(
|
|
peerID: peerID,
|
|
nickname: "alice-renamed",
|
|
noisePublicKey: Data([1, 2, 3]),
|
|
signingPublicKey: Data([4, 5, 6]),
|
|
isConnected: true,
|
|
now: firstSeen.addingTimeInterval(1)
|
|
)
|
|
|
|
#expect(!reconnect.isNewPeer)
|
|
#expect(reconnect.wasDisconnected)
|
|
#expect(reconnect.previousNickname == "alice")
|
|
#expect(registry.info(for: peerID)?.nickname == "alice-renamed")
|
|
}
|
|
|
|
@Test("reachability keeps recent verified offline peers only when mesh is attached")
|
|
func reachabilityRequiresMeshAttachmentForOfflinePeers() {
|
|
let offlinePeer = PeerID(str: "1122334455667788")
|
|
let connectedPeer = PeerID(str: "8877665544332211")
|
|
let now = Date()
|
|
|
|
var isolatedRegistry = BLEPeerRegistry()
|
|
isolatedRegistry.upsert(BLEPeerInfo(
|
|
peerID: offlinePeer,
|
|
nickname: "offline",
|
|
isConnected: false,
|
|
noisePublicKey: nil,
|
|
signingPublicKey: nil,
|
|
isVerifiedNickname: true,
|
|
lastSeen: now
|
|
))
|
|
|
|
#expect(!isolatedRegistry.isReachable(offlinePeer, now: now))
|
|
|
|
var attachedRegistry = isolatedRegistry
|
|
attachedRegistry.upsert(BLEPeerInfo(
|
|
peerID: connectedPeer,
|
|
nickname: "connected",
|
|
isConnected: true,
|
|
noisePublicKey: nil,
|
|
signingPublicKey: nil,
|
|
isVerifiedNickname: true,
|
|
lastSeen: now
|
|
))
|
|
|
|
#expect(attachedRegistry.isReachable(offlinePeer, now: now))
|
|
}
|
|
|
|
@Test("connectivity reconciliation disconnects inactive peers and prunes expired offline peers")
|
|
func reconcileConnectivityUpdatesAndPrunesPeerState() {
|
|
var registry = BLEPeerRegistry()
|
|
let inactiveConnectedPeer = PeerID(str: "1122334455667788")
|
|
let expiredOfflinePeer = PeerID(str: "8877665544332211")
|
|
let now = Date()
|
|
|
|
registry.upsert(BLEPeerInfo(
|
|
peerID: inactiveConnectedPeer,
|
|
nickname: "inactive",
|
|
isConnected: true,
|
|
noisePublicKey: nil,
|
|
signingPublicKey: nil,
|
|
isVerifiedNickname: true,
|
|
lastSeen: now.addingTimeInterval(-TransportConfig.blePeerInactivityTimeoutSeconds - 1)
|
|
))
|
|
registry.upsert(BLEPeerInfo(
|
|
peerID: expiredOfflinePeer,
|
|
nickname: "expired",
|
|
isConnected: false,
|
|
noisePublicKey: nil,
|
|
signingPublicKey: nil,
|
|
isVerifiedNickname: true,
|
|
lastSeen: .distantPast
|
|
))
|
|
|
|
let changes = registry.reconcileConnectivity(now: now, linkStates: [:])
|
|
|
|
#expect(changes.disconnectedPeerIDs == [inactiveConnectedPeer])
|
|
#expect(changes.removedPeers.map(\.peerID) == [expiredOfflinePeer])
|
|
#expect(registry.info(for: inactiveConnectedPeer)?.isConnected == false)
|
|
#expect(registry.info(for: expiredOfflinePeer) == nil)
|
|
}
|
|
}
|