Possibility to add custom snapshot generation logic

Author: martinmitrevski

CHANGELOG.md

@@ -6,6 +6,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 ### ✅ Added
 - Configuration for stack based navigation for iPads
 - Customization of the reactions background
+- Possibility to add custom snapshot generation logic
 
 # [4.21.0](https://github.com/GetStream/stream-chat-swiftui/releases/tag/4.21.0)
 _September 02, 2022_

Sources/StreamChatSwiftUI/Utils/SnapshotCreator.swift

@@ -21,10 +21,14 @@ public class DefaultSnapshotCreator: SnapshotCreator {
     public init() { /* Public init. */ }
 
     public func makeSnapshot(for view: AnyView) -> UIImage {
-        let currentSnapshot: UIImage?
-        guard let view: UIView = topVC()?.view else {
+        guard let uiView: UIView = topVC()?.view else {
             return images.snapshot
         }
+        return makeSnapshot(from: uiView)
+    }
+    
+    func makeSnapshot(from view: UIView) -> UIImage {
+        let currentSnapshot: UIImage?
         UIGraphicsBeginImageContext(view.frame.size)
         if let currentGraphicsContext = UIGraphicsGetCurrentContext() {
             view.layer.render(in: currentGraphicsContext)

StreamChatSwiftUITests/Tests/ChatChannel/MessageListView_Tests.swift

@@ -60,6 +60,22 @@ class MessageListView_Tests: StreamChatTestCase {
         assertSnapshot(matching: messageListView, as: .image)
     }
 
+    func test_messageListView_snapshot() {
+        // Given
+        let channelConfig = ChannelConfig(reactionsEnabled: true)
+        let messageListView = makeMessageListView(channelConfig: channelConfig)
+            .applyDefaultSize()
+        
+        // When
+        let snapshotCreator = DefaultSnapshotCreator()
+        let viewController = UIHostingController(rootView: messageListView)
+        let snapshot = snapshotCreator.makeSnapshot(from: viewController.view)
+        let view = Image(uiImage: snapshot)
+        
+        // Then
+        assertSnapshot(matching: view, as: .image)
+    }
+    
     func test_messageListView_snapshotFallback() {
         // Given
         let channelConfig = ChannelConfig(reactionsEnabled: true)