suixinkan_ios_new/suixinkanTests/AccountContextTests.swift

//
//  AccountContextTests.swift
//  suixinkanTests
//
//  Created by Codex on 2026/6/20.
//

import XCTest
@testable import suixinkan

@MainActor
/// 账号上下文测试，覆盖景区、门店和角色权限的选择恢复。
final class AccountContextTests: XCTestCase {
    /// 测试账号缓存前缀与 Android 的账号作用域规则一致。
    func testAccountCachePrefixUsesAccountTypeAndUserId() {
        let context = AccountContext()

        context.applyLogin(
            profile: AccountProfile(userId: "101", displayName: "测试账号"),
            accountType: "store_user"
        )

        XCTAssertEqual(context.accountCachePrefix, "store_user_101")
    }

    /// 测试 replaceScopes 能按缓存 ID 恢复当前景区和门店。
    func testReplaceScopesRestoresCurrentSelectionsById() {
        let context = AccountContext()
        let scenicScopes = [
            BusinessScope(id: 1, name: "默认景区", kind: .scenic),
            BusinessScope(id: 2, name: "缓存景区", kind: .scenic)
        ]
        let storeScopes = [
            BusinessScope(id: 10, name: "默认门店", kind: .store),
            BusinessScope(id: 20, name: "缓存门店", kind: .store)
        ]

        context.replaceScopes(
            scenic: scenicScopes,
            stores: storeScopes,
            currentScenicId: 2,
            currentStoreId: 20
        )

        XCTAssertEqual(context.currentScenic?.id, 2)
        XCTAssertEqual(context.currentStore?.id, 20)
    }

    /// 测试显式切换景区账号时会清空旧门店，而不是保留上一次门店选择。
    func testReplaceScopesExplicitSelectionClearsStoreForScenicAccount() {
        let context = AccountContext()
        context.replaceScopes(
            scenic: [BusinessScope(id: 128, name: "伊犁那拉提景区-5A", kind: .scenic)],
            stores: [BusinessScope(id: 37, name: "随心旅拍", kind: .store, parentScenicId: 128)],
            currentScenicId: 128,
            currentStoreId: 37,
            explicitSelection: true
        )
        XCTAssertEqual(context.currentStore?.name, "随心旅拍")

        context.applyLogin(accountType: V9ScenicUser.accountTypeValue, businessUserId: 41, displayTitle: "伊犁那拉提景区-5A")
        context.replaceScopes(
            scenic: [BusinessScope(id: 128, name: "伊犁那拉提景区-5A", kind: .scenic)],
            stores: [BusinessScope(id: 37, name: "随心旅拍", kind: .store, parentScenicId: 128)],
            currentScenicId: 128,
            currentStoreId: nil,
            explicitSelection: true
        )

        XCTAssertEqual(context.accountType, V9ScenicUser.accountTypeValue)
        XCTAssertNil(context.currentStore)
        XCTAssertEqual(context.currentAccountDisplayTitle, "伊犁那拉提景区-5A")
    }

    /// 测试缓存 ID 不存在时会回退到首个可用作用域。
    func testReplaceScopesFallsBackToFirstScopeWhenCachedIdIsMissing() {
        let context = AccountContext()

        context.replaceScopes(
            scenic: [BusinessScope(id: 1, name: "默认景区", kind: .scenic)],
            stores: [BusinessScope(id: 10, name: "默认门店", kind: .store)],
            currentScenicId: 999,
            currentStoreId: 999
        )

        XCTAssertEqual(context.currentScenic?.id, 1)
        XCTAssertEqual(context.currentStore?.id, 10)
    }

    /// 测试切换景区时会优先选择同景区门店。
    func testSelectScenicResolvesStoreWithinSelectedScenic() {
        let context = AccountContext()
        context.replaceScopes(
            scenic: [
                BusinessScope(id: 1, name: "西湖景区", kind: .scenic),
                BusinessScope(id: 2, name: "东湖景区", kind: .scenic)
            ],
            stores: [
                BusinessScope(id: 10, name: "西湖门店", kind: .store, parentScenicId: 1),
                BusinessScope(id: 20, name: "东湖门店", kind: .store, parentScenicId: 2)
            ],
            currentScenicId: 1,
            currentStoreId: 10
        )

        context.selectScenic(id: 2)

        XCTAssertEqual(context.currentScenic?.id, 2)
        XCTAssertEqual(context.currentStore?.id, 20)
    }

    /// 测试切换角色时会同步角色景区并重新匹配门店。
    func testSelectRoleUpdatesScenicScopesAndStoreSelection() {
        let accountContext = AccountContext()
        let permissionContext = PermissionContext()
        accountContext.replaceScopes(
            scenic: [BusinessScope(id: 1, name: "旧景区", kind: .scenic)],
            stores: [
                BusinessScope(id: 10, name: "旧门店", kind: .store, parentScenicId: 1),
                BusinessScope(id: 20, name: "新门店", kind: .store, parentScenicId: 2)
            ],
            currentScenicId: 1,
            currentStoreId: 10
        )
        permissionContext.replaceRolePermissions([
            RolePermissionResponse(
                role: RoleInfo(id: 1, name: "旧角色"),
                scenic: [ScenicInfo(id: 1, name: "旧景区")]
            ),
            RolePermissionResponse(
                role: RoleInfo(id: 2, name: "新角色"),
                scenic: [ScenicInfo(id: 2, name: "新景区")]
            )
        ])

        permissionContext.selectRole(id: 2, accountContext: accountContext)

        XCTAssertEqual(permissionContext.currentRole?.id, 2)
        XCTAssertEqual(accountContext.scenicScopes.map(\.id), [2])
        XCTAssertEqual(accountContext.currentScenic?.id, 2)
        XCTAssertEqual(accountContext.currentStore?.id, 20)
    }

    /// 测试权限上下文会递归展开权限 URI。
    func testPermissionContextFlattensPermissionURIs() {
        let permissionContext = PermissionContext()
        permissionContext.replaceRolePermissions([
            RolePermissionResponse(
                role: RoleInfo(
                    id: 1,
                    name: "运营",
                    permission: [
                        PermissionItem(
                            id: 1,
                            name: "首页",
                            uri: "/home",
                            children: [
                                PermissionItem(id: 2, name: "订单", uri: "/orders")
                            ]
                        )
                    ]
                ),
                scenic: []
            )
        ])

        XCTAssertTrue(permissionContext.canAccess("/home"))
        XCTAssertTrue(permissionContext.canAccess("/orders"))
        XCTAssertFalse(permissionContext.canAccess("/settings"))
    }
}