nit

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)