Edit the default notification settings (#1468)

2026-11-02 12:07:55 +00:00 · 2023-08-11 16:10:23 +02:00
parent f413a6b01a
commit 9a52b968f1
23 changed files with 760 additions and 48 deletions
@@ -0,0 +1,203 @@
+//
+// Copyright 2022 New Vector Ltd
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+
+import MatrixRustSDK
+import XCTest
+
+@testable import ElementX
+
+@MainActor
+class NotificationSettingsEditScreenViewModelTests: XCTestCase {
+    private var viewModel: NotificationSettingsEditScreenViewModelProtocol!
+    private var notificationSettingsProxy: NotificationSettingsProxyMock!
+    
+    private var context: NotificationSettingsEditScreenViewModelType.Context {
+        viewModel.context
+    }
+    
+    @MainActor override func setUpWithError() throws {
+        notificationSettingsProxy = NotificationSettingsProxyMock(with: NotificationSettingsProxyMockConfiguration())
+        notificationSettingsProxy.getDefaultRoomNotificationModeIsEncryptedIsOneToOneReturnValue = .allMessages
+    }
+    
+    func testFetchSettings() async throws {
+        notificationSettingsProxy.getDefaultRoomNotificationModeIsEncryptedIsOneToOneClosure = { isEncrypted, isOneToOne in
+            switch (isEncrypted, isOneToOne) {
+            case (_, true):
+                return .allMessages
+            case (_, _):
+                return .mentionsAndKeywordsOnly
+            }
+        }
+        viewModel = NotificationSettingsEditScreenViewModel(isDirect: false,
+                                                            notificationSettingsProxy: notificationSettingsProxy)
+
+        let deferred = deferFulfillment(viewModel.context.$viewState.map(\.defaultMode)
+            .first(where: { !$0.isNil }))
+        viewModel.fetchInitialContent()
+        try await deferred.fulfill()
+        
+        // `getDefaultRoomNotificationModeIsEncryptedIsOneToOne` must have been called twice (for encrypted and unencrypted group chats)
+        let invocations = notificationSettingsProxy.getDefaultRoomNotificationModeIsEncryptedIsOneToOneReceivedInvocations
+        
+        XCTAssertEqual(invocations.count, 2)
+        // First call for encrypted group chats
+        XCTAssertEqual(invocations[0].isEncrypted, true)
+        XCTAssertEqual(invocations[0].isOneToOne, false)
+        // Second call for unencrypted group chats
+        XCTAssertEqual(invocations[1].isEncrypted, false)
+        XCTAssertEqual(invocations[1].isOneToOne, false)
+        
+        XCTAssertEqual(context.viewState.defaultMode, .mentionsAndKeywordsOnly)
+        XCTAssertNil(context.viewState.bindings.alertInfo)
+    }
+    
+    func testSetModeAllMessages() async throws {
+        notificationSettingsProxy.getDefaultRoomNotificationModeIsEncryptedIsOneToOneReturnValue = .mentionsAndKeywordsOnly
+        viewModel = NotificationSettingsEditScreenViewModel(isDirect: false,
+                                                            notificationSettingsProxy: notificationSettingsProxy)
+        let deferred = deferFulfillment(viewModel.context.$viewState.map(\.defaultMode)
+            .first(where: { !$0.isNil }))
+        viewModel.fetchInitialContent()
+        try await deferred.fulfill()
+        
+        // Set mode to .allMessages
+        let deferredViewState = deferFulfillment(context.$viewState
+            .map(\.pendingMode)
+            .removeDuplicates()
+            .collect(3).first())
+        context.send(viewAction: .setMode(.allMessages))
+        let pendingModes = try await deferredViewState.fulfill()
+        
+        XCTAssertEqual(pendingModes, [nil, .allMessages, nil])
+        
+        // `setDefaultRoomNotificationModeIsEncryptedIsOneToOneMode` must have been called twice (for encrypted and unencrypted group chats)
+        let invocations = notificationSettingsProxy.setDefaultRoomNotificationModeIsEncryptedIsOneToOneModeReceivedInvocations
+        XCTAssertEqual(notificationSettingsProxy.setDefaultRoomNotificationModeIsEncryptedIsOneToOneModeCallsCount, 2)
+        // First call for encrypted group chats
+        XCTAssertEqual(invocations[0].isEncrypted, true)
+        XCTAssertEqual(invocations[0].isOneToOne, false)
+        XCTAssertEqual(invocations[0].mode, .allMessages)
+        // Second call for unencrypted group chats
+        XCTAssertEqual(invocations[1].isEncrypted, false)
+        XCTAssertEqual(invocations[1].isOneToOne, false)
+        XCTAssertEqual(invocations[1].mode, .allMessages)
+        
+        // The default mode should be updated
+        let deferredNewViewState = deferFulfillment(context.$viewState
+            .map(\.defaultMode)
+            .first(where: { $0 == .allMessages }))
+        try await deferredNewViewState.fulfill()
+                                                    
+        XCTAssertEqual(context.viewState.defaultMode, .allMessages)
+        XCTAssertNil(context.viewState.bindings.alertInfo)
+    }
+    
+    func testSetModeMentions() async throws {
+        viewModel = NotificationSettingsEditScreenViewModel(isDirect: false,
+                                                            notificationSettingsProxy: notificationSettingsProxy)
+        let deferred = deferFulfillment(viewModel.context.$viewState.map(\.defaultMode)
+            .first(where: { !$0.isNil }))
+        viewModel.fetchInitialContent()
+        try await deferred.fulfill()
+
+        // Set mode to .allMessages
+        let deferredViewState = deferFulfillment(context.$viewState
+            .map(\.pendingMode)
+            .removeDuplicates()
+            .collect(3).first())
+        context.send(viewAction: .setMode(.mentionsAndKeywordsOnly))
+        let pendingModes = try await deferredViewState.fulfill()
+        
+        XCTAssertEqual(pendingModes, [nil, .mentionsAndKeywordsOnly, nil])
+        
+        // `setDefaultRoomNotificationModeIsEncryptedIsOneToOneMode` must have been called twice (for encrypted and unencrypted group chats)
+        let invocations = notificationSettingsProxy.setDefaultRoomNotificationModeIsEncryptedIsOneToOneModeReceivedInvocations
+        XCTAssertEqual(notificationSettingsProxy.setDefaultRoomNotificationModeIsEncryptedIsOneToOneModeCallsCount, 2)
+        // First call for encrypted group chats
+        XCTAssertEqual(invocations[0].isEncrypted, true)
+        XCTAssertEqual(invocations[0].isOneToOne, false)
+        XCTAssertEqual(invocations[0].mode, .mentionsAndKeywordsOnly)
+        // Second call for unencrypted group chats
+        XCTAssertEqual(invocations[1].isEncrypted, false)
+        XCTAssertEqual(invocations[1].isOneToOne, false)
+        XCTAssertEqual(invocations[1].mode, .mentionsAndKeywordsOnly)
+        
+        // The default mode should be updated
+        let deferredNewViewState = deferFulfillment(context.$viewState
+            .map(\.defaultMode)
+            .first(where: { $0 == .mentionsAndKeywordsOnly }))
+        try await deferredNewViewState.fulfill()
+                                                    
+        XCTAssertEqual(context.viewState.defaultMode, .mentionsAndKeywordsOnly)
+        XCTAssertNil(context.viewState.bindings.alertInfo)
+    }
+    
+    func testSetModeDirectChats() async throws {
+        notificationSettingsProxy.getDefaultRoomNotificationModeIsEncryptedIsOneToOneReturnValue = .mentionsAndKeywordsOnly
+        // Initialize for direct chats
+        viewModel = NotificationSettingsEditScreenViewModel(isDirect: true,
+                                                            notificationSettingsProxy: notificationSettingsProxy)
+        let deferred = deferFulfillment(viewModel.context.$viewState.map(\.defaultMode)
+            .first(where: { !$0.isNil }))
+        viewModel.fetchInitialContent()
+        try await deferred.fulfill()
+
+        // Set mode to .allMessages
+        let deferredViewState = deferFulfillment(context.$viewState
+            .map(\.pendingMode)
+            .removeDuplicates()
+            .collect(3).first())
+        context.send(viewAction: .setMode(.allMessages))
+        let pendingModes = try await deferredViewState.fulfill()
+        
+        XCTAssertEqual(pendingModes, [nil, .allMessages, nil])
+        
+        // `setDefaultRoomNotificationModeIsEncryptedIsOneToOneMode` must have been called twice (for encrypted and unencrypted direct chats)
+        let invocations = notificationSettingsProxy.setDefaultRoomNotificationModeIsEncryptedIsOneToOneModeReceivedInvocations
+        XCTAssertEqual(notificationSettingsProxy.setDefaultRoomNotificationModeIsEncryptedIsOneToOneModeCallsCount, 2)
+        // First call for encrypted direct chats
+        XCTAssertEqual(invocations[0].isEncrypted, true)
+        XCTAssertEqual(invocations[0].isOneToOne, true)
+        XCTAssertEqual(invocations[0].mode, .allMessages)
+        // Second call for unencrypted direct chats
+        XCTAssertEqual(invocations[1].isEncrypted, false)
+        XCTAssertEqual(invocations[1].isOneToOne, true)
+        XCTAssertEqual(invocations[1].mode, .allMessages)
+    }
+    
+    func testSetModeFailure() async throws {
+        notificationSettingsProxy.getDefaultRoomNotificationModeIsEncryptedIsOneToOneReturnValue = .mentionsAndKeywordsOnly
+        notificationSettingsProxy.setDefaultRoomNotificationModeIsEncryptedIsOneToOneModeThrowableError = NotificationSettingsError.Generic(message: "error")
+        viewModel = NotificationSettingsEditScreenViewModel(isDirect: true,
+                                                            notificationSettingsProxy: notificationSettingsProxy)
+        let deferred = deferFulfillment(viewModel.context.$viewState.map(\.defaultMode)
+            .first(where: { !$0.isNil }))
+        viewModel.fetchInitialContent()
+        try await deferred.fulfill()
+
+        // Set mode to .allMessages
+        let deferredViewState = deferFulfillment(context.$viewState
+            .map(\.pendingMode)
+            .removeDuplicates()
+            .collect(3).first())
+        context.send(viewAction: .setMode(.allMessages))
+        let pendingModes = try await deferredViewState.fulfill()
+        
+        XCTAssertEqual(pendingModes, [nil, .allMessages, nil])
+        XCTAssertNotNil(context.viewState.bindings.alertInfo)
+    }
+}