qwen-code/integration-tests/interactive/file-system-interactive.test.ts

/**
 * @license
 * Copyright 2025 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import { expect, describe, it, beforeEach, afterEach } from 'vitest';
import { TestRig, type, printDebugInfo } from '../test-helper.js';

describe('Interactive file system', () => {
  let rig: TestRig;

  beforeEach(() => {
    rig = new TestRig();
  });

  afterEach(async () => {
    await rig.cleanup();
  });

  it.skipIf(process.platform === 'win32')(
    'should perform a read-then-write sequence in interactive mode',
    async () => {
      const fileName = 'version.txt';
      await rig.setup('interactive-read-then-write', {
        settings: {
          security: {
            auth: {
              selectedType: 'openai',
            },
          },
        },
      });
      rig.createFile(fileName, '1.0.0');

      const { ptyProcess } = rig.runInteractive();

      // Wait for the app to be ready
      const isReady = await rig.waitForText('Type your message');
      expect(
        isReady,
        'CLI did not start up in interactive mode correctly',
      ).toBe(true);

      // Step 1: Read the file
      const readPrompt = `Read the version from ${fileName}`;
      await type(ptyProcess, readPrompt);
      await type(ptyProcess, '\r');

      const readCall = await rig.waitForToolCall('read_file');
      expect(readCall, 'Expected to find a read_file tool call').toBe(true);

      const containsExpectedVersion = await rig.waitForText('1.0.0');
      expect(
        containsExpectedVersion,
        'Expected to see version "1.0.0" in output',
      ).toBe(true);

      // Step 2: Write the file
      const writePrompt = `now change the version to 1.0.1 in the file`;
      await type(ptyProcess, writePrompt);
      await type(ptyProcess, '\r');

      const toolCall = await rig.waitForAnyToolCall(['write_file', 'edit']);

      if (!toolCall) {
        printDebugInfo(rig, rig._interactiveOutput, {
          toolCall,
        });
      }

      expect(toolCall, 'Expected to find a write_file or edit tool call').toBe(
        true,
      );

      // The tool call is logged once the model issues it, but the turn may
      // still be settling (a failed edit can be retried) and the model may
      // write more than just '1.0.1'. Poll the file until it contains the new
      // version, matching the lenient assertion used by the non-interactive
      // sibling test (file-system.test.ts uses .toContain('1.0.1')).
      const updated = await rig.poll(
        () => rig.readFile(fileName).includes('1.0.1'),
        rig.getDefaultTimeout(),
        200,
      );
      if (!updated) {
        printDebugInfo(rig, rig._interactiveOutput, { toolCall });
      }
      expect(updated, 'Expected file content to contain 1.0.1').toBe(true);
    },
  );
});