Mock settingStore (#845)

This commit is contained in:
Chenlei Hu
2024-09-16 16:25:32 +09:00
committed by GitHub
parent 74fa4a2c2d
commit 091b8a74fb
4 changed files with 38 additions and 27 deletions

View File

@@ -1,3 +1,5 @@
import type { ComfySettingsDialog } from '@/scripts/ui/settings'
import type { ComfyApp } from '@/scripts/app'
import '../../src/scripts/api'
const fs = require('fs')
@@ -97,3 +99,28 @@ export function mockApi(config: APIConfig = {}) {
}
}))
}
export const mockSettingStore = () => {
let app: ComfyApp | null = null
const mockedSettingStore = {
addSettings(settings: ComfySettingsDialog) {
app = settings.app
},
set(key: string, value: any) {
app?.ui.settings.setSettingValue(key, value)
},
get(key: string) {
return (
app?.ui.settings.getSettingValue(key) ??
app?.ui.settings.getSettingDefaultValue(key)
)
}
}
jest.mock('@/stores/settingStore', () => ({
useSettingStore: jest.fn(() => mockedSettingStore)
}))
}