Fix snapshot tests

GetStream · Nov 20, 2024 · 2696168 · 2696168
1 parent 5a0bc11
commit 2696168
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/DemoApp/Sources/ViewModifiers/CallModifier/DemoCallModifier.swift b/DemoApp/Sources/ViewModifiers/CallModifier/DemoCallModifier.swift
@@ -29,6 +29,7 @@ struct DemoCallModifier<Factory: ViewFactory>: ViewModifier {
         contentView(content)
     }
 
+    @MainActor
     @ViewBuilder
     private func contentView(_ rootView: Content) -> some View {
         if
@@ -41,6 +42,7 @@ struct DemoCallModifier<Factory: ViewFactory>: ViewModifier {
                     type: call.callType,
                     id: call.callId,
                     handleParticipationWithLifecycle: false,
+                    showsLeaveCallButton: true,
                     onFullScreenStateChange: { [weak viewModel] in viewModel?.hideUIElements = $0 }
                 )
             }

diff --git a/Sources/StreamVideoSwiftUI/Livestreaming/LivestreamPlayer.swift b/Sources/StreamVideoSwiftUI/Livestreaming/LivestreamPlayer.swift
@@ -11,6 +11,7 @@ public struct LivestreamPlayer: View {
     @Injected(\.colors) var colors
 
     var handleParticipationWithLifecycle: Bool
+    var showsLeaveCallButton: Bool
     var onFullScreenStateChange: ((Bool) -> Void)?
 
     @StateObject var state: CallState
@@ -22,6 +23,7 @@ public struct LivestreamPlayer: View {
         muted: Bool = false,
         showParticipantCount: Bool = true,
         handleParticipationWithLifecycle: Bool = true,
+        showsLeaveCallButton: Bool = false,
         onFullScreenStateChange: ((Bool) -> Void)? = nil
     ) {
         let viewModel = LivestreamPlayerViewModel(
@@ -33,6 +35,7 @@ public struct LivestreamPlayer: View {
         _viewModel = StateObject(wrappedValue: viewModel)
         _state = StateObject(wrappedValue: viewModel.call.state)
         self.handleParticipationWithLifecycle = handleParticipationWithLifecycle
+        self.showsLeaveCallButton = showsLeaveCallButton
         self.onFullScreenStateChange = onFullScreenStateChange
     }
 
@@ -90,8 +93,10 @@ public struct LivestreamPlayer: View {
                                 LivestreamButton(imageName: "viewfinder") {
                                     viewModel.update(fullScreen: !viewModel.fullScreen)
                                 }
-                                LivestreamButton(imageName: "phone.down.fill") {
-                                    viewModel.leaveLivestream()
+                                if showsLeaveCallButton {
+                                    LivestreamButton(imageName: "phone.down.fill") {
+                                        viewModel.leaveLivestream()
+                                    }
                                 }
                             }
                             .padding()