Add more tests

Author: testableapple

StreamChatSwiftUITests/Tests/ChatChannel/ChatChannelViewModel_Tests.swift

@@ -17,19 +17,19 @@ class ChatChannelViewModel_Tests: StreamChatTestCase {
         channelController.channel_mock = initialChannel
         let viewModel = ChatChannelViewModel(channelController: channelController)
         XCTAssertEqual(initialChannel, viewModel.channel)
-        
+
         // When
         let updatedChannel = ChatChannel.mock(cid: cid)
         channelController.channel_mock = updatedChannel
         channelController.delegate?.channelController(
             channelController,
             didUpdateChannel: .update(updatedChannel)
         )
-        
+
         // Then
         XCTAssertEqual(updatedChannel, viewModel.channel)
     }
-    
+
     func test_chatChannelVM_messagesLoaded() {
         // Given
         let channelController = makeChannelController()
@@ -485,63 +485,212 @@ class ChatChannelViewModel_Tests: StreamChatTestCase {
         // Then
         XCTAssert(viewModel.threadMessage == nil)
     }
-    
+
     func test_chatChannelVM_jumpToInitialMessage() {
         // Given
         let message = ChatMessage.mock()
         let channelController = makeChannelController(messages: [message])
         let viewModel = ChatChannelViewModel(channelController: channelController)
-        
+
         // When
         let shouldJump = viewModel.jumpToMessage(messageId: message.messageId)
-    
+
         // Then
         XCTAssert(shouldJump == true)
     }
-    
+
     func test_chatChannelVM_jumpToAvailableMessage() {
         // Given
         let message1 = ChatMessage.mock()
         let message2 = ChatMessage.mock()
         let channelController = makeChannelController(messages: [message1, message2])
         let viewModel = ChatChannelViewModel(channelController: channelController)
-        
+
         // When
         let shouldJump = viewModel.jumpToMessage(messageId: message2.messageId)
-    
+
         // Then
         XCTAssert(shouldJump == true)
     }
-    
+
     func test_chatChannelVM_jumpToUnavailableMessage() {
         // Given
         let message1 = ChatMessage.mock()
         let message2 = ChatMessage.mock()
         let message3 = ChatMessage.mock()
         let channelController = makeChannelController(messages: [message1, message2])
         let viewModel = ChatChannelViewModel(channelController: channelController)
-        
+
         // When
         let shouldJump = viewModel.jumpToMessage(messageId: message3.messageId)
-    
+
         // Then
         XCTAssert(shouldJump == false)
     }
-    
+
     func test_chatChannelVM_jumpToUnknownMessage() {
         // Given
         let message1 = ChatMessage.mock()
         let message2 = ChatMessage.mock()
         let channelController = makeChannelController(messages: [message1, message2])
         let viewModel = ChatChannelViewModel(channelController: channelController)
-        
+
         // When
         let shouldJump = viewModel.jumpToMessage(messageId: .unknownMessageId)
-    
+
         // Then
         XCTAssert(shouldJump == false)
     }
-    
+
+    func test_chatChannelVM_jumpToMessage_setsHighlightedMessageId() {
+        // Given
+        let message1 = ChatMessage.mock()
+        let message2 = ChatMessage.mock()
+        let channelController = makeChannelController(messages: [message1, message2])
+        let viewModel = ChatChannelViewModel(channelController: channelController)
+        let testExpectation = XCTestExpectation(description: "Highlight should be set")
+        testExpectation.assertForOverFulfill = false
+
+        // When
+        let shouldJump = viewModel.jumpToMessage(messageId: message2.messageId)
+
+        // Then
+        XCTAssert(shouldJump == true)
+        DispatchQueue.main.asyncAfter(deadline: .now() + 0.2) {
+            XCTAssertEqual(viewModel.highlightedMessageId, message2.messageId)
+            testExpectation.fulfill()
+        }
+
+        wait(for: [testExpectation], timeout: 1.0)
+    }
+
+    func test_chatChannelVM_jumpToMessage_clearsHighlightedMessageId() {
+        // Given
+        let message1 = ChatMessage.mock()
+        let message2 = ChatMessage.mock()
+        let channelController = makeChannelController(messages: [message1, message2])
+        let viewModel = ChatChannelViewModel(channelController: channelController)
+        let testExpectation = XCTestExpectation(description: "Highlight should be cleared")
+
+        // When
+        _ = viewModel.jumpToMessage(messageId: message2.messageId)
+
+        // Then
+        DispatchQueue.main.asyncAfter(deadline: .now() + 0.8) {
+            XCTAssertNil(viewModel.highlightedMessageId)
+            testExpectation.fulfill()
+        }
+
+        wait(for: [testExpectation], timeout: 1.5)
+    }
+
+    func test_chatChannelVM_jumpToMessage_setsScrolledId() {
+        // Given
+        let message1 = ChatMessage.mock()
+        let message2 = ChatMessage.mock()
+        let channelController = makeChannelController(messages: [message1, message2])
+        let viewModel = ChatChannelViewModel(channelController: channelController)
+
+        // When
+        _ = viewModel.jumpToMessage(messageId: message2.messageId)
+
+        // Then
+        XCTAssertEqual(viewModel.scrolledId, message2.messageId)
+    }
+
+    func test_chatChannelVM_selectedMessageThread_setsJumpToReplyId() {
+        // Given
+        let channelController = makeChannelController()
+        let viewModel = ChatChannelViewModel(channelController: channelController)
+        let message = ChatMessage.mock(
+            id: .unique,
+            cid: .unique,
+            text: "Test message",
+            author: .mock(id: .unique)
+        )
+
+        // When
+        NotificationCenter.default.post(
+            name: NSNotification.Name(MessageRepliesConstants.selectedMessageThread),
+            object: nil,
+            userInfo: [MessageRepliesConstants.selectedMessage: message]
+        )
+
+        // Then
+        XCTAssertEqual(viewModel.threadMessage, message)
+        XCTAssertTrue(viewModel.threadMessageShown)
+        XCTAssertEqual(viewModel.messageCachingUtils.jumpToReplyId, message.messageId)
+    }
+
+    func test_chatChannelVM_selectedMessageThread_withThreadReplyMessage() {
+        // Given
+        let channelController = makeChannelController()
+        let viewModel = ChatChannelViewModel(channelController: channelController)
+        let parentMessage = ChatMessage.mock(
+            id: .unique,
+            cid: .unique,
+            text: "Parent message",
+            author: .mock(id: .unique)
+        )
+        let replyMessage = ChatMessage.mock(
+            id: .unique,
+            cid: .unique,
+            text: "Reply message",
+            author: .mock(id: .unique),
+            parentMessageId: parentMessage.id
+        )
+
+        // When
+        NotificationCenter.default.post(
+            name: NSNotification.Name(MessageRepliesConstants.selectedMessageThread),
+            object: nil,
+            userInfo: [
+                MessageRepliesConstants.selectedMessage: parentMessage,
+                MessageRepliesConstants.threadReplyMessage: replyMessage
+            ]
+        )
+
+        // Then
+        XCTAssertEqual(viewModel.threadMessage, parentMessage)
+        XCTAssertTrue(viewModel.threadMessageShown)
+        XCTAssertEqual(viewModel.messageCachingUtils.jumpToReplyId, replyMessage.messageId)
+    }
+
+    func test_chatChannelVM_threadViewHighlightsMessage() {
+        // Given
+        let channelController = makeChannelController()
+        let messageController = ChatMessageControllerSUI_Mock.mock(
+            chatClient: chatClient,
+            cid: .unique,
+            messageId: .unique
+        )
+        let viewModel = ChatChannelViewModel(
+            channelController: channelController,
+            messageController: messageController
+        )
+        let replyMessageId = "reply-message-id"
+        viewModel.messageCachingUtils.jumpToReplyId = replyMessageId
+        let testExpectation = XCTestExpectation(description: "Highlight should be set in thread")
+        testExpectation.assertForOverFulfill = false
+
+        // When - simulate the initialization path that checks jumpToReplyId
+        DispatchQueue.main.asyncAfter(deadline: .now() + 0.6) {
+            // Simulate thread view appearing and checking jumpToReplyId
+            if let jumpToReplyId = viewModel.messageCachingUtils.jumpToReplyId {
+                viewModel.scrolledId = jumpToReplyId
+                DispatchQueue.main.asyncAfter(deadline: .now() + 0.2) {
+                    viewModel.highlightedMessageId = jumpToReplyId
+                    testExpectation.fulfill()
+                }
+            }
+        }
+
+        // Then
+        wait(for: [testExpectation], timeout: 1.0)
+        XCTAssertEqual(viewModel.highlightedMessageId, replyMessageId)
+        XCTAssertEqual(viewModel.scrolledId, replyMessageId)
+    }
+
     func test_chatChannelVM_crashWhenIndexAccess() {
         // Given
         let message1 = ChatMessage.mock()
@@ -550,7 +699,7 @@ class ChatChannelViewModel_Tests: StreamChatTestCase {
         let channelController = makeChannelController(messages: [message1, message2])
         let viewModel = ChatChannelViewModel(channelController: channelController)
         let newMessages = LazyCachedMapCollection(elements: [message1, message2, message3])
-        
+
         // When
         viewModel.dataSource(
             channelDataSource: ChatChannelDataSource(controller: channelController),
@@ -561,11 +710,11 @@ class ChatChannelViewModel_Tests: StreamChatTestCase {
                 .update(message3, index: IndexPath(row: 3, section: 0)) // intentionally invalid path
             ]
         )
-        
+
         // Then
         XCTAssertEqual(3, viewModel.messages.count)
     }
-    
+
     func test_chatChannelVM_keepFirstUnreadIndexSetAfterMarkingTheChannelAsRead() {
         // Given
         let message1 = ChatMessage.mock()
@@ -576,26 +725,26 @@ class ChatChannelViewModel_Tests: StreamChatTestCase {
         let viewModel = ChatChannelViewModel(channelController: channelController)
         viewModel.firstUnreadMessageId = message1.id
         viewModel.throttler = Throttler_Mock(interval: 0)
-        
+
         // When
         viewModel.handleMessageAppear(index: 0, scrollDirection: .down)
-        
+
         // Then
         XCTAssertEqual(1, channelController.markReadCallCount)
         XCTAssertNotNil(viewModel.firstUnreadMessageId)
     }
-    
+
     // MARK: - currentUserMarkedMessageUnread Tests
-    
+
     func test_chatChannelVM_currentUserMarkedMessageUnread_initialValue() {
         // Given
         let channelController = makeChannelController()
         let viewModel = ChatChannelViewModel(channelController: channelController)
-        
+
         // Then
         XCTAssertFalse(viewModel.currentUserMarkedMessageUnread)
     }
-    
+
     func test_chatChannelVM_sendReadEventIfNeeded_whenCurrentUserMarkedMessageUnreadIsTrue() {
         // Given
         let message = ChatMessage.mock()
@@ -604,14 +753,14 @@ class ChatChannelViewModel_Tests: StreamChatTestCase {
         let viewModel = ChatChannelViewModel(channelController: channelController)
         viewModel.currentUserMarkedMessageUnread = true
         viewModel.throttler = Throttler_Mock(interval: 0)
-        
+
         // When
         viewModel.handleMessageAppear(index: 0, scrollDirection: .down)
-        
+
         // Then
         XCTAssertEqual(0, channelController.markReadCallCount)
     }
-    
+
     func test_chatChannelVM_sendReadEventIfNeeded_whenCurrentUserMarkedMessageUnreadIsFalse() {
         // Given
         let message = ChatMessage.mock()
@@ -620,14 +769,14 @@ class ChatChannelViewModel_Tests: StreamChatTestCase {
         let viewModel = ChatChannelViewModel(channelController: channelController)
         viewModel.currentUserMarkedMessageUnread = false
         viewModel.throttler = Throttler_Mock(interval: 0)
-        
+
         // When
         viewModel.handleMessageAppear(index: 0, scrollDirection: .down)
-        
+
         // Then
         XCTAssertEqual(1, channelController.markReadCallCount)
     }
-    
+
     func test_chatChannelVM_sendReadEventIfNeeded_whenChannelHasNoUnreadMessages() {
         // Given
         let message = ChatMessage.mock()
@@ -636,14 +785,14 @@ class ChatChannelViewModel_Tests: StreamChatTestCase {
         let viewModel = ChatChannelViewModel(channelController: channelController)
         viewModel.currentUserMarkedMessageUnread = false
         viewModel.throttler = Throttler_Mock(interval: 0)
-        
+
         // When
         viewModel.handleMessageAppear(index: 0, scrollDirection: .down)
-        
+
         // Then
         XCTAssertEqual(0, channelController.markReadCallCount)
     }
-    
+
     func test_chatChannelVM_sendReadEventIfNeeded_whenChannelIsNil() {
         // Given
         let message = ChatMessage.mock()
@@ -652,10 +801,10 @@ class ChatChannelViewModel_Tests: StreamChatTestCase {
         let viewModel = ChatChannelViewModel(channelController: channelController)
         viewModel.currentUserMarkedMessageUnread = false
         viewModel.throttler = Throttler_Mock(interval: 0)
-        
+
         // When
         viewModel.handleMessageAppear(index: 0, scrollDirection: .down)
-        
+
         // Then
         XCTAssertEqual(0, channelController.markReadCallCount)
     }