mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2025-12-22 17:00:14 +08:00
8353f331f1
- Refactored tests in MainTextBlock and ThinkingBlock to utilize the usePreference hook for managing user settings. - Updated snapshots in DraggableVirtualList test to reflect changes in class names. - Enhanced export tests to ensure proper handling of markdown formatting and citation footnotes. - Mocked additional dependencies globally for improved test reliability.
98 lines
2.7 KiB
TypeScript
98 lines
2.7 KiB
TypeScript
import { vi } from 'vitest'
|
|
|
|
/**
|
|
* Mock PreferenceService for testing
|
|
* Provides common preference defaults used across the application
|
|
*/
|
|
|
|
// Default preference values used in tests
|
|
export const mockPreferenceDefaults: Record<string, any> = {
|
|
// Export preferences
|
|
'data.export.markdown.force_dollar_math': false,
|
|
'data.export.markdown.exclude_citations': false,
|
|
'data.export.markdown.standardize_citations': true,
|
|
'data.export.markdown.show_model_name': false,
|
|
'data.export.markdown.show_model_provider': false,
|
|
|
|
// UI preferences
|
|
'ui.language': 'zh-CN',
|
|
'ui.theme': 'light',
|
|
'ui.font_size': 14,
|
|
|
|
// AI preferences
|
|
'ai.default_model': 'gpt-4',
|
|
'ai.temperature': 0.7,
|
|
'ai.max_tokens': 2000,
|
|
|
|
// Feature flags
|
|
'feature.web_search': true,
|
|
'feature.reasoning': false,
|
|
'feature.tool_calling': true,
|
|
|
|
// User preferences
|
|
'user.name': 'MockUser',
|
|
|
|
// App preferences
|
|
'app.user.name': 'MockUser',
|
|
'app.language': 'zh-CN',
|
|
|
|
// Add more defaults as needed
|
|
}
|
|
|
|
/**
|
|
* Mock implementation of PreferenceService
|
|
*/
|
|
export const createMockPreferenceService = (customDefaults: Record<string, any> = {}) => {
|
|
const mergedDefaults = { ...mockPreferenceDefaults, ...customDefaults }
|
|
|
|
return {
|
|
get: vi.fn((key: string) => {
|
|
const value = mergedDefaults[key]
|
|
return Promise.resolve(value !== undefined ? value : null)
|
|
}),
|
|
|
|
getMultiple: vi.fn((keys: Record<string, string>) => {
|
|
const result: Record<string, any> = {}
|
|
Object.entries(keys).forEach(([alias, key]) => {
|
|
const value = mergedDefaults[key]
|
|
result[alias] = value !== undefined ? value : null
|
|
})
|
|
return Promise.resolve(result)
|
|
}),
|
|
|
|
set: vi.fn((key: string, value: any) => {
|
|
mergedDefaults[key] = value
|
|
return Promise.resolve()
|
|
}),
|
|
|
|
setMultiple: vi.fn((values: Record<string, any>) => {
|
|
Object.assign(mergedDefaults, values)
|
|
return Promise.resolve()
|
|
}),
|
|
|
|
delete: vi.fn((key: string) => {
|
|
delete mergedDefaults[key]
|
|
return Promise.resolve()
|
|
}),
|
|
|
|
clear: vi.fn(() => {
|
|
Object.keys(mergedDefaults).forEach(key => delete mergedDefaults[key])
|
|
return Promise.resolve()
|
|
}),
|
|
|
|
// Internal state access for testing
|
|
_getMockState: () => ({ ...mergedDefaults }),
|
|
_resetMockState: () => {
|
|
Object.keys(mergedDefaults).forEach(key => delete mergedDefaults[key])
|
|
Object.assign(mergedDefaults, mockPreferenceDefaults, customDefaults)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Default mock instance
|
|
export const mockPreferenceService = createMockPreferenceService()
|
|
|
|
// Export for easy mocking in individual tests
|
|
export const MockPreferenceService = {
|
|
preferenceService: mockPreferenceService
|
|
} |