Serialize data for LibraryElement.getImportsFor

tests/compiler/dart2js/serialization/model_test_helper.dart

 // Copyright (c) 2015, 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.
 
 library dart2js.serialization_model_test;
 
 import 'dart:async';
 import 'dart:io';
 import 'package:async_helper/async_helper.dart';
 import 'package:expect/expect.dart';
 import 'package:compiler/src/closure.dart';
 import 'package:compiler/src/commandline_options.dart';
+import 'package:compiler/src/common.dart';
 import 'package:compiler/src/compiler.dart';
+import 'package:compiler/src/deferred_load.dart';
 import 'package:compiler/src/elements/elements.dart';
 import 'package:compiler/src/filenames.dart';
 import 'package:compiler/src/js_backend/js_backend.dart';
 import 'package:compiler/src/serialization/equivalence.dart';
 import 'package:compiler/src/tree/nodes.dart';
 import 'package:compiler/src/universe/class_set.dart';
 import '../memory_compiler.dart';
 import 'helper.dart';
 import 'test_data.dart';
 import 'test_helper.dart';
 
 main(List<String> args) {
   asyncTest(() async {
     Arguments arguments = new Arguments.from(args);
     SerializedData serializedData =
         await serializeDartCore(arguments: arguments);
     if (arguments.filename != null) {
       Uri entryPoint = Uri.base.resolve(nativeToUriPath(arguments.filename));
+      print('----------------------------------------------------------------');
+      print('serialize ${entryPoint}');
+      print('----------------------------------------------------------------');
+      SerializationResult result = await serialize(
+          entryPoint,
+          memorySourceFiles: serializedData.toMemorySourceFiles(),
+          resolutionInputs: serializedData.toUris(),
+          dataUri: Uri.parse('memory:test.data'));
       await checkModels(entryPoint,
-          sourceFiles: serializedData.toMemorySourceFiles(),
-          resolutionInputs: serializedData.toUris());
+          sourceFiles: serializedData.toMemorySourceFiles(
+              result.serializedData.toMemorySourceFiles()),
+          resolutionInputs: serializedData.toUris(
+              result.serializedData.toUris()));
     } else {
       Uri entryPoint = Uri.parse('memory:main.dart');
       await arguments.forEachTest(serializedData, TESTS, checkModels);
     }
   });
 }
 
 Future checkModels(
     Uri entryPoint,
     {Map<String, String> sourceFiles: const <String, String>{},
      List<Uri> resolutionInputs,
      int index,
      Test test,
      bool verbose: false}) async {
   String testDescription = test != null ? test.name : '${entryPoint}';
   String id = index != null ? '$index: ' : '';
   print('------------------------------------------------------------------');
   print('compile normal ${id}${testDescription}');
   print('------------------------------------------------------------------');
   Compiler compilerNormal = compilerFor(
       memorySourceFiles: sourceFiles,
       options: [Flags.analyzeOnly]);
   compilerNormal.resolution.retainCachesForTesting = true;
   await compilerNormal.run(entryPoint);
   compilerNormal.phase = Compiler.PHASE_DONE_RESOLVING;
   compilerNormal.world.populate();
   compilerNormal.backend.onResolutionComplete();
+  compilerNormal.deferredLoadTask.onResolutionComplete(
+      compilerNormal.mainFunction);
 
   print('------------------------------------------------------------------');
   print('compile deserialized ${id}${testDescription}');
   print('------------------------------------------------------------------');
   Compiler compilerDeserialized = compilerFor(
       memorySourceFiles: sourceFiles,
       resolutionInputs: resolutionInputs,
       options: [Flags.analyzeOnly]);
   compilerDeserialized.resolution.retainCachesForTesting = true;
   await compilerDeserialized.run(entryPoint);
   compilerDeserialized.phase = Compiler.PHASE_DONE_RESOLVING;
   compilerDeserialized.world.populate();
   compilerDeserialized.backend.onResolutionComplete();
+  compilerDeserialized.deferredLoadTask.onResolutionComplete(
+      compilerDeserialized.mainFunction);
 
   checkAllImpacts(
       compilerNormal, compilerDeserialized,
       verbose: verbose);
 
   checkSets(
       compilerNormal.resolverWorld.directlyInstantiatedClasses,
       compilerDeserialized.resolverWorld.directlyInstantiatedClasses,
       "Directly instantiated classes mismatch",
       areElementsEquivalent,
@@ skipping 38 matching lines @@
       "Compiler.enabledInvokeOn mismatch");
   Expect.equals(compilerNormal.enabledFunctionApply,
       compilerDeserialized.enabledFunctionApply,
       "Compiler.enabledFunctionApply mismatch");
   Expect.equals(compilerNormal.enabledRuntimeType,
       compilerDeserialized.enabledRuntimeType,
       "Compiler.enabledRuntimeType mismatch");
   Expect.equals(compilerNormal.hasIsolateSupport,
       compilerDeserialized.hasIsolateSupport,
       "Compiler.hasIsolateSupport mismatch");
+  Expect.equals(
+      compilerNormal.deferredLoadTask.isProgramSplit,
+      compilerDeserialized.deferredLoadTask.isProgramSplit,
+      "isProgramSplit mismatch");
 }
 
 void checkElements(
     Compiler compiler1, Compiler compiler2,
     Element element1, Element element2,
     {bool verbose: false}) {
   if (element1.isFunction ||
       element1.isConstructor ||
       (element1.isField && element1.isInstanceMember)) {
     AstElement astElement1 = element1;
@@ skipping 61 matching lines @@
                 const CheckStrategy());
           });
     }
   }
   JavaScriptBackend backend1 = compiler1.backend;
   JavaScriptBackend backend2 = compiler2.backend;
   Expect.equals(
       backend1.inlineCache.getCurrentCacheDecisionForTesting(element1),
       backend2.inlineCache.getCurrentCacheDecisionForTesting(element2),
       "Inline cache decision mismatch for $element1 vs $element2");
+
+  checkOutputUnits(compiler1, compiler2, element1, element2);
 }
 
 void checkMixinUses(
     Compiler compiler1, Compiler compiler2,
     ClassElement class1, ClassElement class2,
     {bool verbose: false}) {
 
   checkSets(
       compiler1.world.mixinUsesOf(class1),
       compiler2.world.mixinUsesOf(class2),
@@ skipping 93 matching lines @@
       areElementsEquivalent);
   return true;
 }
 
 String nodeToString(Node node) {
   String text = '$node';
   if (text.length > 40) {
     return '(${node.runtimeType}) ${text.substring(0, 37)}...';
   }
   return '(${node.runtimeType}) $text';
+}
+
+void checkOutputUnits(Compiler compiler1, Compiler compiler2, Element element1,
+    Element element2) {
+  OutputUnit outputUnit1 =
+      compiler1.deferredLoadTask.outputUnitForElement(element1);
+  OutputUnit outputUnit2 =
+      compiler2.deferredLoadTask.outputUnitForElement(element2);
+  check(outputUnit1, outputUnit2,
+      'OutputUnit.isMainOutput for $element1 vs $element2',
+      outputUnit1.isMainOutput, outputUnit2.isMainOutput);
+  checkSetEquivalence(outputUnit1, outputUnit2,
+      'OutputUnit.imports for $element1 vs $element2',
+      outputUnit1.imports, outputUnit2.imports,
+      (a, b) => areElementsEquivalent(a.declaration, b.declaration));
 }