Add test coverage to previews

Author: nuno-vieira

Sources/StreamChatSwiftUI/ChatChannelList/ChatChannelListItem.swift

@@ -107,7 +107,7 @@ public struct ChatChannelListItem<Factory: ViewFactory>: View {
                 }
             }
             if utils.messageListConfig.draftMessagesEnabled, let draftText = channel.draftMessageText {
-                HStack {
+                HStack(spacing: 2) {
                     Text("\(L10n.Message.Preview.draft):")
                         .font(fonts.caption1).bold()
                         .foregroundColor(Color(colors.highlightedAccentBackground))

Sources/StreamChatSwiftUI/ChatThreadList/ChatThreadListItem.swift

@@ -186,7 +186,7 @@ struct ChatThreadListItemContentView: View {
                     .font(fonts.subheadlineBold)
                 HStack {
                     if let draftText {
-                        HStack {
+                        HStack(spacing: 2) {
                             draftPrefixView
                             SubtitleText(text: draftText)
                         }

StreamChatSwiftUITests/Tests/ChatChannelList/ChatChannelListItemView_Tests.swift

@@ -18,8 +18,9 @@ final class ChatChannelListItemView_Tests: StreamChatTestCase {
         streamChat?.utils.channelHeaderLoader.placeholder2 = circleImage
         streamChat?.utils.channelHeaderLoader.placeholder3 = circleImage
         streamChat?.utils.channelHeaderLoader.placeholder4 = circleImage
+        streamChat?.utils.messageListConfig = .init(draftMessagesEnabled: true)
     }
-    
+
     func test_channelListItem_audioMessage() throws {
         // Given
         let message = try mockAudioMessage(text: "Audio", isSentByCurrentUser: true)
@@ -202,6 +203,46 @@ final class ChatChannelListItemView_Tests: StreamChatTestCase {
         assertSnapshot(matching: view, as: .image(perceptualPrecision: precision))
     }
 
+    func test_channelListItem_draftMessage() throws {
+        // Given
+        let message = DraftMessage.mock(text: "Draft message")
+        let channel = ChatChannel.mock(cid: .unique, previewMessage: .mock(), draftMessage: message)
+
+        // When
+        let view = ChatChannelListItem(
+            channel: channel,
+            channelName: "Test",
+            avatar: .circleImage,
+            onlineIndicatorShown: true,
+            onItemTap: { _ in }
+        )
+        .frame(width: defaultScreenSize.width)
+        
+        // Then
+        assertSnapshot(matching: view, as: .image(perceptualPrecision: precision))
+    }
+
+    func test_channelListItem_draftMessageWithAttachment() throws {
+        // Given
+        let message = DraftMessage.mock(text: "Draft message", attachments: [.dummy(payload: try JSONEncoder().encode(
+            ImageAttachmentPayload(title: "Test", imageRemoteURL: .localYodaImage, file: .init(url: .localYodaImage))
+        ))])
+        let channel = ChatChannel.mock(cid: .unique, previewMessage: .mock(), draftMessage: message)
+
+        // When
+        let view = ChatChannelListItem(
+            channel: channel,
+            channelName: "Test",
+            avatar: .circleImage,
+            onlineIndicatorShown: true,
+            onItemTap: { _ in }
+        )
+        .frame(width: defaultScreenSize.width)
+
+        // Then
+        assertSnapshot(matching: view, as: .image(perceptualPrecision: precision))
+    }
+
     // MARK: - private
 
     private func mockAudioMessage(text: String, isSentByCurrentUser: Bool) throws -> ChatMessage {

StreamChatSwiftUITests/Tests/ChatChannelList/__Snapshots__/ChatChannelListItemView_Tests/test_channelListItem_draftMessage.1.png

@@ -23,6 +23,7 @@ final class ChatThreadListItemView_Tests: StreamChatTestCase {
         streamChat?.utils.channelHeaderLoader.placeholder2 = circleImage
         streamChat?.utils.channelHeaderLoader.placeholder3 = circleImage
         streamChat?.utils.channelHeaderLoader.placeholder4 = circleImage
+        streamChat?.utils.messageListConfig = .init(draftMessagesEnabled: true)
 
         currentUser = ChatUser.mock(id: StreamChatTestCase.currentUserId, name: "Vader", imageURL: nil)
 
@@ -130,6 +131,32 @@ final class ChatThreadListItemView_Tests: StreamChatTestCase {
 
         assertSnapshot(matching: view, as: .image(perceptualPrecision: precision))
     }
+
+    func test_threadListItem_whenDraftMessage() {
+        let thread = mockThread
+            .with(parentMessage: .mock(text: "Parent", draftReply: .mock(text: "Draft message")))
+
+        let view = ChatThreadListItem(thread: thread)
+            .frame(width: defaultScreenSize.width)
+
+        assertSnapshot(matching: view, as: .image(perceptualPrecision: precision))
+    }
+
+    func test_threadListItem_whenDraftMessageHasAttachment() throws {
+        let message = DraftMessage.mock(text: "Draft message", attachments: [.dummy(payload: try JSONEncoder().encode(
+            ImageAttachmentPayload(title: "Test", imageRemoteURL: .localYodaImage, file: .init(url: .localYodaImage))
+        ))])
+        let thread = mockThread
+            .with(parentMessage: .mock(
+                text: "Parent",
+                draftReply: message
+            ))
+
+        let view = ChatThreadListItem(thread: thread)
+            .frame(width: defaultScreenSize.width)
+
+        assertSnapshot(matching: view, as: .image(perceptualPrecision: precision))
+    }
 }
 
 extension ChatThreadListItem {