pkg/vm/test/transformations/protobuf_aware_treeshaker/treeshaker_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:kernel/target/targets.dart';
import 'package:kernel/ast.dart';
import 'package:kernel/kernel.dart';
import 'package:kernel/binary/limited_ast_to_binary.dart';
import 'package:path/path.dart' as path;
import 'package:test/test.dart';

import 'package:vm/transformations/protobuf_aware_treeshaker/transformer.dart'
    as treeshaker;

import '../../common_test_utils.dart';

final String pkgVmDir = Platform.script.resolve('../../..').toFilePath();

runTestCase(Uri source) async {
  final target = new TestingVmTarget(new TargetFlags());
  Component component =
      await compileTestCaseToKernelProgram(source, target: target);

  List<Class> messageClasses = component.libraries
      .expand(
        (lib) => lib.classes.where((klass) =>
            klass.superclass != null &&
            klass.superclass.name == "GeneratedMessage"),
      )
      .toList();

  treeshaker.transformComponent(component, {}, TestingVmTarget(TargetFlags()),
      collectInfo: true, enableAsserts: false);

  for (Class messageClass in messageClasses) {
    expect(messageClass.enclosingLibrary.classes.contains(messageClass),
        messageClass.name.endsWith('Keep'));
  }

  final systemTempDir = Directory.systemTemp;
  final file =
      new File('${systemTempDir.path}/${source.pathSegments.last}.dill');
  try {
    final sink = file.openWrite();
    final printer =
        LimitedBinaryPrinter(sink, (lib) => true, /*excludeUriToSource=*/ true);

    printer.writeComponentFile(component);
    await sink.close();

    ProcessResult result =
        Process.runSync(Platform.resolvedExecutable, [file.path]);
    expect(result.exitCode, 0);
  } finally {
    if (file.existsSync()) {
      file.deleteSync();
    }
  }
}

main() async {
  final testCases = Directory(path.join(
    pkgVmDir,
    'testcases',
    'transformations',
    'protobuf_aware_treeshaker',
    'lib',
  )).listSync().where((f) => f.path.endsWith('_test.dart'));
  for (final entry in testCases) {
    test(entry.path, () => runTestCase(entry.uri));
  }
}