139 lines
4.5 KiB
TypeScript
139 lines
4.5 KiB
TypeScript
/*
|
|
* 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';
|
|
|
|
// Analog zustand
|
|
vi.mock('zustand/react/shallow', () => ({
|
|
useShallow: fn => fn,
|
|
}));
|
|
|
|
// Simulation foundation-sdk
|
|
const mockUseSpace = vi.fn();
|
|
vi.mock('@coze-arch/foundation-sdk', () => ({
|
|
useSpace: (...args) => mockUseSpace(...args),
|
|
}));
|
|
|
|
// Simulated store
|
|
vi.mock('../../src/space/store', () => ({
|
|
useSpaceAuthStore: vi.fn(),
|
|
}));
|
|
|
|
// Import the actual module, make sure to import it after simulation
|
|
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];
|
|
|
|
// Simulate useSpace Return space object
|
|
mockUseSpace.mockReturnValue(mockSpace);
|
|
|
|
// Emulate useSpaceAuthStore returns isReady and role
|
|
(useSpaceAuthStore as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
|
|
isReady: true,
|
|
role: mockRoles,
|
|
});
|
|
|
|
// Render hook
|
|
const { result } = renderHook(() => useSpaceRole(spaceId));
|
|
|
|
// Verify that useSpace is called, passing in the correct spaceId.
|
|
expect(mockUseSpace).toHaveBeenCalledWith(spaceId);
|
|
|
|
// Verify that useSpaceAuthStore is called, passing in the correct selector
|
|
expect(useSpaceAuthStore).toHaveBeenCalled();
|
|
|
|
// Verify that the return value is as expected
|
|
expect(result.current).toEqual(mockRoles);
|
|
});
|
|
|
|
it('应该在 space 不存在时抛出错误', () => {
|
|
const spaceId = 'test-space-id';
|
|
|
|
// Simulate useSpace returns null
|
|
mockUseSpace.mockReturnValue(null);
|
|
|
|
// Use vi.spyOn to listen to console.error to prevent test output error messages
|
|
vi.spyOn(console, 'error').mockImplementation(() => {
|
|
// Empty implementation to prevent error output
|
|
});
|
|
|
|
// Error thrown while validating render 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' };
|
|
|
|
// Simulate useSpace Return space object
|
|
mockUseSpace.mockReturnValue(mockSpace);
|
|
|
|
// Emulate useSpaceAuthStore returns isReady to false
|
|
(useSpaceAuthStore as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
|
|
isReady: false,
|
|
role: null,
|
|
});
|
|
|
|
// Use vi.spyOn to listen to console.error to prevent test output error messages
|
|
vi.spyOn(console, 'error').mockImplementation(() => {
|
|
// Empty implementation to prevent error output
|
|
});
|
|
|
|
// Error thrown while validating render 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' };
|
|
|
|
// Simulate useSpace Return space object
|
|
mockUseSpace.mockReturnValue(mockSpace);
|
|
|
|
// Emulate useSpaceAuthStore returns isReady as true, but role as null
|
|
(useSpaceAuthStore as unknown as ReturnType<typeof vi.fn>).mockReturnValue({
|
|
isReady: true,
|
|
role: null,
|
|
});
|
|
|
|
// Use vi.spyOn to listen to console.error to prevent test output error messages
|
|
vi.spyOn(console, 'error').mockImplementation(() => {
|
|
// Empty implementation to prevent error output
|
|
});
|
|
|
|
// Error thrown while validating render hook
|
|
expect(() => useSpaceRole(spaceId)).toThrow(
|
|
`Can not get space role of space: ${spaceId}`,
|
|
);
|
|
});
|
|
});
|