Separate prelinked and unlinked parts of summaries.

pkg/analyzer/test/src/summary/resynthesize_test.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 test.src.serialization.elements_test;
 
 import 'package:analyzer/src/generated/element.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:analyzer/src/summary/builder.dart';
 import 'package:analyzer/src/summary/format.dart';
@@ skipping 399 matching lines @@
     expect(resynthesized.uri, original.uri);
   }
 
   void compareVariableElements(VariableElementImpl resynthesized,
       VariableElementImpl original, String desc) {
     compareElements(resynthesized, original, desc);
     compareTypes(resynthesized.type, original.type, desc);
     // TODO(paulberry): test initializer
   }
 
-  PrelinkedLibrary getSummaryFor(LibraryElement lib) {
-    BuilderContext ctx = new BuilderContext();
-    List<int> summary = serializeLibrary(ctx, lib, typeProvider).toBuffer();
-    return new PrelinkedLibrary.fromBuffer(summary);
-  }
-
   LibraryElementImpl resynthesizeLibrary(
       Source source, LibraryElementImpl original, bool allowErrors) {
     if (!allowErrors) {
       assertNoErrors(source);
     }
     String uri = source.uri.toString();
     addLibrary('dart:core');
     return resynthesizeLibraryElement(uri, original);
   }
 
   LibraryElementImpl resynthesizeLibraryElement(
       String uri, LibraryElementImpl original) {
-    Map<String, PrelinkedLibrary> summaries = <String, PrelinkedLibrary>{
-      uri: getSummaryFor(original)
-    };
+    Map<String, UnlinkedUnit> unlinkedSummaries = <String, UnlinkedUnit>{};
+    PrelinkedLibrary getPrelinkedSummaryFor(LibraryElement lib) {
+      BuilderContext ctx = new BuilderContext();
+      LibrarySerializationResult serialized =
+          serializeLibrary(ctx, lib, typeProvider);
+      for (int i = 0; i < serialized.unlinkedUnits.length; i++) {
+        unlinkedSummaries[serialized.unitUris[i]] =
+            new UnlinkedUnit.fromBuffer(serialized.unlinkedUnits[i].toBuffer());
+      }
+      return new PrelinkedLibrary.fromBuffer(serialized.prelinked.toBuffer());
+    }
+    Map<String, PrelinkedLibrary> prelinkedSummaries =
+        <String, PrelinkedLibrary>{uri: getPrelinkedSummaryFor(original)};
     for (Source source in otherLibrarySources) {
       LibraryElement original = resolve2(source);
       String uri = source.uri.toString();
-      summaries[uri] = getSummaryFor(original);
+      prelinkedSummaries[uri] = getPrelinkedSummaryFor(original);
     }
-    PrelinkedLibrary getSummary(String uri) {
-      PrelinkedLibrary serializedLibrary = summaries[uri];
+    PrelinkedLibrary getPrelinkedSummary(String uri) {
+      PrelinkedLibrary serializedLibrary = prelinkedSummaries[uri];
       if (serializedLibrary == null) {
-        fail('Unexpectedly tried to get serialized library for $uri');
+        fail('Unexpectedly tried to get prelinked summary for $uri');
       }
       return serializedLibrary;
     }
+    UnlinkedUnit getUnlinkedSummary(String uri) {
+      UnlinkedUnit serializedUnit = unlinkedSummaries[uri];
+      if (serializedUnit == null) {
+        fail('Unexpectedly tried to get unlinked summary for $uri');
+      }
+      return serializedUnit;
+    }
     SummaryResynthesizer resynthesizer = new SummaryResynthesizer(
-        analysisContext, getSummary, analysisContext.sourceFactory);
+        analysisContext,
+        getPrelinkedSummary,
+        getUnlinkedSummary,
+        analysisContext.sourceFactory);
     LibraryElementImpl resynthesized = resynthesizer.getLibraryElement(uri);
     // Check that no other summaries needed to be resynthesized to resynthesize
     // the library element.
     expect(resynthesizer.resynthesisCount, 1);
     return resynthesized;
   }
 
   test_class_alias() {
     checkLibrary('class C = D with E, F; class D {} class E {} class F {}');
   }
@@ skipping 552 matching lines @@
   }
 
   test_variable_const() {
     checkLibrary('const int i = 0;');
   }
 
   test_variables() {
     checkLibrary('int i; int j;');
   }
 }