packages/angular/cli/src/commands/mcp/tools/onpush-zoneless-migration/migrate-test-file.ts - main - githubmirror/angular-cli

packages/angular/cli/src/commands/mcp/tools/onpush-zoneless-migration/migrate-test-file.ts

72 строки2 KB

Charles Lyding

refactor(@angular/cli): migrate MCP zoneless tool to host abstraction

08 май 2026, 15:51

08 май 2026, 15:5166ec762

/**
 * @license
 * Copyright Google LLC All Rights Reserved.
 *
 * Use of this source code is governed by an MIT-style license that can be
 * found in the LICENSE file at https://angular.dev/license
 */

import { join } from 'node:path';
import type { SourceFile } from 'typescript';
import { type Host } from '../../host';
import { findAngularJsonDir } from '../../workspace-utils';
import { createFixResponseForZoneTests, createProvideZonelessForTestsSetupPrompt } from './prompts';
import { loadTypescript } from './ts-utils';
import { MigrationResponse } from './types';

export async function migrateTestFile(
  sourceFile: SourceFile,
  host: Host,
): Promise<MigrationResponse | null> {
  const ts = await loadTypescript();
  // Check if tests use zoneless either by default through `initTestEnvironment` or by explicitly calling `provideZonelessChangeDetection`.
  let testsUseZonelessChangeDetection = await searchForGlobalZoneless(sourceFile.fileName, host);
  if (!testsUseZonelessChangeDetection) {
    ts.forEachChild(sourceFile, function visit(node) {
      if (
        ts.isCallExpression(node) &&
        node.expression.getText(sourceFile) === 'provideZonelessChangeDetection'
      ) {
        testsUseZonelessChangeDetection = true;

        return;
      }
      ts.forEachChild(node, visit);
    });
  }

  if (!testsUseZonelessChangeDetection) {
    // Tests do not use zoneless, so we provide instructions to set it up.
    return createProvideZonelessForTestsSetupPrompt(sourceFile.fileName);
  }

  // At this point, tests are using zoneless, so we look for any explicit uses of `provideZoneChangeDetection` that need to be fixed.
  return createFixResponseForZoneTests(sourceFile);
}

export async function searchForGlobalZoneless(startPath: string, host: Host): Promise<boolean> {
  const angularJsonDir = findAngularJsonDir(startPath, host);
  if (!angularJsonDir) {
    // Cannot determine project root, fallback to original behavior or assume false.
    // For now, let's assume no global setup if angular.json is not found.
    return false;
  }

  try {
    const files = host.glob('**/*.ts', { cwd: angularJsonDir });
    for await (const file of files) {
      const fullPath = join(file.parentPath, file.name);
      const content = await host.readFile(fullPath, 'utf-8');
      if (
        content.includes('initTestEnvironment') &&
        content.includes('provideZonelessChangeDetection')
      ) {
        return true;
      }
    }
  } catch (e) {
    return false;
  }

  return false;
}