pkg/vm/test/unlinked_ast_to_text_test.dart

// Copyright (c) 2019, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'dart:io';

import 'package:args/args.dart';
import 'package:expect/expect.dart';
import 'package:kernel/kernel.dart';
import 'package:kernel/binary/ast_to_binary.dart';
import 'package:vm/kernel_front_end.dart';

main() async {
  final outDir = Directory.systemTemp.createTempSync(
    "incremental_load_from_dill_test",
  );

  final dillFile = outDir.uri.resolve("dart2js.dart.dill").toFilePath();
  final unlinkedDillFile =
      outDir.uri.resolve("dart2js.dart.unlinked.dill").toFilePath();
  final unlinkedDillTxtFile =
      outDir.uri.resolve("dart2js.dart.unlinked dill.txt").toFilePath();

  final executable = File(Platform.executable).uri;
  final executableName = Platform.isWindows ? 'dart.exe' : 'dart';
  late final String platformDill;
  if (executable.path.endsWith('ReleaseX64/$executableName')) {
    platformDill = executable.resolve('vm_platform.dill').toFilePath();
  } else if (executable.path.endsWith('dart-sdk/bin/$executableName')) {
    platformDill =
        executable.resolve('../lib/_internal/vm_platform.dill').toFilePath();
  } else {
    print(
      'Skipping test due to not being run .../ReleaseX64/$executableName '
      'or .../dart-sdk/bin/$executableName.',
    );
    return;
  }

  try {
    final arguments = <String>[
      '--platform=$platformDill',
      '--output=$dillFile',
      'pkg/compiler/lib/src/dart2js.dart',
    ];

    // Compile dart2js.dart.
    final ArgParser argParser = createCompilerArgParser();
    final int exitCode = await runCompiler(
      argParser.parse(arguments),
      '<usage>',
    );
    Expect.equals(0, exitCode);

    // Load the dart2js.dart.dill and write the unlinked version.
    final component = loadComponentFromBinary(dillFile);
    final sink = File(unlinkedDillFile).openWrite();
    final printer = BinaryPrinter(
      sink,
      libraryFilter: (lib) => !lib.importUri.isScheme('dart'),
    );
    printer.writeComponentFile(component);
    await sink.close();

    // Ensure we can load the unlinked dill file and ensure it doesn't include
    // core libraries.
    final unlinkedComponent = loadComponentFromBinary(unlinkedDillFile);
    final coreLibraryCount =
        unlinkedComponent.libraries
            .where(
              (lib) => lib.importUri.isScheme('dart') && lib.members.isNotEmpty,
            )
            .length;
    Expect.equals(0, coreLibraryCount);

    // Ensure we can print the unlinked kernel to text.
    writeComponentToText(unlinkedComponent, path: unlinkedDillTxtFile);
  } finally {
    outDir.deleteSync(recursive: true);
  }
}