97 lines
2.8 KiB
TypeScript
97 lines
2.8 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 } from 'vitest';
|
|
import { renderHook } from '@testing-library/react';
|
|
import {
|
|
userStoreService,
|
|
type UserInfo,
|
|
type UserLabel,
|
|
} from '@coze-studio/user-store';
|
|
|
|
import { useUserSenderInfo } from '../../src/bot/use-user-sender-info';
|
|
|
|
// Mock dependencies
|
|
vi.mock('@coze-studio/user-store', () => ({
|
|
userStoreService: {
|
|
useUserLabel: vi.fn(),
|
|
useUserInfo: vi.fn(),
|
|
},
|
|
}));
|
|
|
|
describe('useUserSenderInfo', () => {
|
|
const mockUserLabel = {
|
|
id: 'label-1',
|
|
name: 'Test Label',
|
|
} as unknown as UserLabel;
|
|
const mockUserInfo: Partial<UserInfo> = {
|
|
avatar_url: 'https://example.com/avatar.jpg',
|
|
name: 'Test User',
|
|
user_id_str: '12345',
|
|
app_user_info: {
|
|
user_unique_name: 'test_user',
|
|
},
|
|
};
|
|
|
|
it('should return null when userInfo is not available', () => {
|
|
vi.mocked(userStoreService.useUserLabel).mockReturnValue(mockUserLabel);
|
|
vi.mocked(userStoreService.useUserInfo).mockReturnValue(null);
|
|
|
|
const { result } = renderHook(() => useUserSenderInfo());
|
|
|
|
expect(result.current).toBeNull();
|
|
});
|
|
|
|
it('should return formatted user sender info when userInfo is available', () => {
|
|
vi.mocked(userStoreService.useUserLabel).mockReturnValue(mockUserLabel);
|
|
vi.mocked(userStoreService.useUserInfo).mockReturnValue(
|
|
mockUserInfo as UserInfo,
|
|
);
|
|
|
|
const { result } = renderHook(() => useUserSenderInfo());
|
|
|
|
expect(result.current).toEqual({
|
|
url: mockUserInfo.avatar_url,
|
|
nickname: mockUserInfo.name,
|
|
id: mockUserInfo.user_id_str,
|
|
userUniqueName: mockUserInfo.app_user_info?.user_unique_name,
|
|
userLabel: mockUserLabel,
|
|
});
|
|
});
|
|
|
|
it('should handle missing optional fields', () => {
|
|
const partialUserInfo: Partial<UserInfo> = {
|
|
user_id_str: '12345',
|
|
app_user_info: {},
|
|
};
|
|
|
|
vi.mocked(userStoreService.useUserLabel).mockReturnValue(mockUserLabel);
|
|
vi.mocked(userStoreService.useUserInfo).mockReturnValue(
|
|
partialUserInfo as UserInfo,
|
|
);
|
|
|
|
const { result } = renderHook(() => useUserSenderInfo());
|
|
|
|
expect(result.current).toEqual({
|
|
url: '',
|
|
nickname: '',
|
|
id: partialUserInfo.user_id_str,
|
|
userUniqueName: '',
|
|
userLabel: mockUserLabel,
|
|
});
|
|
});
|
|
});
|