203 lines
9.2 KiB
Swift
203 lines
9.2 KiB
Swift
//
|
|
// Copyright 2025 Element Creations Ltd.
|
|
// Copyright 2022-2025 New Vector 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
|
|
|
|
@MainActor
|
|
struct RoomChangeRolesScreenViewModelTests {
|
|
var viewModel: RoomChangeRolesScreenViewModelProtocol!
|
|
var roomProxy: JoinedRoomProxyMock!
|
|
|
|
var context: RoomChangeRolesScreenViewModelType.Context {
|
|
viewModel.context
|
|
}
|
|
|
|
@Test
|
|
mutating func initialStateAdministrators() {
|
|
setup(mode: .administrator)
|
|
#expect(context.viewState.membersToPromote == [])
|
|
#expect(context.viewState.membersToDemote == [])
|
|
#expect(context.viewState.administrators == context.viewState.visibleAdministrators)
|
|
#expect(context.viewState.moderators == context.viewState.visibleModerators)
|
|
#expect(context.viewState.users == context.viewState.visibleUsers)
|
|
#expect(context.viewState.membersWithRole.count == 2)
|
|
#expect(context.viewState.membersWithRole.first?.id == RoomMemberProxyMock.mockAdmin.userID)
|
|
#expect(!context.viewState.hasChanges)
|
|
#expect(!context.viewState.isSearching)
|
|
}
|
|
|
|
@Test
|
|
mutating func initialStateModerators() {
|
|
setup(mode: .moderator)
|
|
#expect(context.viewState.membersToPromote == [])
|
|
#expect(context.viewState.membersToDemote == [])
|
|
#expect(context.viewState.administrators == context.viewState.visibleAdministrators)
|
|
#expect(context.viewState.moderators == context.viewState.visibleModerators)
|
|
#expect(context.viewState.users == context.viewState.visibleUsers)
|
|
#expect(context.viewState.membersWithRole.count == 3)
|
|
#expect(context.viewState.membersWithRole.first { $0.id == RoomMemberProxyMock.mockModerator.userID } != nil)
|
|
#expect(!context.viewState.hasChanges)
|
|
#expect(!context.viewState.isSearching)
|
|
}
|
|
|
|
@Test
|
|
mutating func toggleUserOn() throws {
|
|
setup(mode: .moderator)
|
|
let firstUser = try #require(context.viewState.users.first { !context.viewState.isMemberSelected($0) },
|
|
"There should be a regular user available to promote.")
|
|
|
|
context.send(viewAction: .toggleMember(firstUser))
|
|
|
|
#expect(context.viewState.membersToPromote == [firstUser])
|
|
#expect(context.viewState.membersToDemote == [])
|
|
#expect(context.viewState.membersWithRole.count == 4)
|
|
#expect(context.viewState.membersWithRole.contains(firstUser))
|
|
#expect(context.viewState.hasChanges)
|
|
}
|
|
|
|
@Test
|
|
mutating func toggleUserOff() throws {
|
|
try toggleUserOn()
|
|
let firstUser = try #require(context.viewState.membersToPromote.first,
|
|
"There should be a regular user available to promote.")
|
|
|
|
// Then toggle off
|
|
context.send(viewAction: .toggleMember(firstUser))
|
|
|
|
#expect(context.viewState.membersToPromote == [])
|
|
#expect(context.viewState.membersToDemote == [])
|
|
#expect(context.viewState.membersWithRole.count == 3)
|
|
#expect(!context.viewState.membersWithRole.contains(firstUser))
|
|
#expect(!context.viewState.hasChanges)
|
|
}
|
|
|
|
@Test
|
|
mutating func demoteToggledUser() throws {
|
|
try toggleUserOn()
|
|
let firstUser = try #require(context.viewState.membersToPromote.first,
|
|
"There should be a regular user available to promote.")
|
|
|
|
// Then demote
|
|
context.send(viewAction: .demoteMember(firstUser))
|
|
|
|
#expect(context.viewState.membersToPromote == [])
|
|
#expect(context.viewState.membersToDemote == [])
|
|
#expect(context.viewState.membersWithRole.count == 3)
|
|
#expect(!context.viewState.membersWithRole.contains(firstUser))
|
|
#expect(!context.viewState.hasChanges)
|
|
}
|
|
|
|
@Test
|
|
mutating func toggleModeratorOff() throws {
|
|
initialStateModerators()
|
|
let existingModerator = try #require(context.viewState.membersWithRole.first { $0.role == .moderator },
|
|
"There should be a member with the role before we begin.")
|
|
|
|
context.send(viewAction: .toggleMember(existingModerator))
|
|
|
|
#expect(context.viewState.membersToPromote == [])
|
|
#expect(context.viewState.membersToDemote == [existingModerator])
|
|
#expect(context.viewState.membersWithRole.count == 2)
|
|
#expect(!context.viewState.membersWithRole.contains(existingModerator))
|
|
#expect(context.viewState.hasChanges)
|
|
}
|
|
|
|
@Test
|
|
mutating func toggleModeratorOn() throws {
|
|
try toggleModeratorOff()
|
|
let demotedMember = try #require(context.viewState.membersToDemote.first,
|
|
"There should be a member with the role before we begin.")
|
|
|
|
// Then toggle back on
|
|
context.send(viewAction: .toggleMember(demotedMember))
|
|
|
|
#expect(context.viewState.membersToPromote == [])
|
|
#expect(context.viewState.membersToDemote == [])
|
|
#expect(context.viewState.membersWithRole.count == 3)
|
|
#expect(context.viewState.membersWithRole.contains(demotedMember))
|
|
#expect(!context.viewState.hasChanges)
|
|
}
|
|
|
|
@Test
|
|
mutating func demoteModerator() throws {
|
|
initialStateModerators()
|
|
let existingModerator = try #require(context.viewState.membersWithRole.first { $0.role == .moderator },
|
|
"There should be a member with the role before we begin.")
|
|
|
|
context.send(viewAction: .demoteMember(existingModerator))
|
|
|
|
#expect(context.viewState.membersToPromote == [])
|
|
#expect(context.viewState.membersToDemote == [existingModerator])
|
|
#expect(context.viewState.membersWithRole.count == 2)
|
|
#expect(!context.viewState.membersWithRole.contains(existingModerator))
|
|
#expect(context.viewState.hasChanges)
|
|
}
|
|
|
|
@Test
|
|
mutating func saveModeratorChanges() async throws {
|
|
// Given the change roles view model for moderators.
|
|
setup(mode: .moderator)
|
|
|
|
let firstUser = try #require(context.viewState.users.first { !context.viewState.isMemberSelected($0) },
|
|
"There should be a regular user to begin with.")
|
|
let existingModerator = try #require(context.viewState.membersWithRole.first { $0.role == .moderator },
|
|
"There should be a moderator to begin with.")
|
|
|
|
// When promoting a regular user and demoting a moderator.
|
|
context.send(viewAction: .toggleMember(firstUser))
|
|
context.send(viewAction: .toggleMember(existingModerator))
|
|
context.send(viewAction: .save)
|
|
|
|
try await Task.sleep(for: .milliseconds(100))
|
|
|
|
// Then no warning should be shown, and the call to update the users should be made straight away.
|
|
#expect(roomProxy.updatePowerLevelsForUsersCalled)
|
|
#expect(roomProxy.updatePowerLevelsForUsersReceivedUpdates?.count == 2)
|
|
#expect(roomProxy.updatePowerLevelsForUsersReceivedUpdates?.contains { $0.userID == existingModerator.id && $0.powerLevel == 0 } == true)
|
|
#expect(roomProxy.updatePowerLevelsForUsersReceivedUpdates?.contains { $0.userID == firstUser.id && $0.powerLevel == 50 } == true)
|
|
}
|
|
|
|
@Test
|
|
mutating func savePromotedAdministrator() async throws {
|
|
// Given the change roles view model for administrators.
|
|
setup(mode: .administrator)
|
|
#expect(context.alertInfo == nil)
|
|
|
|
let firstUser = try #require(context.viewState.users.first { !context.viewState.isMemberSelected($0) },
|
|
"There should be a regular user to begin with.")
|
|
|
|
// When saving changes to promote a user to an administrator.
|
|
context.send(viewAction: .toggleMember(firstUser))
|
|
context.send(viewAction: .save)
|
|
|
|
// Then an alert should be shown to warn the action cannot be undone.
|
|
#expect(context.alertInfo != nil)
|
|
|
|
// When confirming the prompt
|
|
context.alertInfo?.primaryButton.action?()
|
|
try await Task.sleep(for: .milliseconds(100))
|
|
|
|
// Then the user should be made into an administrator.
|
|
#expect(roomProxy.updatePowerLevelsForUsersCalled)
|
|
#expect(roomProxy.updatePowerLevelsForUsersReceivedUpdates?.count == 1)
|
|
#expect(roomProxy.updatePowerLevelsForUsersReceivedUpdates?.contains { $0.userID == firstUser.id && $0.powerLevel == 100 } == true)
|
|
}
|
|
|
|
// MARK: - Helpers
|
|
|
|
private mutating func setup(mode: RoomRole) {
|
|
roomProxy = JoinedRoomProxyMock(.init(members: .allMembersAsAdmin))
|
|
viewModel = RoomChangeRolesScreenViewModel(mode: mode,
|
|
roomProxy: roomProxy,
|
|
mediaProvider: MediaProviderMock(configuration: .init()),
|
|
userIndicatorController: UserIndicatorControllerMock(),
|
|
analytics: ServiceLocator.shared.analytics)
|
|
}
|
|
}
|