sugarr/test/app/tools/MovieSearchTool.test.ts

import { describe, expect, it } from 'vitest';
import { MovieSearchTool } from '../../../src/app/tools/MovieSearchTool';
import type { RunContext } from '../../../src/core/llm/RunContext';

describe('MovieSearchTool.buildVariables', () => {
  it('passes the query and requested count through as prompt variables', () => {
    const tool = new MovieSearchTool(
      {} as unknown as ConstructorParameters<typeof MovieSearchTool>[0],
      {} as unknown as ConstructorParameters<typeof MovieSearchTool>[1]
    ) as unknown as { buildVariables(input: unknown, ctx: RunContext): Record<string, unknown> };

    const variables = tool.buildVariables({ query: 'sci-fi like Blade Runner', count: 5 }, {} as RunContext);
    expect(variables).toEqual({ query: 'sci-fi like Blade Runner', count: 5 });
  });
});