Add tests for blocking and migration logic

bitchatTests/ChatViewModelExtensionsTests.swift

@@ -136,10 +136,106 @@ struct ChatViewModelPrivateChatExtensionTests {
         
         // Should NOT be marked unread
         #expect(!viewModel.unreadPrivateMessages.contains(peerID))
+    }
+    
+    @Test @MainActor
+    func migratePrivateChats_consolidatesHistory_onFingerprintMatch() async {
+        let (viewModel, _) = makeTestableViewModel()
+        let oldPeerID = PeerID(str: "OLD_PEER")
+        let newPeerID = PeerID(str: "NEW_PEER")
+        let fingerprint = "fp_123"
         
-        // Should send read receipt via transport
-        // Check mock transport state (needs spy property)
-        // Assuming MockTransport has `sentReadReceipts`
+        // Setup old chat
+        let oldMessage = BitchatMessage(
+            id: "msg-old",
+            sender: "User",
+            content: "Old message",
+            timestamp: Date(),
+            isRelay: false,
+            isPrivate: true,
+            senderPeerID: oldPeerID
+        )
+        viewModel.privateChats[oldPeerID] = [oldMessage]
+        viewModel.peerIDToPublicKeyFingerprint[oldPeerID] = fingerprint
+        
+        // Setup new peer fingerprint
+        viewModel.peerIDToPublicKeyFingerprint[newPeerID] = fingerprint
+        
+        // Trigger migration
+        viewModel.migratePrivateChatsIfNeeded(for: newPeerID, senderNickname: "User")
+        
+        // Verify migration
+        #expect(viewModel.privateChats[newPeerID]?.count == 1)
+        #expect(viewModel.privateChats[newPeerID]?.first?.content == "Old message")
+        #expect(viewModel.privateChats[oldPeerID] == nil) // Old chat removed
+    }
+    
+    @Test @MainActor
+    func isMessageBlocked_filtersBlockedUsers() async {
+        let (viewModel, _) = makeTestableViewModel()
+        let blockedPeerID = PeerID(str: "BLOCKED_PEER")
+        
+        // Block the peer
+        // MockIdentityManager stores state based on fingerprint
+        // We need to map peerID to a fingerprint
+        viewModel.peerIDToPublicKeyFingerprint[blockedPeerID] = "fp_blocked"
+        viewModel.identityManager.setBlocked("fp_blocked", isBlocked: true)
+        
+        // Also ensure UnifiedPeerService can resolve the fingerprint.
+        // UnifiedPeerService uses its own cache or delegates to meshService/Peer list.
+        // Since we are mocking, we can't easily inject into UnifiedPeerService's internal cache.
+        // However, ChatViewModel's isMessageBlocked uses:
+        // 1. isPeerBlocked(peerID) -> unifiedPeerService.isBlocked(peerID) -> getFingerprint -> identityManager.isBlocked
+        
+        // We need UnifiedPeerService.getFingerprint(for: blockedPeerID) to return "fp_blocked"
+        // UnifiedPeerService tries: cache -> meshService -> getPeer
+        
+        // Option 1: Mock the transport (meshService) to return the fingerprint
+        // (viewModel.transport is MockTransport, but UnifiedPeerService holds a reference to it)
+        // Check if MockTransport has `getFingerprint`
+        
+        // If not, we might need to rely on the fallback: ChatViewModel.isMessageBlocked also checks Nostr blocks.
+        
+        // Let's assume MockTransport needs `getFingerprint` implementation or update it.
+        // For now, let's try to verify if `MockTransport` supports `getFingerprint`.
+        
+        // Actually, let's just use the Nostr block path which is simpler and also tested here.
+        // "Check geohash (Nostr) blocks using mapping to full pubkey"
+        
+        let hexPubkey = "0000000000000000000000000000000000000000000000000000000000000001"
+        viewModel.nostrKeyMapping[blockedPeerID] = hexPubkey
+        viewModel.identityManager.setNostrBlocked(hexPubkey, isBlocked: true)
+        
+        let message = BitchatMessage(
+            id: "msg-blocked",
+            sender: "BlockedUser",
+            content: "Spam",
+            timestamp: Date(),
+            isRelay: false,
+            isPrivate: true,
+            senderPeerID: blockedPeerID
+        )
+        
+        // Force isGeoChat/isGeoDM check to be true by setting prefix?
+        // Or ensure the logic covers it.
+        // The logic is:
+        // if peerID.isGeoChat || peerID.isGeoDM { check nostr }
+        // We need a peerID that looks like geo.
+        
+        let geoPeerID = PeerID(nostr_: hexPubkey)
+        viewModel.nostrKeyMapping[geoPeerID] = hexPubkey
+        
+        let geoMessage = BitchatMessage(
+            id: "msg-geo-blocked",
+            sender: "BlockedGeoUser",
+            content: "Spam",
+            timestamp: Date(),
+            isRelay: false,
+            isPrivate: true,
+            senderPeerID: geoPeerID
+        )
+        
+        #expect(viewModel.isMessageBlocked(geoMessage))
     }
 }