Fix preview of knocking.

element-hq · Jul 18, 2024 · 797a0d0 · 797a0d0
1 parent 230841c
commit 797a0d0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/ElementX/Sources/Screens/JoinRoomScreen/JoinRoomScreenViewModel.swift b/ElementX/Sources/Screens/JoinRoomScreen/JoinRoomScreenViewModel.swift
@@ -22,6 +22,7 @@ typealias JoinRoomScreenViewModelType = StateStoreViewModel<JoinRoomScreenViewSt
 class JoinRoomScreenViewModel: JoinRoomScreenViewModelType, JoinRoomScreenViewModelProtocol {
     private let roomID: String
     private let via: [String]
+    private let allowKnocking: Bool // For preview tests only, actions aren't sent.
     private let clientProxy: ClientProxyProtocol
     private let userIndicatorController: UserIndicatorControllerProtocol
 
@@ -32,11 +33,13 @@ class JoinRoomScreenViewModel: JoinRoomScreenViewModelType, JoinRoomScreenViewMo
 
     init(roomID: String,
          via: [String],
+         allowKnocking: Bool = false,
          clientProxy: ClientProxyProtocol,
          mediaProvider: MediaProviderProtocol,
          userIndicatorController: UserIndicatorControllerProtocol) {
         self.roomID = roomID
         self.via = via
+        self.allowKnocking = allowKnocking
         self.clientProxy = clientProxy
         self.userIndicatorController = userIndicatorController
 
@@ -98,8 +101,8 @@ class JoinRoomScreenViewModel: JoinRoomScreenViewModelType, JoinRoomScreenViewMo
             state.mode = .join
         } else if roomDetails.isInvited {
             state.mode = .invited
-        } else if roomDetails.canKnock { // Knocking is not supported yet, treat it as .unknown
-            state.mode = .unknown
+        } else if roomDetails.canKnock, allowKnocking { // Knocking is not supported yet, the flag is purely for preview tests.
+            state.mode = .knock
         } else {
             state.mode = .unknown
         }

diff --git a/ElementX/Sources/Screens/JoinRoomScreen/View/JoinRoomScreen.swift b/ElementX/Sources/Screens/JoinRoomScreen/View/JoinRoomScreen.swift
@@ -169,6 +169,7 @@ struct JoinRoomScreen_Previews: PreviewProvider, TestablePreview {
 
         return JoinRoomScreenViewModel(roomID: "1",
                                        via: [],
+                                       allowKnocking: true,
                                        clientProxy: clientProxy,
                                        mediaProvider: MockMediaProvider(),
                                        userIndicatorController: ServiceLocator.shared.userIndicatorController)