From 6bf36b2d5593a2cb9a884bde4239449319ead316 Mon Sep 17 00:00:00 2001 From: Stefan Ceriu Date: Mon, 31 Jul 2023 14:22:09 +0300 Subject: [PATCH] Fix unit tests --- .../Sources/Screens/RoomScreen/RoomScreenViewModel.swift | 1 + .../RoomNotificationSettingsScreenViewModelTests.swift | 6 +++--- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/ElementX/Sources/Screens/RoomScreen/RoomScreenViewModel.swift b/ElementX/Sources/Screens/RoomScreen/RoomScreenViewModel.swift index ef7bf6131..e6a84e54c 100644 --- a/ElementX/Sources/Screens/RoomScreen/RoomScreenViewModel.swift +++ b/ElementX/Sources/Screens/RoomScreen/RoomScreenViewModel.swift @@ -304,6 +304,7 @@ class RoomScreenViewModel: RoomScreenViewModelType, RoomScreenViewModelProtocol selectivelyUpdateTimelineItems(timelineItemsDictionary: timelineItemsDictionary) } else { state.timelineViewState.itemsDictionary = timelineItemsDictionary + state.timelineViewState.renderedTimelineIDs = Array(timelineItemsDictionary.keys) } } diff --git a/UnitTests/Sources/RoomNotificationSettingsScreenViewModelTests.swift b/UnitTests/Sources/RoomNotificationSettingsScreenViewModelTests.swift index cc4c2969c..7ab923863 100644 --- a/UnitTests/Sources/RoomNotificationSettingsScreenViewModelTests.swift +++ b/UnitTests/Sources/RoomNotificationSettingsScreenViewModelTests.swift @@ -122,7 +122,7 @@ class RoomNotificationSettingsScreenViewModelTests: XCTestCase { try await deferredState.fulfill() do { - var deferredViewState = deferFulfillment(context.$viewState.collect(2).first()) + let deferredViewState = deferFulfillment(context.$viewState.collect(2).first()) context.send(viewAction: .setCustomMode(.allMessages)) try await deferredViewState.fulfill() @@ -132,7 +132,7 @@ class RoomNotificationSettingsScreenViewModelTests: XCTestCase { } do { - var deferredViewState = deferFulfillment(context.$viewState.collect(2).first()) + let deferredViewState = deferFulfillment(context.$viewState.collect(2).first()) context.send(viewAction: .setCustomMode(.mute)) try await deferredViewState.fulfill() @@ -142,7 +142,7 @@ class RoomNotificationSettingsScreenViewModelTests: XCTestCase { } do { - var deferredViewState = deferFulfillment(context.$viewState.collect(2).first()) + let deferredViewState = deferFulfillment(context.$viewState.collect(2).first()) context.send(viewAction: .setCustomMode(.mentionsAndKeywordsOnly)) try await deferredViewState.fulfill()