110 lines
4.4 KiB
Swift
110 lines
4.4 KiB
Swift
//
|
|
// 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 Combine
|
|
import XCTest
|
|
|
|
@testable import ElementX
|
|
|
|
@MainActor
|
|
class InviteUsersScreenViewModelTests: XCTestCase {
|
|
var viewModel: InviteUsersScreenViewModelProtocol!
|
|
var clientProxy: MockClientProxy!
|
|
var userDiscoveryService: UserDiscoveryServiceMock!
|
|
|
|
private var cancellables: Set<AnyCancellable> = []
|
|
|
|
var context: InviteUsersScreenViewModel.Context {
|
|
viewModel.context
|
|
}
|
|
|
|
func testSelectUser() {
|
|
setupWithRoomType(roomType: .draft)
|
|
XCTAssertTrue(context.viewState.selectedUsers.isEmpty)
|
|
context.send(viewAction: .toggleUser(.mockAlice))
|
|
XCTAssertTrue(context.viewState.selectedUsers.count == 1)
|
|
XCTAssertEqual(context.viewState.selectedUsers.first?.userID, UserProfile.mockAlice.userID)
|
|
}
|
|
|
|
func testReselectUser() {
|
|
setupWithRoomType(roomType: .draft)
|
|
XCTAssertTrue(context.viewState.selectedUsers.isEmpty)
|
|
context.send(viewAction: .toggleUser(.mockAlice))
|
|
XCTAssertEqual(context.viewState.selectedUsers.count, 1)
|
|
XCTAssertEqual(context.viewState.selectedUsers.first?.userID, UserProfile.mockAlice.userID)
|
|
context.send(viewAction: .toggleUser(.mockAlice))
|
|
XCTAssertTrue(context.viewState.selectedUsers.isEmpty)
|
|
}
|
|
|
|
func testDeselectUser() {
|
|
setupWithRoomType(roomType: .draft)
|
|
XCTAssertTrue(context.viewState.selectedUsers.isEmpty)
|
|
context.send(viewAction: .toggleUser(.mockAlice))
|
|
XCTAssertEqual(context.viewState.selectedUsers.count, 1)
|
|
XCTAssertEqual(context.viewState.selectedUsers.first?.userID, UserProfile.mockAlice.userID)
|
|
context.send(viewAction: .toggleUser(.mockAlice))
|
|
XCTAssertTrue(context.viewState.selectedUsers.isEmpty)
|
|
}
|
|
|
|
func testInviteButton() async {
|
|
let mockedMembers: [RoomMemberProxyMock] = [.mockAlice, .mockBob]
|
|
setupWithRoomType(roomType: .room(members: mockedMembers, userIndicatorController: MockUserIndicatorController()))
|
|
_ = await viewModel.context.$viewState.values.first(where: { $0.membershipState.isEmpty == false })
|
|
context.send(viewAction: .toggleUser(.mockAlice))
|
|
|
|
Task.detached(priority: .low) {
|
|
await self.context.send(viewAction: .proceed)
|
|
}
|
|
|
|
let action = await viewModel.actions.values.first()
|
|
|
|
guard case let .invite(members) = action else {
|
|
XCTFail("Sent action should be 'invite'")
|
|
return
|
|
}
|
|
|
|
XCTAssertEqual(members, [RoomMemberProxyMock.mockAlice.userID])
|
|
}
|
|
|
|
private func setupWithRoomType(roomType: InviteUsersScreenRoomType) {
|
|
let usersSubject = CurrentValueSubject<[UserProfile], Never>([])
|
|
clientProxy = .init(userID: "")
|
|
userDiscoveryService = UserDiscoveryServiceMock()
|
|
userDiscoveryService.fetchSuggestionsReturnValue = .success([])
|
|
userDiscoveryService.searchProfilesWithReturnValue = .success([])
|
|
usersSubject.send([])
|
|
let viewModel = InviteUsersScreenViewModel(selectedUsers: usersSubject.asCurrentValuePublisher(), roomType: roomType, mediaProvider: MockMediaProvider(), userDiscoveryService: userDiscoveryService)
|
|
viewModel.state.usersSection = .init(type: .suggestions, users: [.mockAlice, .mockBob, .mockCharlie])
|
|
self.viewModel = viewModel
|
|
|
|
viewModel.actions.sink { action in
|
|
switch action {
|
|
case .toggleUser(let user):
|
|
var selectedUsers = usersSubject.value
|
|
if let index = selectedUsers.firstIndex(where: { $0.userID == user.userID }) {
|
|
selectedUsers.remove(at: index)
|
|
} else {
|
|
selectedUsers.append(user)
|
|
}
|
|
usersSubject.send(selectedUsers)
|
|
default:
|
|
break
|
|
}
|
|
}
|
|
.store(in: &cancellables)
|
|
}
|
|
}
|