pkg/vm/test/transformations/ffi_test.dart

// Copyright (c) 2021, 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/ast.dart';
import 'package:kernel/class_hierarchy.dart';
import 'package:kernel/core_types.dart';
import 'package:kernel/kernel.dart';
import 'package:kernel/target/targets.dart';
import 'package:kernel/verifier.dart';
import 'package:test/test.dart';
import 'package:vm/kernel_front_end.dart'
    show runGlobalTransformations, ErrorDetector, KernelCompilationArguments;
import 'package:vm/modular/target/vm.dart' show VmTarget;
import 'package:vm/modular/transformations/ffi/native.dart'
    show transformLibraries;

import '../common_test_utils.dart';

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

class TestDiagnosticReporter extends DiagnosticReporter<Object, Object> {
  @override
  void report(
    Object message,
    int charOffset,
    int length,
    Uri? fileUri, {
    List<Object>? context,
  }) {
    /* nop */
  }
}

runTestCaseJit(Uri source) async {
  final target = VmTarget(TargetFlags());

  Component component = await compileTestCaseToKernelProgram(
    source,
    target: target,
    experimentalFlags: [],
  );

  final coreTypes = CoreTypes(component);

  transformLibraries(
    component,
    coreTypes,
    ClassHierarchy(component, coreTypes),
    component.libraries,
    TestDiagnosticReporter(),
    /*referenceFromIndex=*/ null,
  );

  verifyComponent(
    target,
    VerificationStage.afterModularTransformations,
    component,
  );

  final actual = kernelLibraryToString(component.mainMethod!.enclosingLibrary);

  compareResultWithExpectationsFile(source, actual);
}

runTestCaseAot(Uri source) async {
  final target = VmTarget(TargetFlags(supportMirrors: false));

  Component component = await compileTestCaseToKernelProgram(
    source,
    target: target,
    experimentalFlags: [],
  );

  final nopErrorDetector = ErrorDetector();
  runGlobalTransformations(
    target,
    component,
    nopErrorDetector,
    KernelCompilationArguments(
      useGlobalTypeFlowAnalysis: true,
      enableAsserts: false,
      useProtobufTreeShakerV2: true,
      treeShakeWriteOnlyFields: true,
    ),
  );

  verifyComponent(
    target,
    VerificationStage.afterGlobalTransformations,
    component,
  );

  final actual = kernelLibraryToString(component.mainMethod!.enclosingLibrary);

  compareResultWithExpectationsFile(source, actual, expectFilePostfix: '.aot');
}

void main(List<String> args) {
  assert(args.isEmpty || args.length == 1);
  String? filter;
  if (args.isNotEmpty) {
    filter = args.first;
  }

  group('ffi-transformations', () {
    final testCasesDir = Directory(pkgVmDir + 'testcases/transformations/ffi');

    for (var entry
        in testCasesDir
            .listSync(recursive: true, followLinks: false)
            .reversed) {
      if (entry.path.endsWith(".dart") &&
          (filter == null || entry.path.contains(filter))) {
        test(entry.path, () => runTestCaseJit(entry.uri));
        test('${entry.path} aot', () => runTestCaseAot(entry.uri));
      }
    }
  });
}