Add max selections support for polls (#5726)

2026-11-02 21:33:26 +00:00 · 2026-06-22 14:39:12 +03:00
parent 173a927bf8
commit 30e4e6339d
49 changed files with 506 additions and 251 deletions
@@ -0,0 +1,74 @@
+//
+// Copyright 2026 Element Creations Ltd.
+//
+// SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-Element-Commercial.
+// Please see LICENSE files in the repository root for full details.
+//
+
+@testable import ElementX
+import Testing
+
+struct PollTests {
+    @Test
+    func singleSelectionReplacesSelectedAnswer() {
+        let poll = Poll.mock(question: "Question?",
+                             maxSelections: 1,
+                             options: [
+                                 .mock(id: "first", text: "First", isSelected: true),
+                                 .mock(id: "second", text: "Second")
+                             ])
+        
+        #expect(poll.answerIDsAfterSelecting(optionID: "second") == ["second"])
+        #expect(poll.answerIDsAfterSelecting(optionID: "first") == nil)
+    }
+    
+    @Test
+    func multipleSelectionAddsAnswerUntilLimit() {
+        let poll = Poll.mock(question: "Question?",
+                             maxSelections: 2,
+                             options: [
+                                 .mock(id: "first", text: "First", isSelected: true),
+                                 .mock(id: "second", text: "Second"),
+                                 .mock(id: "third", text: "Third")
+                             ])
+        
+        #expect(poll.answerIDsAfterSelecting(optionID: "second") == ["first", "second"])
+    }
+    
+    @Test
+    func multipleSelectionCanRemoveAnswerWhenMoreThanOneIsSelected() {
+        let poll = Poll.mock(question: "Question?",
+                             maxSelections: 2,
+                             options: [
+                                 .mock(id: "first", text: "First", isSelected: true),
+                                 .mock(id: "second", text: "Second", isSelected: true),
+                                 .mock(id: "third", text: "Third")
+                             ])
+        
+        #expect(poll.answerIDsAfterSelecting(optionID: "first") == ["second"])
+    }
+    
+    @Test
+    func multipleSelectionDoesNotAddAnswerOverLimit() {
+        let poll = Poll.mock(question: "Question?",
+                             maxSelections: 2,
+                             options: [
+                                 .mock(id: "first", text: "First", isSelected: true),
+                                 .mock(id: "second", text: "Second", isSelected: true),
+                                 .mock(id: "third", text: "Third")
+                             ])
+        
+        #expect(poll.answerIDsAfterSelecting(optionID: "third") == nil)
+    }
+}
+
+private extension Poll.Option {
+    static func mock(id: String, text: String, isSelected: Bool = false) -> Self {
+        .init(id: id,
+              text: text,
+              votes: 0,
+              allVotes: 0,
+              isSelected: isSelected,
+              isWinning: false)
+    }
+}