Fix: TimelinePoll code refactoring

2026-04-29 12:46:58 +02:00 · 2023-04-26 15:31:07 +02:00
parent 782ad8465c
commit b563043b19
11 changed files with 143 additions and 117 deletions
@@ -48,22 +48,21 @@ enum MockTimelinePollScreenState: MockScreenState, CaseIterable {
                                       hasBeenEdited: false,
                                       hasDecryptionError: false)
        
-        let viewModel = TimelinePollViewModel(timelinePollDetails: poll)
+        let viewModel: TimelinePollViewModel
        
        switch self {
        case .loading:
-            viewModel.updateWithPollState(.loading)
+            viewModel = TimelinePollViewModel(timelinePollDetailsState: .loading)
        case .invalidStartEvent:
-            viewModel.updateWithPollState(.invalidStartEvent)
+            viewModel = TimelinePollViewModel(timelinePollDetailsState: .errored)
        default:
-            viewModel.updateWithPollState(.loaded)
+            viewModel = TimelinePollViewModel(timelinePollDetailsState: .loaded(poll))
        }
        
        if self == .withAlert {
            viewModel.showAnsweringFailure()
        }
        
-        
        return ([viewModel], AnyView(TimelinePollView(viewModel: viewModel.context)))
    }
 }