Add tests for AccountAction.updateNotificationSettings

woocommerce · Dec 20, 2024 · 018501e · 018501e
1 parent 9d38e56
commit 018501e
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 0 deletions.
diff --git a/Yosemite/YosemiteTests/Mocks/Networking/Remote/MockAccountRemote.swift b/Yosemite/YosemiteTests/Mocks/Networking/Remote/MockAccountRemote.swift
@@ -55,6 +55,11 @@ final class MockAccountRemote {
     func whenCreatingAccount(thenReturn result: Result<CreateAccountResult, CreateAccountError>) {
         createAccountResult = result
     }
+
+    /// Returns  the value when `updateNotificationSettings` is called.
+    func whenUpdatingNotificationSettings(thenReturn result: Result<Void, Error>) {
+        updateNotificationSettingsResult = result
+    }
 }
 
 extension MockAccountRemote {

diff --git a/Yosemite/YosemiteTests/Stores/AccountStoreTests.swift b/Yosemite/YosemiteTests/Stores/AccountStoreTests.swift
@@ -884,6 +884,56 @@ final class AccountStoreTests: XCTestCase {
         XCTAssertTrue(result.isFailure)
         XCTAssertEqual(result.failure as NSError?, error)
     }
+
+    // MARK: - updateNotificationSettings
+
+    func test_updateNotificationSettings_returns_success_upon_success() {
+        // Given
+        let network = MockNetwork()
+        let accountRemote = MockAccountRemote()
+        accountRemote.whenUpdatingNotificationSettings(thenReturn: .success(()))
+        let accountStore = AccountStore(dispatcher: dispatcher,
+                                        storageManager: storageManager,
+                                        network: network,
+                                        remote: accountRemote)
+
+        // When
+        let result: Result<Void, Error> = waitFor { promise in
+            let notificationSettings = NotificationSettings.createSettings(deviceID: 132, enabledSites: [23], disabledSites: [44, 66])
+            let action = AccountAction.updateNotificationSettings(notificationSettings: notificationSettings) { result in
+                promise(result)
+            }
+            accountStore.onAction(action)
+        }
+
+        // Then
+        XCTAssertTrue(result.isSuccess)
+    }
+
+    func test_updateNotificationSettings_relays_error_upon_failure() {
+        // Given
+        let network = MockNetwork()
+        let accountRemote = MockAccountRemote()
+        let error = NSError(domain: "notifications", code: 33)
+        accountRemote.whenUpdatingNotificationSettings(thenReturn: .failure(error))
+        let accountStore = AccountStore(dispatcher: dispatcher,
+                                        storageManager: storageManager,
+                                        network: network,
+                                        remote: accountRemote)
+
+        // When
+        let result: Result<Void, Error> = waitFor { promise in
+            let notificationSettings = NotificationSettings.createSettings(deviceID: 132, enabledSites: [23], disabledSites: [44, 66])
+            let action = AccountAction.updateNotificationSettings(notificationSettings: notificationSettings) { result in
+                promise(result)
+            }
+            accountStore.onAction(action)
+        }
+
+        // Then
+        XCTAssertTrue(result.isFailure)
+        XCTAssertEqual(result.failure as NSError?, error)
+    }
 }
 
 // MARK: - Private Methods