mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-01-26 19:09:52 +00:00
b264685052
## Summary See https://typescript-eslint.io/blog/project-service/ for context. Creates a browser_tests specific tsconfig so that they can be linted. Does not add a package.json script to do the linting yet, but `pnpm exec eslint browser_tests` should work for now. ┆Issue is synchronized with this [Notion page](https://www.notion.so/PR-5633-lint-add-tsconfig-for-browser_tests-fix-existing-violations-2726d73d3650819d8ef2c4b0abc31e14) by [Unito](https://www.unito.io)
78 lines
2.0 KiB
TypeScript
78 lines
2.0 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
|
|
|
import type { ComfyNodeDef } from '@/schemas/nodeDefSchema'
|
|
import { NodeSearchService } from '@/services/nodeSearchService'
|
|
import { ComfyNodeDefImpl } from '@/stores/nodeDefStore'
|
|
|
|
const EXAMPLE_NODE_DEFS: ComfyNodeDefImpl[] = (
|
|
[
|
|
{
|
|
input: {
|
|
required: {
|
|
ckpt_name: [['model1.safetensors', 'model2.ckpt'], {}]
|
|
}
|
|
},
|
|
output: ['MODEL', 'CLIP', 'VAE'],
|
|
output_is_list: [false, false, false],
|
|
output_name: ['MODEL', 'CLIP', 'VAE'],
|
|
name: 'CheckpointLoaderSimple',
|
|
display_name: 'Load Checkpoint',
|
|
description: '',
|
|
python_module: 'nodes',
|
|
category: 'loaders',
|
|
output_node: false
|
|
},
|
|
{
|
|
input: {
|
|
required: {
|
|
samples: ['LATENT'],
|
|
batch_index: [
|
|
'INT',
|
|
{
|
|
default: 0,
|
|
min: 0,
|
|
max: 63
|
|
}
|
|
],
|
|
length: [
|
|
'INT',
|
|
{
|
|
default: 1,
|
|
min: 1,
|
|
max: 64
|
|
}
|
|
]
|
|
}
|
|
},
|
|
output: ['LATENT'],
|
|
output_is_list: [false],
|
|
output_name: ['LATENT'],
|
|
name: 'LatentFromBatch',
|
|
display_name: 'Latent From Batch',
|
|
description: '',
|
|
python_module: 'nodes',
|
|
category: 'latent/batch',
|
|
output_node: false
|
|
}
|
|
] as ComfyNodeDef[]
|
|
).map((nodeDef: ComfyNodeDef) => {
|
|
const def = new ComfyNodeDefImpl(nodeDef)
|
|
def['postProcessSearchScores'] = (s) => s
|
|
return def
|
|
})
|
|
|
|
describe('nodeSearchService', () => {
|
|
it('searches with input filter', () => {
|
|
const service = new NodeSearchService(EXAMPLE_NODE_DEFS)
|
|
const inputFilter = service.inputTypeFilter
|
|
expect(
|
|
service.searchNode('L', [{ filterDef: inputFilter, value: 'LATENT' }])
|
|
).toHaveLength(1)
|
|
// Wildcard should match all.
|
|
expect(
|
|
service.searchNode('L', [{ filterDef: inputFilter, value: '*' }])
|
|
).toHaveLength(2)
|
|
expect(service.searchNode('L')).toHaveLength(2)
|
|
})
|
|
})
|