Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-01-26 19:09:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Set default empty record for InputSpec (#225)

Browse Source
This commit is contained in:
Chenlei Hu
2024-07-25 21:29:22 -04:00
committed by GitHub
parent c97ff6fd85
commit 9d56bb4e0e
2 changed files with 18 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/stores/nodeDefStore.ts
View File

@@ -77,10 +77,10 @@ export class CustomInputSpec extends BaseInputSpec {}
export class ComfyInputsSpec {
@Transform(({ value }) => ComfyInputsSpec.transformInputSpecRecord(value))
required?: Record<string, BaseInputSpec>
required: Record<string, BaseInputSpec> = {}
@Transform(({ value }) => ComfyInputsSpec.transformInputSpecRecord(value))
optional?: Record<string, BaseInputSpec>
optional: Record<string, BaseInputSpec> = {}
hidden?: Record<string, any>

32
tests-ui/tests/nodeDef.test.ts
View File

@@ -47,11 +47,11 @@ describe('ComfyInputsSpec', () => {
const result = plainToClass(ComfyInputsSpec, plainObject)
expect(result.required?.intInput).toBeInstanceOf(IntInputSpec)
expect(result.required?.stringInput).toBeInstanceOf(StringInputSpec)
expect(result.required.intInput).toBeInstanceOf(IntInputSpec)
expect(result.required.stringInput).toBeInstanceOf(StringInputSpec)
const intInput = result.required?.intInput as IntInputSpec
const stringInput = result.required?.stringInput as StringInputSpec
const intInput = result.required.intInput as IntInputSpec
const stringInput = result.required.stringInput as StringInputSpec
expect(intInput.min).toBe(0)
expect(intInput.max).toBe(100)
@@ -75,11 +75,11 @@ describe('ComfyInputsSpec', () => {
const result = plainToClass(ComfyInputsSpec, plainObject)
expect(result.optional?.booleanInput).toBeInstanceOf(BooleanInputSpec)
expect(result.optional?.floatInput).toBeInstanceOf(FloatInputSpec)
expect(result.optional.booleanInput).toBeInstanceOf(BooleanInputSpec)
expect(result.optional.floatInput).toBeInstanceOf(FloatInputSpec)
const booleanInput = result.optional?.booleanInput as BooleanInputSpec
const floatInput = result.optional?.floatInput as FloatInputSpec
const booleanInput = result.optional.booleanInput as BooleanInputSpec
const floatInput = result.optional.floatInput as FloatInputSpec
expect(booleanInput.default).toBe(true)
expect(booleanInput.labelOn).toBe('Yes')
@@ -98,9 +98,9 @@ describe('ComfyInputsSpec', () => {
const result = plainToClass(ComfyInputsSpec, plainObject)
expect(result.optional?.comboInput).toBeInstanceOf(ComboInputSpec)
expect(result.optional?.comboInput.type).toBe('COMBO')
expect(result.optional?.comboInput.default).toBe(2)
expect(result.optional.comboInput).toBeInstanceOf(ComboInputSpec)
expect(result.optional.comboInput.type).toBe('COMBO')
expect(result.optional.comboInput.default).toBe(2)
})
it('should handle custom input specs', () => {
@@ -112,9 +112,9 @@ describe('ComfyInputsSpec', () => {
const result = plainToClass(ComfyInputsSpec, plainObject)
expect(result.optional?.customInput).toBeInstanceOf(CustomInputSpec)
expect(result.optional?.customInput.type).toBe('CUSTOM_TYPE')
expect(result.optional?.customInput.default).toBe('custom value')
expect(result.optional.customInput).toBeInstanceOf(CustomInputSpec)
expect(result.optional.customInput.type).toBe('CUSTOM_TYPE')
expect(result.optional.customInput.default).toBe('custom value')
})
it('should not transform hidden fields', () => {
@@ -138,8 +138,8 @@ describe('ComfyInputsSpec', () => {
const result = plainToClass(ComfyInputsSpec, plainObject)
expect(result).toBeInstanceOf(ComfyInputsSpec)
expect(result.required).toBeUndefined()
expect(result.optional).toBeUndefined()
expect(result.required).toEqual({})
expect(result.optional).toEqual({})
expect(result.hidden).toBeUndefined()
})
})