Refactor tests

Swap/Controller/ParticipantController.swift

@@ -70,6 +70,7 @@ enum ParticipantError: Error, Equatable {
 func ==(lhs: ParticipantError, rhs: ParticipantError) -> Bool {
     switch (lhs, rhs) {
     case (.notValid(let leftErrors), .notValid(let rightErrors)): return leftErrors == rightErrors
+    case (.notEnoughPicked(let leftCount), .notEnoughPicked(let rightCount)): return leftCount == rightCount
     default: return true
     }
 }

Swap/Info.plist

@@ -19,7 +19,7 @@
 	<key>CFBundleSignature</key>
 	<string>????</string>
 	<key>CFBundleVersion</key>
-	<string>598</string>
+	<string>604</string>
 	<key>LSRequiresIPhoneOS</key>
 	<true/>
 	<key>UILaunchStoryboardName</key>

SwapTests/Controller/ParticipantControllerTest.swift

@@ -28,32 +28,38 @@ class ParticipantControllerTest: XCTestCase {
         self.controller = ParticipantController(self.participant, parent: self.parent)
     }
 
-    func testToggle_withCurrentParticipant() {
-        let expected = [String]()
-
-        self.controller.toggle(self.participant)
-
-        XCTAssertEqual(self.controller.cantPick, expected)
-    }
-
     func testToggle_withParticipantNotPicked() {
         let expected = [Constant.other.id]
 
-        self.controller.toggle(Constant.other)
+        XCTAssertNoThrow(try self.controller.toggle(Constant.other), "toggle() did throw, but was not supposed to.")
 
         XCTAssertEqual(self.controller.cantPick, expected)
     }
 
-    func testPick_withAlreadyParticipantPicked() {
+    func testToggle_withAlreadyParticipantPicked() {
         let expected = [String]()
         self.participant.cantPick = [Constant.other.id]
         self.controller = ParticipantController(self.participant, parent: self.parent)
 
-        self.controller.toggle(Constant.other)
+        XCTAssertNoThrow(try self.controller.toggle(Constant.other), "toggle() did throw, but was not supposed to.")
 
         XCTAssertEqual(self.controller.cantPick, expected)
     }
 
+    func testToggle_shouldThrowCantToggleYourself_withCurrentParticipant() {
+        XCTAssertThrowsError(try self.controller.toggle(self.participant), "toggle() did throw, but was not supposed to.") { (error) in
+            XCTAssertEqual(error as! ParticipantError, ParticipantError.cantToggleYourself)
+        }
+    }
+
+    func testToggle_shouldThrowNotEnoughPicked() {
+        self.parent.setReturnedCount(integer: 1)
+
+        XCTAssertThrowsError(try self.controller.toggle(Constant.other), "toggle() did throw, but was not supposed to.") { (error) in
+            XCTAssertEqual(error as! ParticipantError, ParticipantError.notEnoughPicked(count: 0))
+        }
+    }
+
     func testSave_ShouldNotThrow_WithAllValid() {
         _ = self.participant.setUpValidateAreInvalid()
 

SwapTests/Mocks/ParticipantsControllerMock.swift

@@ -11,6 +11,19 @@ import Foundation
 
 class ParticipantsControllerMock: ParticipantsController {
 
+    // Count
+    private var mockCount: Int?
+    func setReturnedCount(integer: Int) {
+        self.mockCount = integer
+    }
 
+    override var count: Int {
+        get {
+            if let mock = self.mockCount {
+                return mock
+            }
+            return super.count
+        }
+    }
 
 }