Compute ids for IR nodes and check equivalence with the AST.

tests/compiler/dart2js/inference/id_equivalence_test.dart

+// Copyright (c) 2017, 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:async_helper/async_helper.dart';
+import 'package:compiler/src/compiler.dart';
+import 'package:compiler/src/commandline_options.dart';
+import 'package:compiler/src/elements/elements.dart';
+import 'package:compiler/src/kernel/kernel.dart';
+import 'package:compiler/src/tree/nodes.dart' as ast;
+import 'package:expect/expect.dart';
+import 'package:kernel/ast.dart' as ir;
+import 'inference_test_helper.dart';
+import 'enumerator.dart';
+
+main() {
+  asyncTest(() async {
+    Directory dataDir = new Directory.fromUri(Platform.script.resolve('data'));
+    await for (FileSystemEntity entity in dataDir.list()) {
+      print('Checking ${entity.uri}');
+      String annotatedCode = await new File.fromUri(entity.uri).readAsString();
+      await checkCode(annotatedCode, checkMemberEquivalence,
+          options: [Flags.useKernel]);
+    }
+  });
+}
+
+/// Check that the ids in [expectedMap] map to equivalent nodes/elements in
+/// the AST and kernel IR.
+void checkMemberEquivalence(
+    Compiler compiler, Map<Id, String> expectedMap, MemberElement member) {
+  ResolvedAst resolvedAst = member.resolvedAst;
+  if (resolvedAst.kind != ResolvedAstKind.PARSED) return;
+  Map<Id, ast.Node> astMap = <Id, ast.Node>{};
+  Map<Id, Element> elementMap = <Id, Element>{};
+  AstIdFinder astFinder = new AstIdFinder(resolvedAst.elements);
+  for (Id id in expectedMap.keys.toList()) {
+    var result = astFinder.find(resolvedAst.node, id);
+    if (result is ast.Node) {
+      astMap[id] = result;
+    } else if (result is AstElement) {
+      elementMap[id] = result;
+    }
+  }
+
+  Kernel kernel = compiler.backend.kernelTask.kernel;
+  Map<Id, ir.Node> irMap = <Id, ir.Node>{};
+  ir.Node node = kernel.elementToIr(member);
+  IrIdFinder irFinder = new IrIdFinder();
+  for (Id id in expectedMap.keys.toList()) {
+    ir.Node result = irFinder.find(node, id);
+    if (result != null) {
+      irMap[id] = result;
+    }
+  }
+
+  elementMap.forEach((Id id, AstElement element) {
+    ir.Node irNode = irMap[id];
+    Expect.equals(kernel.elementToIr(element), irNode,
+        "Element mismatch on $id = $element");
+    expectedMap.remove(id);
+    irMap.remove(id);
+  });
+  astMap.forEach((Id id, ast.Node astNode) {
+    ir.Node irNode = irMap[id];
+    Expect.equals(
+        kernel.nodeToAst[irNode], astNode, "Node mismatch on $id = $astNode");
+    expectedMap.remove(id);
+    irMap.remove(id);
+  });
+  Expect.isTrue(irMap.isEmpty, "Extra IR ids: $irMap");
+}