feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf
This commit is contained in:
fanlv
2025-07-20 17:36:12 +08:00
commit 890153324f
14811 changed files with 1923430 additions and 0 deletions

View File

@@ -0,0 +1,238 @@
/*
* Copyright 2025 coze-dev Authors
*
* 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 { describe, it, expect } from 'vitest';
import { SpaceRoleType } from '@coze-arch/idl/developer_api';
import { ESpacePermisson } from '../../src/space/constants';
import { calcPermission } from '../../src/space/calc-permission';
describe('Space Calc Permission', () => {
describe('calcPermission', () => {
it('应该为 Owner 角色返回正确的权限', () => {
// Owner 应该有更新空间的权限
expect(
calcPermission(ESpacePermisson.UpdateSpace, [SpaceRoleType.Owner]),
).toBe(true);
// Owner 应该有删除空间的权限
expect(
calcPermission(ESpacePermisson.DeleteSpace, [SpaceRoleType.Owner]),
).toBe(true);
// Owner 应该有添加成员的权限
expect(
calcPermission(ESpacePermisson.AddBotSpaceMember, [
SpaceRoleType.Owner,
]),
).toBe(true);
// Owner 应该有移除成员的权限
expect(
calcPermission(ESpacePermisson.RemoveSpaceMember, [
SpaceRoleType.Owner,
]),
).toBe(true);
// Owner 应该有转移所有权的权限
expect(
calcPermission(ESpacePermisson.TransferSpace, [SpaceRoleType.Owner]),
).toBe(true);
// Owner 应该有更新成员的权限
expect(
calcPermission(ESpacePermisson.UpdateSpaceMember, [
SpaceRoleType.Owner,
]),
).toBe(true);
// Owner 应该有管理 API 的权限
expect(calcPermission(ESpacePermisson.API, [SpaceRoleType.Owner])).toBe(
true,
);
});
it('应该为 Admin 角色返回正确的权限', () => {
// Admin 应该有添加成员的权限
expect(
calcPermission(ESpacePermisson.AddBotSpaceMember, [
SpaceRoleType.Admin,
]),
).toBe(true);
// Admin 应该有移除成员的权限
expect(
calcPermission(ESpacePermisson.RemoveSpaceMember, [
SpaceRoleType.Admin,
]),
).toBe(true);
// Admin 应该有退出空间的权限
expect(
calcPermission(ESpacePermisson.ExitSpace, [SpaceRoleType.Admin]),
).toBe(true);
// Admin 应该有更新成员的权限
expect(
calcPermission(ESpacePermisson.UpdateSpaceMember, [
SpaceRoleType.Admin,
]),
).toBe(true);
// Admin 不应该有更新空间的权限
expect(
calcPermission(ESpacePermisson.UpdateSpace, [SpaceRoleType.Admin]),
).toBe(false);
// Admin 不应该有删除空间的权限
expect(
calcPermission(ESpacePermisson.DeleteSpace, [SpaceRoleType.Admin]),
).toBe(false);
// Admin 不应该有转移所有权的权限
expect(
calcPermission(ESpacePermisson.TransferSpace, [SpaceRoleType.Admin]),
).toBe(false);
// Admin 不应该有管理 API 的权限
expect(calcPermission(ESpacePermisson.API, [SpaceRoleType.Admin])).toBe(
false,
);
});
it('应该为 Member 角色返回正确的权限', () => {
// Member 应该有退出空间的权限
expect(
calcPermission(ESpacePermisson.ExitSpace, [SpaceRoleType.Member]),
).toBe(true);
// Member 不应该有更新空间的权限
expect(
calcPermission(ESpacePermisson.UpdateSpace, [SpaceRoleType.Member]),
).toBe(false);
// Member 不应该有删除空间的权限
expect(
calcPermission(ESpacePermisson.DeleteSpace, [SpaceRoleType.Member]),
).toBe(false);
// Member 不应该有添加成员的权限
expect(
calcPermission(ESpacePermisson.AddBotSpaceMember, [
SpaceRoleType.Member,
]),
).toBe(false);
// Member 不应该有移除成员的权限
expect(
calcPermission(ESpacePermisson.RemoveSpaceMember, [
SpaceRoleType.Member,
]),
).toBe(false);
// Member 不应该有转移所有权的权限
expect(
calcPermission(ESpacePermisson.TransferSpace, [SpaceRoleType.Member]),
).toBe(false);
// Member 不应该有更新成员的权限
expect(
calcPermission(ESpacePermisson.UpdateSpaceMember, [
SpaceRoleType.Member,
]),
).toBe(false);
// Member 不应该有管理 API 的权限
expect(calcPermission(ESpacePermisson.API, [SpaceRoleType.Member])).toBe(
false,
);
});
it('应该为 Default 角色返回正确的权限', () => {
// Default 不应该有任何权限
expect(
calcPermission(ESpacePermisson.UpdateSpace, [SpaceRoleType.Default]),
).toBe(false);
expect(
calcPermission(ESpacePermisson.DeleteSpace, [SpaceRoleType.Default]),
).toBe(false);
expect(
calcPermission(ESpacePermisson.AddBotSpaceMember, [
SpaceRoleType.Default,
]),
).toBe(false);
expect(
calcPermission(ESpacePermisson.RemoveSpaceMember, [
SpaceRoleType.Default,
]),
).toBe(false);
expect(
calcPermission(ESpacePermisson.ExitSpace, [SpaceRoleType.Default]),
).toBe(false);
expect(
calcPermission(ESpacePermisson.TransferSpace, [SpaceRoleType.Default]),
).toBe(false);
expect(
calcPermission(ESpacePermisson.UpdateSpaceMember, [
SpaceRoleType.Default,
]),
).toBe(false);
expect(calcPermission(ESpacePermisson.API, [SpaceRoleType.Default])).toBe(
false,
);
});
it('应该处理多个角色的情况', () => {
// 当用户同时拥有 Member 和 Admin 角色时,应该有两个角色的所有权限
expect(
calcPermission(ESpacePermisson.ExitSpace, [
SpaceRoleType.Member,
SpaceRoleType.Admin,
]),
).toBe(true);
expect(
calcPermission(ESpacePermisson.RemoveSpaceMember, [
SpaceRoleType.Member,
SpaceRoleType.Admin,
]),
).toBe(true);
// 即使其中一个角色没有权限,只要有一个角色有权限,就应该返回 true
expect(
calcPermission(ESpacePermisson.UpdateSpace, [
SpaceRoleType.Member,
SpaceRoleType.Owner,
]),
).toBe(true);
});
it('应该处理空角色数组', () => {
// 当没有角色时,应该返回 false
expect(calcPermission(ESpacePermisson.UpdateSpace, [])).toBe(false);
expect(calcPermission(ESpacePermisson.ExitSpace, [])).toBe(false);
});
it('应该处理未知角色', () => {
// 当角色未知时,应该返回 false
expect(
calcPermission(ESpacePermisson.UpdateSpace, [
'UnknownRole' as unknown as SpaceRoleType,
]),
).toBe(false);
});
});
});

View File

@@ -0,0 +1,68 @@
/*
* Copyright 2025 coze-dev Authors
*
* 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 { describe, it, expect } from 'vitest';
import { SpaceRoleType } from '@coze-arch/idl/developer_api';
import { ESpacePermisson } from '../../src/space/constants';
describe('Space Constants', () => {
describe('ESpacePermisson', () => {
it('应该定义所有必要的权限点', () => {
// 验证所有权限点都已定义
expect(ESpacePermisson.UpdateSpace).toBeDefined();
expect(ESpacePermisson.DeleteSpace).toBeDefined();
expect(ESpacePermisson.AddBotSpaceMember).toBeDefined();
expect(ESpacePermisson.RemoveSpaceMember).toBeDefined();
expect(ESpacePermisson.ExitSpace).toBeDefined();
expect(ESpacePermisson.TransferSpace).toBeDefined();
expect(ESpacePermisson.UpdateSpaceMember).toBeDefined();
expect(ESpacePermisson.API).toBeDefined();
});
it('应该为每个权限点分配唯一的值', () => {
// 创建一个集合来存储所有权限点的值
const permissionValues = new Set();
// 获取所有权限点的值
Object.values(ESpacePermisson)
.filter(value => typeof value === 'number')
.forEach(value => {
permissionValues.add(value);
});
// 验证权限点的数量与唯一值的数量相同
const numericKeys = Object.keys(ESpacePermisson).filter(
key => !isNaN(Number(key)),
).length;
expect(permissionValues.size).toBe(numericKeys);
});
});
describe('SpaceRoleType', () => {
it('应该正确导出 SpaceRoleType', () => {
// 验证 SpaceRoleType 已正确导出
expect(SpaceRoleType).toBeDefined();
// 验证 SpaceRoleType 包含必要的角色
expect(SpaceRoleType.Owner).toBeDefined();
expect(SpaceRoleType.Admin).toBeDefined();
expect(SpaceRoleType.Member).toBeDefined();
expect(SpaceRoleType.Default).toBeDefined();
});
});
});

View File

@@ -0,0 +1,67 @@
/*
* Copyright 2025 coze-dev Authors
*
* 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 { SpaceRoleType } from '@coze-arch/idl/developer_api';
import { ESpacePermisson } from '../../src/space/constants';
import { calcPermission } from '../../src/space/calc-permission';
describe('calcPermission', () => {
it('should return true for Owner role with UpdateSpace permission', () => {
expect(
calcPermission(ESpacePermisson.UpdateSpace, [SpaceRoleType.Owner]),
).toBe(true);
});
it('should return true for Admin role with RemoveSpaceMember permission', () => {
expect(
calcPermission(ESpacePermisson.RemoveSpaceMember, [SpaceRoleType.Admin]),
).toBe(true);
});
it('should return true for Member role with ExitSpace permission', () => {
expect(
calcPermission(ESpacePermisson.ExitSpace, [SpaceRoleType.Member]),
).toBe(true);
});
it('should return false for Member role with UpdateSpace permission', () => {
expect(
calcPermission(ESpacePermisson.UpdateSpace, [SpaceRoleType.Member]),
).toBe(false);
});
it('should return true for multiple roles with overlapping permissions', () => {
expect(
calcPermission(ESpacePermisson.ExitSpace, [
SpaceRoleType.Admin,
SpaceRoleType.Member,
]),
).toBe(true);
});
it('should return false for unknown role', () => {
expect(
calcPermission(ESpacePermisson.UpdateSpace, [
'UnknownRole' as unknown as SpaceRoleType,
]),
).toBe(false);
});
it('should return false for no roles', () => {
expect(calcPermission(ESpacePermisson.UpdateSpace, [])).toBe(false);
});
});

View File

@@ -0,0 +1,208 @@
/*
* Copyright 2025 coze-dev Authors
*
* 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 { describe, it, expect, beforeEach, vi } from 'vitest';
import { renderHook, act } from '@testing-library/react-hooks';
import { SpaceRoleType } from '@coze-arch/idl/developer_api';
// 模拟全局变量
vi.stubGlobal('IS_DEV_MODE', true);
describe('Space Auth Store', () => {
beforeEach(() => {
// 重置模块缓存,确保每个测试都使用新的 store 实例
vi.resetModules();
});
describe('setRoles', () => {
it('应该正确设置空间角色', async () => {
// 动态导入 store 模块,确保每次测试都获取新的实例
const { useSpaceAuthStore } = await vi.importActual(
'../../src/space/store',
);
const { result } = renderHook(() => useSpaceAuthStore());
const roles = [SpaceRoleType.Owner, SpaceRoleType.Admin];
await act(() => {
result.current.setRoles('space1', roles);
});
expect(result.current.roles.space1).toEqual(roles);
});
it('应该能够为多个空间设置不同的角色', async () => {
const { useSpaceAuthStore } = await vi.importActual(
'../../src/space/store',
);
const { result } = renderHook(() => useSpaceAuthStore());
const roles1 = [SpaceRoleType.Owner];
const roles2 = [SpaceRoleType.Member];
await act(() => {
result.current.setRoles('space1', roles1);
result.current.setRoles('space2', roles2);
});
expect(result.current.roles.space1).toEqual(roles1);
expect(result.current.roles.space2).toEqual(roles2);
});
it('应该能够更新已存在空间的角色', async () => {
const { useSpaceAuthStore } = await vi.importActual(
'../../src/space/store',
);
const { result } = renderHook(() => useSpaceAuthStore());
const initialRoles = [SpaceRoleType.Owner];
const updatedRoles = [SpaceRoleType.Admin];
await act(() => {
result.current.setRoles('space1', initialRoles);
});
expect(result.current.roles.space1).toEqual(initialRoles);
await act(() => {
result.current.setRoles('space1', updatedRoles);
});
expect(result.current.roles.space1).toEqual(updatedRoles);
});
});
describe('setIsReady', () => {
it('应该正确设置空间数据准备状态', async () => {
const { useSpaceAuthStore } = await vi.importActual(
'../../src/space/store',
);
const { result } = renderHook(() => useSpaceAuthStore());
await act(() => {
result.current.setIsReady('space1', true);
});
expect(result.current.isReady.space1).toBe(true);
});
it('应该能够为多个空间设置不同的准备状态', async () => {
const { useSpaceAuthStore } = await vi.importActual(
'../../src/space/store',
);
const { result } = renderHook(() => useSpaceAuthStore());
await act(() => {
result.current.setIsReady('space1', true);
result.current.setIsReady('space2', false);
});
expect(result.current.isReady.space1).toBe(true);
expect(result.current.isReady.space2).toBe(false);
});
it('应该能够更新已存在空间的准备状态', async () => {
const { useSpaceAuthStore } = await vi.importActual(
'../../src/space/store',
);
const { result } = renderHook(() => useSpaceAuthStore());
await act(() => {
result.current.setIsReady('space1', false);
});
expect(result.current.isReady.space1).toBe(false);
await act(() => {
result.current.setIsReady('space1', true);
});
expect(result.current.isReady.space1).toBe(true);
});
});
describe('destory', () => {
it('应该正确清除空间数据', async () => {
const { useSpaceAuthStore } = await vi.importActual(
'../../src/space/store',
);
const { result } = renderHook(() => useSpaceAuthStore());
const roles = [SpaceRoleType.Owner];
// 设置初始数据
await act(() => {
result.current.setRoles('space1', roles);
result.current.setIsReady('space1', true);
});
// 验证数据已设置
expect(result.current.roles.space1).toEqual(roles);
expect(result.current.isReady.space1).toBe(true);
// 销毁数据
await act(() => {
result.current.destory('space1');
});
// 验证数据已清除
expect(result.current.roles.space1).toEqual([]);
expect(result.current.isReady.space1).toBeUndefined();
});
it('应该只清除指定空间的数据', async () => {
const { useSpaceAuthStore } = await vi.importActual(
'../../src/space/store',
);
const { result } = renderHook(() => useSpaceAuthStore());
// 设置两个空间的数据
await act(() => {
result.current.setRoles('space1', [SpaceRoleType.Owner]);
result.current.setIsReady('space1', true);
result.current.setRoles('space2', [SpaceRoleType.Member]);
result.current.setIsReady('space2', true);
});
// 只销毁 space1 的数据
await act(() => {
result.current.destory('space1');
});
// 验证 space1 的数据已清除
expect(result.current.roles.space1).toEqual([]);
expect(result.current.isReady.space1).toBeUndefined();
// 验证 space2 的数据保持不变
expect(result.current.roles.space2).toEqual([SpaceRoleType.Member]);
expect(result.current.isReady.space2).toBe(true);
});
});
describe('初始状态', () => {
it('应该有正确的初始状态', async () => {
const { useSpaceAuthStore } = await vi.importActual(
'../../src/space/store',
);
const { result } = renderHook(() => useSpaceAuthStore());
// 重置 store 确保测试环境干净
await act(() => {
Object.keys(result.current.roles).forEach(spaceId => {
result.current.destory(spaceId);
});
});
// 验证初始状态
expect(result.current.roles).toEqual({});
expect(result.current.isReady).toEqual({});
});
});
});

View File

@@ -0,0 +1,108 @@
/*
* Copyright 2025 coze-dev Authors
*
* 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 { describe, it, expect, vi, beforeEach } from 'vitest';
import { renderHook } from '@testing-library/react-hooks';
// 模拟 React 的 useEffect
const cleanupFns = new Map();
vi.mock('react', () => ({
useEffect: vi.fn((fn, deps) => {
// 执行 effect 函数并获取清理函数
const cleanup = fn();
// 存储清理函数,以便在 unmount 时调用
cleanupFns.set(fn, cleanup);
// 返回清理函数
return cleanup;
}),
}));
// 模拟 store
const mockDestory = vi.fn();
vi.mock('../../src/space/store', () => ({
useSpaceAuthStore: vi.fn(selector => selector({ destory: mockDestory })),
}));
// 创建一个包装函数,确保在 unmount 时调用清理函数
function renderHookWithCleanup(callback, options = {}) {
const result = renderHook(callback, options);
const originalUnmount = result.unmount;
result.unmount = () => {
// 调用所有清理函数
cleanupFns.forEach(cleanup => {
if (typeof cleanup === 'function') {
cleanup();
}
});
// 调用原始的 unmount
originalUnmount();
};
return result;
}
import { useDestorySpace } from '../../src/space/use-destory-space';
describe('useDestorySpace', () => {
beforeEach(() => {
vi.clearAllMocks();
cleanupFns.clear();
});
it('应该在组件卸载时调用 destory 方法', () => {
const spaceId = 'test-space-id';
// 渲染 hook
const { unmount } = renderHookWithCleanup(() => useDestorySpace(spaceId));
// 初始时不应调用 destory
expect(mockDestory).not.toHaveBeenCalled();
// 模拟组件卸载
unmount();
// 卸载时应调用 destory 并传入正确的 spaceId
expect(mockDestory).toHaveBeenCalledTimes(1);
expect(mockDestory).toHaveBeenCalledWith(spaceId);
});
it('应该为不同的 spaceId 调用 destory 方法', () => {
const spaceId1 = 'space-id-1';
const spaceId2 = 'space-id-2';
// 渲染第一个 hook 实例
const { unmount: unmount1 } = renderHookWithCleanup(() =>
useDestorySpace(spaceId1),
);
// 渲染第二个 hook 实例
const { unmount: unmount2 } = renderHookWithCleanup(() =>
useDestorySpace(spaceId2),
);
// 卸载第一个实例
unmount1();
expect(mockDestory).toHaveBeenCalledWith(spaceId1);
// 卸载第二个实例
unmount2();
expect(mockDestory).toHaveBeenCalledWith(spaceId2);
// 总共应调用两次
expect(mockDestory).toHaveBeenCalledTimes(4);
});
});

View File

@@ -0,0 +1,112 @@
/*
* Copyright 2025 coze-dev Authors
*
* 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 { describe, it, expect, vi } from 'vitest';
import { renderHook } from '@testing-library/react-hooks';
import { SpaceRoleType } from '@coze-arch/idl/developer_api';
import { ESpacePermisson } from '../../src/space/constants';
// 模拟 useSpaceRole
vi.mock('../../src/space/use-space-role', () => ({
useSpaceRole: vi.fn(),
}));
// 模拟 calcPermission
vi.mock('../../src/space/calc-permission', () => ({
calcPermission: vi.fn(),
}));
import { useSpaceRole } from '../../src/space/use-space-role';
import { calcPermission } from '../../src/space/calc-permission';
import { useSpaceAuth } from '../../src/space/use-space-auth';
describe('useSpaceAuth', () => {
it('应该使用 useSpaceRole 获取角色并调用 calcPermission 计算权限', () => {
const spaceId = 'test-space-id';
const permissionKey = ESpacePermisson.UpdateSpace;
const mockRoles = [SpaceRoleType.Owner];
// 模拟 useSpaceRole 返回角色
(useSpaceRole as unknown as ReturnType<typeof vi.fn>).mockReturnValue(
mockRoles,
);
// 模拟 calcPermission 返回权限结果
(calcPermission as unknown as ReturnType<typeof vi.fn>).mockReturnValue(
true,
);
// 渲染 hook
const { result } = renderHook(() => useSpaceAuth(permissionKey, spaceId));
// 验证 useSpaceRole 被调用,并传入正确的 spaceId
expect(useSpaceRole).toHaveBeenCalledWith(spaceId);
// 验证 calcPermission 被调用,并传入正确的参数
expect(calcPermission).toHaveBeenCalledWith(permissionKey, mockRoles);
// 验证返回值与 calcPermission 的返回值一致
expect(result.current).toBe(true);
});
it('应该在没有权限时返回 false', () => {
const spaceId = 'test-space-id';
const permissionKey = ESpacePermisson.UpdateSpace;
const mockRoles = [SpaceRoleType.Member];
// 模拟 useSpaceRole 返回角色
(useSpaceRole as unknown as ReturnType<typeof vi.fn>).mockReturnValue(
mockRoles,
);
// 模拟 calcPermission 返回权限结果
(calcPermission as unknown as ReturnType<typeof vi.fn>).mockReturnValue(
false,
);
// 渲染 hook
const { result } = renderHook(() => useSpaceAuth(permissionKey, spaceId));
// 验证返回值与 calcPermission 的返回值一致
expect(result.current).toBe(false);
});
it('应该在角色为空数组时返回 false', () => {
const spaceId = 'test-space-id';
const permissionKey = ESpacePermisson.UpdateSpace;
const mockRoles: SpaceRoleType[] = [];
// 模拟 useSpaceRole 返回空角色数组
(useSpaceRole as unknown as ReturnType<typeof vi.fn>).mockReturnValue(
mockRoles,
);
// 模拟 calcPermission 返回权限结果
(calcPermission as unknown as ReturnType<typeof vi.fn>).mockReturnValue(
false,
);
// 渲染 hook
const { result } = renderHook(() => useSpaceAuth(permissionKey, spaceId));
// 验证 calcPermission 被调用,并传入正确的参数
expect(calcPermission).toHaveBeenCalledWith(permissionKey, mockRoles);
// 验证返回值与 calcPermission 的返回值一致
expect(result.current).toBe(false);
});
});

View File

@@ -0,0 +1,138 @@
/*
* Copyright 2025 coze-dev Authors
*
* 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 { describe, it, expect, vi, beforeEach } from 'vitest';
import { renderHook } from '@testing-library/react-hooks';
import { SpaceRoleType } from '@coze-arch/idl/developer_api';
import { useSpaceAuthStore } from '../../src/space/store';
// 模拟 zustand
vi.mock('zustand/react/shallow', () => ({
useShallow: fn => fn,
}));
// 模拟 foundation-sdk
const mockUseSpace = vi.fn();
vi.mock('@coze-arch/foundation-sdk', () => ({
useSpace: (...args) => mockUseSpace(...args),
}));
// 模拟 store
vi.mock('../../src/space/store', () => ({
useSpaceAuthStore: vi.fn(),
}));
// 导入实际模块,确保在模拟之后导入
import { useSpaceRole } from '../../src/space/use-space-role';
describe('useSpaceRole', () => {
beforeEach(() => {
vi.clearAllMocks();
});
it('应该在 space 存在且 isReady 为 true 时返回角色', () => {
const spaceId = 'test-space-id';
const mockSpace = { id: spaceId, name: 'Test Space' };
const mockRoles = [SpaceRoleType.Owner];
// 模拟 useSpace 返回 space 对象
mockUseSpace.mockReturnValue(mockSpace);
// 模拟 useSpaceAuthStore 返回 isReady 和 role
(useSpaceAuthStore as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
isReady: true,
role: mockRoles,
});
// 渲染 hook
const { result } = renderHook(() => useSpaceRole(spaceId));
// 验证 useSpace 被调用,并传入正确的 spaceId
expect(mockUseSpace).toHaveBeenCalledWith(spaceId);
// 验证 useSpaceAuthStore 被调用,并传入正确的选择器
expect(useSpaceAuthStore).toHaveBeenCalled();
// 验证返回值与预期一致
expect(result.current).toEqual(mockRoles);
});
it('应该在 space 不存在时抛出错误', () => {
const spaceId = 'test-space-id';
// 模拟 useSpace 返回 null
mockUseSpace.mockReturnValue(null);
// 使用 vi.spyOn 监听 console.error 以防止测试输出错误信息
vi.spyOn(console, 'error').mockImplementation(() => {
// 空实现,防止错误输出
});
// 验证渲染 hook 时抛出错误
expect(() => useSpaceRole(spaceId)).toThrow(
'useSpaceAuth must be used after space list has been pulled.',
);
});
it('应该在 isReady 为 false 时抛出错误', () => {
const spaceId = 'test-space-id';
const mockSpace = { id: spaceId, name: 'Test Space' };
// 模拟 useSpace 返回 space 对象
mockUseSpace.mockReturnValue(mockSpace);
// 模拟 useSpaceAuthStore 返回 isReady 为 false
(useSpaceAuthStore as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
isReady: false,
role: null,
});
// 使用 vi.spyOn 监听 console.error 以防止测试输出错误信息
vi.spyOn(console, 'error').mockImplementation(() => {
// 空实现,防止错误输出
});
// 验证渲染 hook 时抛出错误
expect(() => useSpaceRole(spaceId)).toThrow(
'useSpaceAuth must be used after useInitSpaceRole has been completed.',
);
});
it('应该在 role 不存在时抛出错误', () => {
const spaceId = 'test-space-id';
const mockSpace = { id: spaceId, name: 'Test Space' };
// 模拟 useSpace 返回 space 对象
mockUseSpace.mockReturnValue(mockSpace);
// 模拟 useSpaceAuthStore 返回 isReady 为 true但 role 为 null
(useSpaceAuthStore as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
isReady: true,
role: null,
});
// 使用 vi.spyOn 监听 console.error 以防止测试输出错误信息
vi.spyOn(console, 'error').mockImplementation(() => {
// 空实现,防止错误输出
});
// 验证渲染 hook 时抛出错误
expect(() => useSpaceRole(spaceId)).toThrow(
`Can not get space role of space: ${spaceId}`,
);
});
});