Add a unit index to prelinked type references in summaries.

pkg/analyzer/test/src/summary/summary_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 'dart:typed_data';
 
 import 'package:analyzer/src/generated/element.dart';
 import 'package:analyzer/src/generated/engine.dart';
@@ skipping 192 matching lines @@
    * in a file reachable via [absoluteUri] and [relativeUri], having name
    * [expectedName].  If [expectedPrefix] is supplied, verify that the type is
    * reached via the given prefix.  If [allowTypeParameters] is true, allow the
    * type reference to supply type parameters.  [expectedKind] is the kind of
    * object referenced.
    */
   void checkTypeRef(UnlinkedTypeRef typeRef, String absoluteUri,
       String relativeUri, String expectedName,
       {String expectedPrefix,
       bool allowTypeParameters: false,
-      PrelinkedReferenceKind expectedKind:
-          PrelinkedReferenceKind.classOrEnum}) {
+      PrelinkedReferenceKind expectedKind: PrelinkedReferenceKind.classOrEnum,
+      int expectedUnit: 0}) {
     expect(typeRef, new isInstanceOf<UnlinkedTypeRef>());
     expect(typeRef.paramReference, 0);
     int index = typeRef.reference;
     UnlinkedReference reference = unlinked.references[index];
     PrelinkedReference referenceResolution = lib.references[index];
     if (absoluteUri == null) {
       expect(referenceResolution.dependency, 0);
     } else {
       checkDependency(referenceResolution.dependency, absoluteUri, relativeUri);
     }
     if (!allowTypeParameters) {
       expect(typeRef.typeArguments, isEmpty);
     }
     if (expectedName == null) {
       expect(reference.name, isEmpty);
     } else {
       expect(reference.name, expectedName);
     }
     if (checkAstDerivedData) {
       if (expectedPrefix == null) {
         expect(reference.prefix, 0);
         expect(unlinked.prefixes[reference.prefix].name, isEmpty);
       } else {
         expect(reference.prefix, isNot(0));
         expect(unlinked.prefixes[reference.prefix].name, expectedPrefix);
       }
     }
     expect(referenceResolution.kind, expectedKind);
+    expect(referenceResolution.unit, expectedUnit);
   }
 
   /**
    * Verify that the given [typeRef] represents a reference to an unresolved
    * type.
    */
   void checkUnresolvedTypeRef(
       UnlinkedTypeRef typeRef, String expectedPrefix, String expectedName) {
     // When serializing from the element model, unresolved type refs lose their
     // name.
@@ skipping 1399 matching lines @@
   }
 
   test_type_reference_to_imported_part() {
     addNamedSource('/a.dart', 'library my.lib; part "b.dart";');
     addNamedSource('/b.dart', 'part of my.lib; class C {}');
     checkTypeRef(
         serializeTypeText('C',
             otherDeclarations: 'library my.lib; import "a.dart";'),
         absUri('/a.dart'),
         'a.dart',
-        'C');
+        'C',
+        expectedUnit: 1);
   }
 
   test_type_reference_to_imported_part_with_prefix() {
     addNamedSource('/a.dart', 'library my.lib; part "b.dart";');
     addNamedSource('/b.dart', 'part of my.lib; class C {}');
     checkTypeRef(
         serializeTypeText('p.C',
             otherDeclarations: 'library my.lib; import "a.dart" as p;'),
         absUri('/a.dart'),
         'a.dart',
         'C',
-        expectedPrefix: 'p');
+        expectedPrefix: 'p',
+        expectedUnit: 1);
   }
 
   test_type_reference_to_internal_class() {
     checkTypeRef(serializeTypeText('C', otherDeclarations: 'class C {}'), null,
         null, 'C');
   }
 
   test_type_reference_to_internal_class_alias() {
     checkTypeRef(
         serializeTypeText('C',
             otherDeclarations: 'class C = D with E; class D {} class E {}'),
         null,
         null,
         'C');
   }
 
   test_type_reference_to_internal_enum() {
     checkTypeRef(serializeTypeText('E', otherDeclarations: 'enum E { value }'),
         null, null, 'E');
   }
 
   test_type_reference_to_local_part() {
     addNamedSource('/a.dart', 'part of my.lib; class C {}');
     checkTypeRef(
         serializeTypeText('C',
             otherDeclarations: 'library my.lib; part "a.dart";'),
         null,
         null,
-        'C');
+        'C',
+        expectedUnit: 1);
   }
 
   test_type_reference_to_nonexistent_file_via_prefix() {
     UnlinkedTypeRef typeRef = serializeTypeText('p.C',
         otherDeclarations: 'import "foo.dart" as p;', allowErrors: true);
     checkUnresolvedTypeRef(typeRef, 'p', 'C');
   }
 
   test_type_reference_to_typedef() {
     checkTypeRef(serializeTypeText('F', otherDeclarations: 'typedef void F();'),
         null, null, 'F',
         expectedKind: PrelinkedReferenceKind.typedef);
   }
 
   test_type_unit_counts_unreferenced_units() {
     addNamedSource('/a.dart', 'library a; part "b.dart"; part "c.dart";');
     addNamedSource('/b.dart', 'part of a;');
     addNamedSource('/c.dart', 'part of a; class C {}');
     UnlinkedTypeRef typeRef =
         serializeTypeText('C', otherDeclarations: 'import "a.dart";');
     // The referenced unit should be 2, since unit 0 is a.dart and unit 1 is
     // b.dart.  a.dart and b.dart are counted even though nothing is imported
     // from them.
-    checkTypeRef(typeRef, absUri('/a.dart'), 'a.dart', 'C');
+    checkTypeRef(typeRef, absUri('/a.dart'), 'a.dart', 'C', expectedUnit: 2);
   }
 
   test_type_unresolved() {
     UnlinkedTypeRef typeRef = serializeTypeText('Foo', allowErrors: true);
     checkUnresolvedTypeRef(typeRef, null, 'Foo');
   }
 
   test_typedef_name() {
     UnlinkedTypedef type = serializeTypedefText('typedef F();');
     expect(type.name, 'F');
@@ skipping 107 matching lines @@
         serializeClassText('class C { static int i; }').fields[0];
     expect(variable.isStatic, isTrue);
   }
 
   test_variable_type() {
     UnlinkedVariable variable =
         serializeVariableText('int i;', variableName: 'i');
     checkTypeRef(variable.type, 'dart:core', 'dart:core', 'int');
   }
 }