From e821b12ec6e508d221b04453c7e195603f3b3cfd Mon Sep 17 00:00:00 2001 From: huchenlei Date: Thu, 19 Sep 2024 11:17:44 +0900 Subject: [PATCH] nit --- tests-ui/tests/nodeSearchService.test.ts | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests-ui/tests/nodeSearchService.test.ts b/tests-ui/tests/nodeSearchService.test.ts index b5f5319a6..fea642ac9 100644 --- a/tests-ui/tests/nodeSearchService.test.ts +++ b/tests-ui/tests/nodeSearchService.test.ts @@ -1,7 +1,6 @@ import { NodeSearchService } from '@/services/nodeSearchService' import { ComfyNodeDefImpl } from '@/stores/nodeDefStore' import { plainToClass } from 'class-transformer' -import { mockNodeDefStore } from '../utils/setup' const EXAMPLE_NODE_DEFS: ComfyNodeDefImpl[] = [ { @@ -52,11 +51,14 @@ const EXAMPLE_NODE_DEFS: ComfyNodeDefImpl[] = [ category: 'latent/batch', output_node: false } -].map((nodeDef) => plainToClass(ComfyNodeDefImpl, nodeDef)) +].map((nodeDef) => { + const def = plainToClass(ComfyNodeDefImpl, nodeDef) + def['postProcessSearchScores'] = (s) => s + return def +}) describe('nodeSearchService', () => { it('searches with input filter', () => { - mockNodeDefStore() const service = new NodeSearchService(EXAMPLE_NODE_DEFS) const inputFilter = service.getFilterById('input') expect(service.searchNode('L', [[inputFilter, 'LATENT']])).toHaveLength(1)