实现权限驱动的首页 Tab 并对齐 Android

2026-07-06 17:27:54 +08:00
parent e04ad8f8f0
commit d79d3003e3
27 changed files with 2573 additions and 10 deletions
--- a/suixinkanTests/HomeViewModelTests.swift
+++ b/suixinkanTests/HomeViewModelTests.swift
@ -0,0 +1,125 @@
+//
+//  HomeViewModelTests.swift
+//  suixinkanTests
+//
+
+import XCTest
+@testable import suixinkan
+
+@MainActor
+/// 首页 ViewModel 测试。
+final class HomeViewModelTests: XCTestCase {
+
+    private var appStore: AppStore!
+
+    override func setUp() {
+        let defaults = UserDefaults(suiteName: "HomeViewModelTests")!
+        defaults.removePersistentDomain(forName: "HomeViewModelTests")
+        appStore = AppStore(defaults: defaults)
+        appStore.userId = "2001"
+        appStore.accountType = "scenic_user"
+        appStore.roleCode = AppRoleCode.photographer.rawValue
+        appStore.roleName = "摄影师"
+        appStore.currentScenicId = 10
+        appStore.currentScenicName = "测试景区"
+    }
+
+    override func tearDown() {
+        appStore.logout()
+        super.tearDown()
+    }
+
+    func testMinimalTopRoleFlag() {
+        appStore.roleCode = AppRoleCode.storeAdmin.rawValue
+        let viewModel = HomeViewModel(appStore: appStore)
+        viewModel.refreshLocalDisplayState()
+        XCTAssertTrue(viewModel.isMinimalTopRole)
+
+        appStore.roleCode = AppRoleCode.photographer.rawValue
+        viewModel.refreshLocalDisplayState()
+        XCTAssertFalse(viewModel.isMinimalTopRole)
+    }
+
+    func testRebuildCommonMenusFiltersUnknownURIs() {
+        appStore.savePermissionItems([
+            HomePermissionItem(id: "1", name: "钱包", uri: "wallet"),
+            HomePermissionItem(id: "2", name: "未知", uri: "unknown_uri"),
+        ])
+        let viewModel = HomeViewModel(appStore: appStore)
+        viewModel.rebuildCommonMenus()
+        XCTAssertEqual(viewModel.commonMenus.map(\.uri), ["wallet"])
+    }
+
+    func testEmptyRolePermissionShowsPermissionDialog() async throws {
+        let responseData = try TestJSON.envelope(data: [RolePermissionResponse]())
+        let session = MockURLSession(responses: [responseData])
+        let api = HomeAPI(client: APIClient(session: session))
+        let viewModel = HomeViewModel(appStore: appStore)
+
+        await viewModel.loadPermissions(api: api, force: true)
+
+        XCTAssertTrue(viewModel.showPermissionDialog)
+        XCTAssertTrue(viewModel.commonMenus.isEmpty)
+    }
+
+    func testRolePermissionBuildsCommonMenusAndStoreCardRole() async throws {
+        appStore.roleCode = AppRoleCode.storeAdmin.rawValue
+        let permissionResponse = RolePermissionResponse(
+            role: RoleInfo(
+                id: 46,
+                name: "店铺管理员",
+                roleCode: AppRoleCode.storeAdmin.rawValue,
+                notes: nil,
+                permission: [
+                    PermissionItem(id: 1, pid: 0, name: "钱包", uri: "wallet", iconSrc: nil, children: nil),
+                    PermissionItem(id: 2, pid: 0, name: "云盘", uri: "cloud_management", iconSrc: nil, children: nil),
+                ]
+            ),
+            scenic: [ScenicInfo(id: 10, name: "测试景区", status: 1)]
+        )
+        let storeResponse = StoreListResponse(
+            total: 1,
+            list: [StoreItem(id: 5, scenicId: 10, name: "测试门店", address: "地址", status: 1, statusText: "营业中")]
+        )
+
+        let permissionJSON = try TestJSON.envelope(data: [permissionResponse])
+        let storeJSON = try TestJSON.envelope(data: storeResponse)
+        let session = MockURLSession(responses: [permissionJSON, storeJSON])
+        let api = HomeAPI(client: APIClient(session: session))
+        let viewModel = HomeViewModel(appStore: appStore)
+
+        await viewModel.loadPermissions(api: api, force: true)
+
+        XCTAssertFalse(viewModel.showPermissionDialog)
+        XCTAssertTrue(viewModel.isMinimalTopRole)
+        XCTAssertEqual(viewModel.commonMenus.count, 2)
+        XCTAssertEqual(viewModel.storeItem?.name, "测试门店")
+    }
+
+    func testMarkNeedsPermissionReload() async throws {
+        let permissionResponse = RolePermissionResponse(
+            role: RoleInfo(
+                id: 41,
+                name: "摄影师",
+                roleCode: AppRoleCode.photographer.rawValue,
+                notes: nil,
+                permission: [
+                    PermissionItem(id: 1, pid: 0, name: "钱包", uri: "wallet", iconSrc: nil, children: nil),
+                ]
+            ),
+            scenic: []
+        )
+        appStore.savePermissionItems([HomePermissionItem(id: "1", name: "钱包", uri: "wallet")])
+
+        let permissionJSON = try TestJSON.envelope(data: [permissionResponse])
+        let session = MockURLSession(responses: [permissionJSON])
+        let api = HomeAPI(client: APIClient(session: session))
+        let viewModel = HomeViewModel(appStore: appStore)
+
+        viewModel.markNeedsPermissionReload()
+        await viewModel.reloadIfNeeded(api: api)
+
+        XCTAssertEqual(viewModel.commonMenus.map(\.uri), ["wallet"])
+        XCTAssertEqual(session.requests.count, 1)
+    }
+}