98 lines
4.0 KiB
Swift
98 lines
4.0 KiB
Swift
//
|
|
// Copyright 2023, 2024 New Vector Ltd.
|
|
//
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
// Please see LICENSE in the repository root for full details.
|
|
//
|
|
|
|
import Combine
|
|
import XCTest
|
|
@testable import Element
|
|
|
|
final class PushRulesUpdaterTests: XCTestCase {
|
|
private var notificationService: MockNotificationSettingsService!
|
|
private var pushRulesUpdater: PushRulesUpdater!
|
|
|
|
override func setUpWithError() throws {
|
|
notificationService = .init()
|
|
notificationService.rules = [MockNotificationPushRule].default
|
|
pushRulesUpdater = .init(notificationSettingsService: notificationService)
|
|
}
|
|
|
|
func testNoRuleIsUpdated() async throws {
|
|
await pushRulesUpdater.syncRulesIfNeeded()
|
|
XCTAssertEqual(notificationService.rules as? [MockNotificationPushRule], [MockNotificationPushRule].default)
|
|
}
|
|
|
|
func testSingleRuleAffected() async throws {
|
|
let targetActions: NotificationActions = .init(notify: true, sound: "default")
|
|
let targetRuleIndex = try mockRule(ruleId: .pollStart, enabled: false, actions: targetActions)
|
|
|
|
await pushRulesUpdater.syncRulesIfNeeded()
|
|
|
|
XCTAssertEqual(self.notificationService.rules[targetRuleIndex].ruleActions, NotificationStandardActions.notifyDefaultSound.actions)
|
|
XCTAssertTrue(self.notificationService.rules[targetRuleIndex].enabled)
|
|
}
|
|
|
|
func testAffectedRulesAreUpdated() async throws {
|
|
let targetActions: NotificationActions = .init(notify: true, sound: "abc")
|
|
try mockRule(ruleId: .allOtherMessages, enabled: true, actions: targetActions)
|
|
let affectedRules: [NotificationPushRuleId] = [.allOtherMessages, .pollStart, .msc3930pollStart, .pollEnd, .msc3930pollEnd]
|
|
|
|
await pushRulesUpdater.syncRulesIfNeeded()
|
|
|
|
for rule in self.notificationService.rules {
|
|
guard let id = rule.pushRuleId else {
|
|
continue
|
|
}
|
|
|
|
if affectedRules.contains(id) {
|
|
XCTAssertEqual(rule.ruleActions, targetActions)
|
|
} else {
|
|
XCTAssertEqual(rule.ruleActions, NotificationStandardActions.notifyDefaultSound.actions)
|
|
}
|
|
}
|
|
}
|
|
|
|
func testAffectedOneToOneRulesAreUpdated() async throws {
|
|
let targetActions: NotificationActions = .init(notify: true, sound: "abc")
|
|
try mockRule(ruleId: .oneToOneRoom, enabled: true, actions: targetActions)
|
|
let affectedRules: [NotificationPushRuleId] = [.oneToOneRoom, .oneToOnePollStart, .msc3930oneToOnePollStart, .oneToOnePollEnd, .msc3930oneToOnePollEnd]
|
|
|
|
await pushRulesUpdater.syncRulesIfNeeded()
|
|
|
|
for rule in self.notificationService.rules {
|
|
guard let id = rule.pushRuleId else {
|
|
continue
|
|
}
|
|
|
|
if affectedRules.contains(id) {
|
|
XCTAssertEqual(rule.ruleActions, targetActions)
|
|
} else {
|
|
XCTAssertEqual(rule.ruleActions, NotificationStandardActions.notifyDefaultSound.actions)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private extension PushRulesUpdaterTests {
|
|
@discardableResult
|
|
func mockRule(ruleId: NotificationPushRuleId, enabled: Bool, actions: NotificationActions) throws -> Int {
|
|
guard let ruleIndex = notificationService.rules.firstIndex(where: { $0.pushRuleId == ruleId }) else {
|
|
throw NSError(domain: "no ruleIndex found", code: 0)
|
|
}
|
|
notificationService.rules[ruleIndex] = MockNotificationPushRule(ruleId: ruleId.rawValue, enabled: enabled, ruleActions: actions)
|
|
return ruleIndex
|
|
}
|
|
}
|
|
|
|
private extension Array where Element == MockNotificationPushRule {
|
|
static var `default`: [MockNotificationPushRule] {
|
|
let ids: [NotificationPushRuleId] = [.oneToOneRoom, .allOtherMessages, .pollStart, .msc3930pollStart, .pollEnd, .msc3930pollEnd, .oneToOnePollStart, .msc3930oneToOnePollStart, .oneToOnePollEnd, .msc3930oneToOnePollEnd]
|
|
|
|
return ids.map {
|
|
MockNotificationPushRule(ruleId: $0.rawValue, enabled: true)
|
|
}
|
|
}
|
|
}
|