bitchat/bitchatTests/ChatTransportEventCoordinatorContextTests.swift

//
// ChatTransportEventCoordinatorContextTests.swift
// bitchatTests
//
// Exercises `ChatTransportEventCoordinator` against a mock
// `ChatTransportEventContext` — proving the coordinator works without a
// `ChatViewModel`, following the `ChatDeliveryCoordinatorContextTests` /
// `ChatPrivateConversationCoordinatorContextTests` exemplars.
//
// Scope note: the coordinator hops every event onto the main actor via an
// internal `Task`; tests drain those tasks with `Task.yield()`. All flows are
// mockable — no singletons are involved at this layer.
//

import Testing
import Foundation
import BitFoundation
@testable import bitchat

// MARK: - Mock Context

/// Lightweight stand-in for `ChatTransportEventContext` proving that
/// `ChatTransportEventCoordinator` is testable without a `ChatViewModel`.
@MainActor
private final class MockChatTransportEventContext: ChatTransportEventContext {
    // Connection & chat state
    var isConnected = false
    var nickname = "me"
    var myPeerID = PeerID(str: "0011223344556677")
    var privateChats: [PeerID: [BitchatMessage]] = [:]

    func privateMessages(for peerID: PeerID) -> [BitchatMessage] {
        privateChats[peerID] ?? []
    }
    var unreadPrivateMessages: Set<PeerID> = []
    var selectedPrivateChatPeer: PeerID?
    private(set) var unmarkedReadReceiptBatches: [[String]] = []
    private(set) var notifyUIChangedCount = 0

    // Conversation store intents (mirror `ConversationStore` semantics)
    @discardableResult
    func appendPrivateMessage(_ message: BitchatMessage, to peerID: PeerID) -> Bool {
        var chat = privateChats[peerID] ?? []
        guard !chat.contains(where: { $0.id == message.id }) else { return false }
        let index = chat.firstIndex(where: { $0.timestamp > message.timestamp }) ?? chat.count
        chat.insert(message, at: index)
        privateChats[peerID] = chat
        return true
    }

    func removePrivateChat(_ peerID: PeerID) {
        privateChats.removeValue(forKey: peerID)
        unreadPrivateMessages.remove(peerID)
    }

    func markPrivateChatUnread(_ peerID: PeerID) {
        unreadPrivateMessages.insert(peerID)
    }

    func markPrivateChatRead(_ peerID: PeerID) {
        unreadPrivateMessages.remove(peerID)
    }

    func unmarkReadReceiptsSent(_ ids: [String]) {
        unmarkedReadReceiptBatches.append(ids)
    }

    func notifyUIChanged() {
        notifyUIChangedCount += 1
    }

    // Inbound message handling
    var blockedMessageIDs: Set<String> = []
    private(set) var handledPrivateMessages: [BitchatMessage] = []
    private(set) var handledPublicMessages: [BitchatMessage] = []
    private(set) var mentionCheckedMessageIDs: [String] = []
    private(set) var hapticMessageIDs: [String] = []

    func isMessageBlocked(_ message: BitchatMessage) -> Bool {
        blockedMessageIDs.contains(message.id)
    }

    func handlePrivateMessage(_ message: BitchatMessage) {
        handledPrivateMessages.append(message)
    }

    func handlePublicMessage(_ message: BitchatMessage) {
        handledPublicMessages.append(message)
    }

    func checkForMentions(_ message: BitchatMessage) {
        mentionCheckedMessageIDs.append(message.id)
    }

    func sendHapticFeedback(for message: BitchatMessage) {
        hapticMessageIDs.append(message.id)
    }

    func parseMentions(from content: String) -> [String] {
        content.contains("@me") ? ["me"] : []
    }

    // Peer identity & sessions
    var blockedPeers: Set<PeerID> = []
    var peersByID: [PeerID: BitchatPeer] = [:]
    var noiseSessionKeysByPeerID: [PeerID: Data] = [:]
    private(set) var stablePeerIDCache: [PeerID: PeerID] = [:]
    private(set) var registeredEphemeralSessions: [PeerID] = []
    private(set) var removedEphemeralSessions: [PeerID] = []

    func isPeerBlocked(_ peerID: PeerID) -> Bool { blockedPeers.contains(peerID) }
    func unifiedPeer(for peerID: PeerID) -> BitchatPeer? { peersByID[peerID] }
    func resolveNickname(for peerID: PeerID) -> String { "anon\(peerID.id.prefix(4))" }
    func registerEphemeralSession(peerID: PeerID) { registeredEphemeralSessions.append(peerID) }
    func removeEphemeralSession(peerID: PeerID) { removedEphemeralSessions.append(peerID) }
    func noiseSessionPublicKeyData(for peerID: PeerID) -> Data? { noiseSessionKeysByPeerID[peerID] }
    func cacheStablePeerID(_ stablePeerID: PeerID, for shortPeerID: PeerID) {
        stablePeerIDCache[shortPeerID] = stablePeerID
    }
    func cachedStablePeerID(for shortPeerID: PeerID) -> PeerID? { stablePeerIDCache[shortPeerID] }

    // Routing & acknowledgements
    private(set) var flushedOutboxPeerIDs: [PeerID] = []
    private(set) var meshDeliveryAcks: [(messageID: String, peerID: PeerID)] = []

    func flushRouterOutbox(for peerID: PeerID) { flushedOutboxPeerIDs.append(peerID) }
    func sendMeshDeliveryAck(for messageID: String, to peerID: PeerID) {
        meshDeliveryAcks.append((messageID, peerID))
    }

    // Delivery status
    var applyMessageDeliveryStatusResult = true
    var deliveryStatusesByMessageID: [String: DeliveryStatus] = [:]
    private(set) var appliedDeliveryStatuses: [(messageID: String, status: DeliveryStatus)] = []

    @discardableResult
    func applyMessageDeliveryStatus(_ messageID: String, status: DeliveryStatus) -> Bool {
        appliedDeliveryStatuses.append((messageID, status))
        return applyMessageDeliveryStatusResult
    }

    func deliveryStatus(for messageID: String) -> DeliveryStatus? {
        deliveryStatusesByMessageID[messageID]
    }

    // Verification payloads
    private(set) var verifyChallengePayloads: [(peerID: PeerID, payload: Data)] = []
    private(set) var verifyResponsePayloads: [(peerID: PeerID, payload: Data)] = []

    func handleVerifyChallengePayload(from peerID: PeerID, payload: Data) {
        verifyChallengePayloads.append((peerID, payload))
    }

    func handleVerifyResponsePayload(from peerID: PeerID, payload: Data) {
        verifyResponsePayloads.append((peerID, payload))
    }
}

// MARK: - Helpers

/// Lets the coordinator's internal `Task { @MainActor … }` hops run.
@MainActor
private func drainMainActorTasks() async {
    for _ in 0..<10 { await Task.yield() }
}

private func makeMessage(
    id: String,
    sender: String = "alice",
    content: String = "hello",
    isPrivate: Bool = false,
    senderPeerID: PeerID? = nil
) -> BitchatMessage {
    BitchatMessage(
        id: id,
        sender: sender,
        content: content,
        timestamp: Date(),
        isRelay: false,
        isPrivate: isPrivate,
        recipientNickname: isPrivate ? "me" : nil,
        senderPeerID: senderPeerID
    )
}

// MARK: - Coordinator Tests Against Mock Context

/// Exercises `ChatTransportEventCoordinator` against
/// `MockChatTransportEventContext` with no `ChatViewModel`.
struct ChatTransportEventCoordinatorContextTests {

    @Test @MainActor
    func didReceiveMessage_routesPrivateAndPublic_skipsBlockedAndEmpty() async {
        let context = MockChatTransportEventContext()
        let coordinator = ChatTransportEventCoordinator(context: context)

        // Blocked messages are dropped before any handling.
        context.blockedMessageIDs = ["blocked"]
        coordinator.didReceiveMessage(makeMessage(id: "blocked"))
        // Empty public content is dropped too.
        coordinator.didReceiveMessage(makeMessage(id: "empty", content: "   "))
        await drainMainActorTasks()
        #expect(context.handledPublicMessages.isEmpty)
        #expect(context.handledPrivateMessages.isEmpty)
        #expect(context.mentionCheckedMessageIDs.isEmpty)

        // Private goes to the private handler, public to the public handler;
        // both get mention checks and haptics.
        coordinator.didReceiveMessage(makeMessage(id: "pm", isPrivate: true))
        coordinator.didReceiveMessage(makeMessage(id: "pub"))
        await drainMainActorTasks()
        #expect(context.handledPrivateMessages.map(\.id) == ["pm"])
        #expect(context.handledPublicMessages.map(\.id) == ["pub"])
        #expect(context.mentionCheckedMessageIDs == ["pm", "pub"])
        #expect(context.hapticMessageIDs == ["pm", "pub"])
    }

    @Test @MainActor
    func didReceivePublicMessage_trimsContentAndParsesMentions() async {
        let context = MockChatTransportEventContext()
        let coordinator = ChatTransportEventCoordinator(context: context)
        let peerID = PeerID(str: "aabbccdd00112233")

        coordinator.didReceivePublicMessage(
            from: peerID,
            nickname: "alice",
            content: "  hi @me  ",
            timestamp: Date(),
            messageID: "m1"
        )
        await drainMainActorTasks()

        #expect(context.handledPublicMessages.count == 1)
        let message = context.handledPublicMessages[0]
        #expect(message.content == "hi @me")
        #expect(message.mentions == ["me"])
        #expect(message.senderPeerID == peerID)
        #expect(context.hapticMessageIDs == ["m1"])
    }

    @Test @MainActor
    func didConnectAndDisconnect_manageSessionsStableIDsAndReadReceipts() async {
        let context = MockChatTransportEventContext()
        let coordinator = ChatTransportEventCoordinator(context: context)
        let peerID = PeerID(str: "1122334455667788")
        let noiseKey = Data(repeating: 0xAB, count: 32)
        context.peersByID[peerID] = BitchatPeer(peerID: peerID, noisePublicKey: noiseKey, nickname: "alice")

        coordinator.didConnectToPeer(peerID)
        await drainMainActorTasks()
        #expect(context.isConnected)
        #expect(context.registeredEphemeralSessions == [peerID])
        #expect(context.stablePeerIDCache[peerID] == PeerID(hexData: noiseKey))
        #expect(context.flushedOutboxPeerIDs == [peerID])
        #expect(context.notifyUIChangedCount == 1)

        // Their messages' read receipts are un-marked on disconnect so READ
        // acks can be re-sent after reconnect; our own messages are not.
        context.privateChats[peerID] = [
            makeMessage(id: "theirs-1", isPrivate: true, senderPeerID: peerID),
            makeMessage(id: "mine-1", sender: "me", isPrivate: true, senderPeerID: context.myPeerID),
            makeMessage(id: "theirs-2", isPrivate: true, senderPeerID: peerID),
        ]
        coordinator.didDisconnectFromPeer(peerID)
        await drainMainActorTasks()
        #expect(context.removedEphemeralSessions == [peerID])
        #expect(context.unmarkedReadReceiptBatches == [["theirs-1", "theirs-2"]])
        #expect(context.notifyUIChangedCount == 2)
    }

    @Test @MainActor
    func didDisconnect_whileViewingChat_migratesConversationToStablePeerID() async {
        let context = MockChatTransportEventContext()
        let coordinator = ChatTransportEventCoordinator(context: context)
        let peerID = PeerID(str: "1122334455667788")
        let noiseKey = Data(repeating: 0xCD, count: 32)
        let stablePeerID = PeerID(hexData: noiseKey)

        // No cached stable ID: it must be derived from the Noise session key.
        context.noiseSessionKeysByPeerID[peerID] = noiseKey
        context.selectedPrivateChatPeer = peerID
        context.unreadPrivateMessages = [peerID]
        context.privateChats[peerID] = [
            makeMessage(id: "m1", isPrivate: true, senderPeerID: peerID),
            makeMessage(id: "mine", sender: "me", isPrivate: true, senderPeerID: context.myPeerID),
        ]

        coordinator.didDisconnectFromPeer(peerID)
        await drainMainActorTasks()

        #expect(context.privateChats[peerID] == nil)
        #expect(context.privateChats[stablePeerID]?.map(\.id) == ["m1", "mine"])
        // Sender IDs migrate to the stable peer ID, except our own.
        #expect(context.privateChats[stablePeerID]?.first?.senderPeerID == stablePeerID)
        #expect(context.privateChats[stablePeerID]?.last?.senderPeerID == context.myPeerID)
        #expect(context.selectedPrivateChatPeer == stablePeerID)
        #expect(context.unreadPrivateMessages == [stablePeerID])
        #expect(context.stablePeerIDCache[peerID] == stablePeerID)
    }

    @Test @MainActor
    func noisePayloads_driveDeliveryStatusAcksAndVerification() async {
        let context = MockChatTransportEventContext()
        let coordinator = ChatTransportEventCoordinator(context: context)
        let peerID = PeerID(str: "99aabbccddeeff00")
        let noiseKey = Data(repeating: 0x44, count: 32)
        context.peersByID[peerID] = BitchatPeer(peerID: peerID, noisePublicKey: noiseKey, nickname: "alice")

        // Inbound private message: decoded, handled, and delivery-acked.
        let packet = PrivateMessagePacket(messageID: "pm-1", content: "hi there")
        coordinator.didReceiveNoisePayload(
            from: peerID,
            type: .privateMessage,
            payload: packet.encode() ?? Data(),
            timestamp: Date()
        )
        await drainMainActorTasks()
        #expect(context.handledPrivateMessages.map(\.id) == ["pm-1"])
        #expect(context.handledPrivateMessages.first?.sender == "alice")
        #expect(context.meshDeliveryAcks.count == 1)
        #expect(context.meshDeliveryAcks.first?.messageID == "pm-1")

        // Delivered / read acks resolve the display name from the unified peer.
        coordinator.didReceiveNoisePayload(from: peerID, type: .delivered, payload: Data("m-1".utf8), timestamp: Date())
        coordinator.didReceiveNoisePayload(from: peerID, type: .readReceipt, payload: Data("m-2".utf8), timestamp: Date())
        await drainMainActorTasks()
        #expect(context.appliedDeliveryStatuses.count == 2)
        #expect(context.appliedDeliveryStatuses[0].messageID == "m-1")
        if case .delivered(let to, _) = context.appliedDeliveryStatuses[0].status {
            #expect(to == "alice")
        } else {
            Issue.record("expected .delivered status")
        }
        if case .read(let by, _) = context.appliedDeliveryStatuses[1].status {
            #expect(by == "alice")
        } else {
            Issue.record("expected .read status")
        }

        // Verification payloads are forwarded untouched.
        coordinator.didReceiveNoisePayload(from: peerID, type: .verifyChallenge, payload: Data([0x01]), timestamp: Date())
        coordinator.didReceiveNoisePayload(from: peerID, type: .verifyResponse, payload: Data([0x02]), timestamp: Date())
        await drainMainActorTasks()
        #expect(context.verifyChallengePayloads.count == 1)
        #expect(context.verifyResponsePayloads.count == 1)

        // Blocked peers' private messages are dropped (no handling, no ack).
        context.blockedPeers = [peerID]
        coordinator.didReceiveNoisePayload(
            from: peerID,
            type: .privateMessage,
            payload: packet.encode() ?? Data(),
            timestamp: Date()
        )
        await drainMainActorTasks()
        #expect(context.handledPrivateMessages.count == 1)
        #expect(context.meshDeliveryAcks.count == 1)
    }
}