124 lines
4.7 KiB
Swift
124 lines
4.7 KiB
Swift
//
|
|
// AllFunctionsViewModelTests.swift
|
|
// suixinkanTests
|
|
//
|
|
|
|
import XCTest
|
|
@testable import suixinkan
|
|
|
|
/// 全部功能 ViewModel 测试。
|
|
final class AllFunctionsViewModelTests: XCTestCase {
|
|
|
|
private var defaults: UserDefaults!
|
|
private var appStore: AppStore!
|
|
private var menuStore: HomeCommonMenuStore!
|
|
|
|
override func setUp() {
|
|
defaults = UserDefaults(suiteName: "AllFunctionsViewModelTests")!
|
|
defaults.removePersistentDomain(forName: "AllFunctionsViewModelTests")
|
|
appStore = AppStore(defaults: defaults)
|
|
menuStore = HomeCommonMenuStore(defaults: defaults)
|
|
appStore.userId = "1001"
|
|
appStore.accountType = "scenic_user"
|
|
appStore.roleCode = AppRoleCode.photographer.rawValue
|
|
}
|
|
|
|
override func tearDown() {
|
|
defaults.removePersistentDomain(forName: "AllFunctionsViewModelTests")
|
|
super.tearDown()
|
|
}
|
|
|
|
func testLoadFunctionsUsesDefaultFirstFourWhenNoSavedURIs() {
|
|
appStore.savePermissionItems([
|
|
HomePermissionItem(id: "1", name: "钱包", uri: "wallet"),
|
|
HomePermissionItem(id: "2", name: "云盘", uri: "cloud_management"),
|
|
HomePermissionItem(id: "3", name: "任务", uri: "task_management"),
|
|
HomePermissionItem(id: "4", name: "设置", uri: "system_settings"),
|
|
HomePermissionItem(id: "5", name: "消息", uri: "message_center"),
|
|
])
|
|
let viewModel = AllFunctionsViewModel(appStore: appStore, commonMenuStore: menuStore)
|
|
|
|
viewModel.loadFunctions()
|
|
|
|
XCTAssertEqual(viewModel.commonMenus.map(\.uri), [
|
|
"wallet", "cloud_management", "task_management", "system_settings",
|
|
])
|
|
XCTAssertEqual(viewModel.moreMenus.map(\.uri), ["message_center"])
|
|
XCTAssertEqual(
|
|
menuStore.savedCommonURIs(
|
|
accountScope: appStore.accountCachePrefix,
|
|
roleCode: appStore.roleCode
|
|
),
|
|
["wallet", "cloud_management", "task_management", "system_settings"]
|
|
)
|
|
}
|
|
|
|
func testLoadFunctionsUsesSavedURIsOverDefault() {
|
|
appStore.savePermissionItems([
|
|
HomePermissionItem(id: "1", name: "钱包", uri: "wallet"),
|
|
HomePermissionItem(id: "2", name: "云盘", uri: "cloud_management"),
|
|
HomePermissionItem(id: "3", name: "任务", uri: "task_management"),
|
|
])
|
|
menuStore.saveCommonURIs(
|
|
["cloud_management"],
|
|
accountScope: appStore.accountCachePrefix,
|
|
roleCode: appStore.roleCode
|
|
)
|
|
let viewModel = AllFunctionsViewModel(appStore: appStore, commonMenuStore: menuStore)
|
|
|
|
viewModel.loadFunctions()
|
|
|
|
XCTAssertEqual(viewModel.commonMenus.map(\.uri), ["cloud_management"])
|
|
XCTAssertEqual(Set(viewModel.moreMenus.map(\.uri)), Set(["wallet", "task_management"]))
|
|
}
|
|
|
|
func testAddToCommonMovesMenuAndPersists() {
|
|
appStore.savePermissionItems([
|
|
HomePermissionItem(id: "1", name: "钱包", uri: "wallet"),
|
|
HomePermissionItem(id: "2", name: "云盘", uri: "cloud_management"),
|
|
])
|
|
menuStore.saveCommonURIs(
|
|
["wallet"],
|
|
accountScope: appStore.accountCachePrefix,
|
|
roleCode: appStore.roleCode
|
|
)
|
|
let viewModel = AllFunctionsViewModel(appStore: appStore, commonMenuStore: menuStore)
|
|
viewModel.loadFunctions()
|
|
let cloudMenu = viewModel.moreMenus.first { $0.uri == "cloud_management" }!
|
|
|
|
viewModel.addToCommon(cloudMenu)
|
|
|
|
XCTAssertEqual(viewModel.commonMenus.map(\.uri), ["wallet", "cloud_management"])
|
|
XCTAssertTrue(viewModel.moreMenus.isEmpty)
|
|
XCTAssertTrue(viewModel.didCustomize)
|
|
XCTAssertEqual(
|
|
menuStore.savedCommonURIs(
|
|
accountScope: appStore.accountCachePrefix,
|
|
roleCode: appStore.roleCode
|
|
),
|
|
["wallet", "cloud_management"]
|
|
)
|
|
}
|
|
|
|
func testRemoveFromCommonMovesMenuAndPersists() {
|
|
appStore.savePermissionItems([
|
|
HomePermissionItem(id: "1", name: "钱包", uri: "wallet"),
|
|
HomePermissionItem(id: "2", name: "云盘", uri: "cloud_management"),
|
|
])
|
|
menuStore.saveCommonURIs(
|
|
["wallet", "cloud_management"],
|
|
accountScope: appStore.accountCachePrefix,
|
|
roleCode: appStore.roleCode
|
|
)
|
|
let viewModel = AllFunctionsViewModel(appStore: appStore, commonMenuStore: menuStore)
|
|
viewModel.loadFunctions()
|
|
let walletMenu = viewModel.commonMenus.first { $0.uri == "wallet" }!
|
|
|
|
viewModel.removeFromCommon(walletMenu)
|
|
|
|
XCTAssertEqual(viewModel.commonMenus.map(\.uri), ["cloud_management"])
|
|
XCTAssertEqual(viewModel.moreMenus.map(\.uri), ["wallet"])
|
|
XCTAssertTrue(viewModel.didCustomize)
|
|
}
|
|
}
|