Fix UI tests

ElementX/Sources/Screens/HomeScreen/View/HomeScreen.swift

@@ -39,7 +39,7 @@ struct HomeScreen: View {
                     }
                 }
             }
-            .animation(.default, value: context.viewState.visibleRooms)
+            .animation(.elementDefault, value: context.viewState.visibleRooms)
             .padding(.horizontal)
             .searchable(text: $context.searchQuery)
         }

ElementX/Sources/Screens/RoomScreen/RoomScreenModels.swift

@@ -46,7 +46,6 @@ struct RoomScreenViewState: BindableState {
     var roomId: String
     var roomTitle = ""
     var roomAvatar: UIImage?
-    var roomEncryptionBadge: UIImage?
     var items: [RoomTimelineViewProvider] = []
     var isBackPaginating = false
     var bindings: RoomScreenViewStateBindings

ElementX/Sources/Screens/RoomScreen/RoomScreenViewModel.swift

@@ -33,8 +33,7 @@ class RoomScreenViewModel: RoomScreenViewModelType, RoomScreenViewModelProtocol
          timelineViewFactory: RoomTimelineViewFactoryProtocol,
          mediaProvider: MediaProviderProtocol,
          roomName: String?,
-         roomAvatarUrl: String? = nil,
-         roomEncryptionBadge: UIImage? = nil) {
+         roomAvatarUrl: String? = nil) {
         self.timelineController = timelineController
         self.timelineViewFactory = timelineViewFactory
         self.mediaProvider = mediaProvider
@@ -42,7 +41,6 @@ class RoomScreenViewModel: RoomScreenViewModelType, RoomScreenViewModelProtocol
         super.init(initialViewState: RoomScreenViewState(roomId: timelineController.roomId,
                                                          roomTitle: roomName ?? "Unknown room 💥",
                                                          roomAvatar: nil,
-                                                         roomEncryptionBadge: roomEncryptionBadge,
                                                          bindings: .init(composerText: "", composerFocused: false)))
         
         timelineController.callbacks

ElementX/Sources/Screens/RoomScreen/View/RoomHeaderView.swift

@@ -37,11 +37,6 @@ struct RoomHeaderView: View {
         ZStack(alignment: .bottomTrailing) {
             roomAvatarImage
                 .clipShape(Circle())
-
-            if let encryptionBadge = context.viewState.roomEncryptionBadge {
-                Image(uiImage: encryptionBadge)
-                    .accessibilityIdentifier("encryptionBadgeIcon")
-            }
         }
         .frame(width: AvatarSize.room(on: .timeline).value, height: AvatarSize.room(on: .timeline).value)
     }
@@ -87,8 +82,7 @@ struct RoomHeaderView_Previews: PreviewProvider {
                                             timelineViewFactory: RoomTimelineViewFactory(),
                                             mediaProvider: MockMediaProvider(),
                                             roomName: "Some Room name",
-                                            roomAvatarUrl: nil,
-                                            roomEncryptionBadge: Asset.Images.encryptionTrusted.image)
+                                            roomAvatarUrl: nil)
 
         RoomHeaderView(context: viewModel.context)
             .previewLayout(.sizeThatFits)

ElementX/Sources/Services/Media/MockMediaProvider.swift

@@ -43,6 +43,10 @@ struct MockMediaProvider: MediaProviderProtocol {
     }
         
     func loadImageFromURLString(_ urlString: String, avatarSize: AvatarSize?) async -> Result<UIImage, MediaProviderError> {
-        .failure(.failedRetrievingImage)
+        guard let image = UIImage(systemName: "photo") else {
+            fatalError()
+        }
+        
+        return .success(image)
     }
 }