Add more tests

Author: laevandus

Tests/StreamChatTests/Controllers/ChannelController/ChannelController_Tests.swift

@@ -4326,6 +4326,86 @@ final class ChannelController_Tests: XCTestCase {
         XCTAssertEqual(updater.markUnread_lastReadMessageId, previousMessageId)
         XCTAssertEqual(updater.markUnread_criteria, MarkUnreadCriteria.messageId(messageId))
     }
+
+    func test_markUnread_whenChannelDoesNotExist_messageTimestamp() {
+        var receivedError: Error?
+        let expectation = self.expectation(description: "Mark Unread completes")
+        controller.markUnread(from: Date()) { result in
+            receivedError = result.error
+            expectation.fulfill()
+        }
+
+        waitForExpectations(timeout: defaultTimeout)
+
+        XCTAssertTrue(receivedError is ClientError.ChannelNotCreatedYet)
+    }
+
+    func test_markUnread_whenReadEventsAreNotEnabled_messageTimestamp() throws {
+        let channel: ChannelPayload = .dummy(
+            channel: .dummy(cid: channelId, ownCapabilities: [])
+        )
+
+        writeAndWaitForMessageUpdates(count: 0, channelChanges: true) { session in
+            try session.saveChannel(payload: channel)
+        }
+
+        var receivedError: Error?
+        let expectation = self.expectation(description: "Mark Unread completes")
+        controller.markUnread(from: Date()) { result in
+            receivedError = result.error
+            expectation.fulfill()
+        }
+
+        waitForExpectations(timeout: defaultTimeout)
+
+        XCTAssertTrue(receivedError is ClientError.ChannelFeatureDisabled)
+    }
+
+    func test_markUnread_whenIsMarkingAsRead_andCurrentUserIdIsPresent_messageTimestamp() throws {
+        let channel: ChannelPayload = .dummy(
+            channel: .dummy(cid: channelId, ownCapabilities: [ChannelCapability.readEvents.rawValue])
+        )
+
+        try client.databaseContainer.writeSynchronously { session in
+            try session.saveChannel(payload: channel)
+        }
+
+        let currentUserId = UserId.unique
+        client.setToken(token: .unique(userId: currentUserId))
+        try simulateMarkingAsRead(userId: currentUserId)
+
+        var receivedError: Error?
+        let expectation = self.expectation(description: "Mark Unread completes")
+        controller.markUnread(from: Date()) { result in
+            receivedError = result.error
+            expectation.fulfill()
+        }
+
+        waitForExpectations(timeout: defaultTimeout)
+
+        XCTAssertNil(receivedError)
+    }
+
+    func test_markUnread_whenIsNotMarkingAsRead_andCurrentUserIdIsNotPresent_messageTimestamp() throws {
+        let channel: ChannelPayload = .dummy(
+            channel: .dummy(cid: channelId, ownCapabilities: [ChannelCapability.readEvents.rawValue])
+        )
+
+        writeAndWaitForMessageUpdates(count: 0, channelChanges: true) { session in
+            try session.saveChannel(payload: channel)
+        }
+
+        var receivedError: Error?
+        let expectation = self.expectation(description: "Mark Unread completes")
+        controller.markUnread(from: Date()) { result in
+            receivedError = result.error
+            expectation.fulfill()
+        }
+
+        waitForExpectations(timeout: defaultTimeout)
+
+        XCTAssertNil(receivedError)
+    }
 
     // MARK: - Load more channel reads
 

Tests/StreamChatTests/StateLayer/Chat_Tests.swift

@@ -1308,6 +1308,44 @@ final class Chat_Tests: XCTestCase {
         await XCTAssertEqual(3, chat.state.channel?.reads.first?.unreadMessagesCount)
     }
 
+    func test_markUnread_whenAPIRequestSucceeds_thenReadStateUpdates_messageTimestamp() async throws {
+        try await setUpChat(
+            usesMockedUpdaters: false,
+            messageCount: 3
+        )
+        let messages = await chat.state.messages
+        let firstMessage = try XCTUnwrap(messages.first)
+        let lastMessage = try XCTUnwrap(messages.first)
+        
+        // Create a read state for the current user
+        try await env.client.databaseContainer.write { session in
+            let payload = ChannelPayload.dummy(
+                channel: .dummy(
+                    cid: self.channelId,
+                    lastMessageAt: lastMessage.createdAt
+                ),
+                channelReads: [
+                    ChannelReadPayload(
+                        user: .dummy(userId: self.currentUserId),
+                        lastReadAt: lastMessage.createdAt,
+                        lastReadMessageId: nil,
+                        unreadMessagesCount: 0,
+                        lastDeliveredAt: nil,
+                        lastDeliveredMessageId: nil
+                    )
+                ]
+            )
+            try session.saveChannel(payload: payload)
+        }
+        
+        env.client.mockAPIClient.test_mockResponseResult(.success(EmptyResponse()))
+        try await chat.markUnread(from: firstMessage.createdAt)
+        XCTAssertNotNil(env.client.mockAPIClient.request_endpoint)
+        
+        await XCTAssertEqual(1, chat.state.channel?.reads.count)
+        await XCTAssertEqual(3, chat.state.channel?.reads.first?.unreadMessagesCount)
+    }
+    
     // MARK: - Message Replies
 
     func test_reply_whenAPIRequestSucceeds_thenStateUpdates() async throws {