diff --git a/web/app/components/base/prompt-editor/plugins/hitl-input-block/__tests__/input-field.spec.tsx b/web/app/components/base/prompt-editor/plugins/hitl-input-block/__tests__/input-field.spec.tsx index 009c5e728e..8394d6468e 100644 --- a/web/app/components/base/prompt-editor/plugins/hitl-input-block/__tests__/input-field.spec.tsx +++ b/web/app/components/base/prompt-editor/plugins/hitl-input-block/__tests__/input-field.spec.tsx @@ -1,15 +1,19 @@ import type { FormInputItem } from '@/app/components/workflow/nodes/human-input/types' import { render, screen } from '@testing-library/react' import userEvent from '@testing-library/user-event' -import { InputVarType } from '@/app/components/workflow/types' +import { InputVarType, VarType } from '@/app/components/workflow/types' import InputField from '../input-field' type VarReferencePickerProps = { onChange: (value: string[]) => void + filterVar?: (payload: { type: VarType }) => boolean } +let lastVarReferencePickerProps: VarReferencePickerProps | undefined + vi.mock('@/app/components/workflow/nodes/_base/components/variable/var-reference-picker', () => ({ default: (props: VarReferencePickerProps) => { + lastVarReferencePickerProps = props return (