Index and search of import prefixes.

pkg/analysis_server/test/services/index/dart_index_contributor_test.dart

 // Copyright (c) 2014, 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.services.src.index.dart_index_contributor;
 
 import 'package:analysis_server/src/services/index/index.dart';
 import 'package:analysis_server/src/services/index/index_contributor.dart';
 import 'package:analysis_server/src/services/index/index_store.dart';
 import 'package:analyzer/src/generated/ast.dart';
@@ skipping 24 matching lines @@
       expected.isQualified,
       expected.isResolved);
 }
 
 
 /**
  * Returns `true` if the [actual] location the expected properties.
  */
 bool _equalsLocationProperties(Location actual, Element expectedElement,
     int expectedOffset, int expectedLength, bool isQualified, bool isResolved) {
-  return expectedElement == actual.element &&
+  return (expectedElement == null || expectedElement == actual.element) &&
       expectedOffset == actual.offset &&
       expectedLength == actual.length &&
       isQualified == actual.isQualified &&
       isResolved == actual.isResolved;
 }
 
 
 bool _equalsRecordedRelation(RecordedRelation recordedRelation,
     Element expectedElement, Relationship expectedRelationship,
     ExpectedLocation expectedLocation) {
   return expectedElement == recordedRelation.element &&
-      expectedRelationship == recordedRelation.relationship &&
+      (expectedRelationship == null ||
+          expectedRelationship == recordedRelation.relationship) &&
       (expectedLocation == null ||
           _equalsLocation(recordedRelation.location, expectedLocation));
 }
 
 
 @ReflectiveTestCase()
 class DartUnitContributorTest extends AbstractSingleUnitTest {
   IndexStore store = new MockIndexStore();
   List<RecordedRelation> recordedRelations = <RecordedRelation>[];
 
@@ skipping 1262 matching lines @@
     // prepare elements
     Element mainElement = findElement('main');
     Element element = findElement('p');
     // verify
     _assertRecordedRelation(
         element,
         IndexConstants.IS_REFERENCED_BY,
         _expectedLocation(mainElement, 'p: 1'));
   }
 
+  void test_isReferencedBy_PrefixElement() {
+    _indexTestUnit('''
+import 'dart:async' as ppp;
+main() {
+  ppp.Future a;
+  ppp.Stream b;
+}
+''');
+    // prepare elements
+    PrefixElement element = findNodeElementAtString('ppp;');
+    Element elementA = findElement('a');
+    Element elementB = findElement('b');
+    // verify
+    _assertRecordedRelation(
+        element,
+        IndexConstants.IS_REFERENCED_BY,
+        _expectedLocation(elementA, 'ppp.Future'));
+    _assertRecordedRelation(
+        element,
+        IndexConstants.IS_REFERENCED_BY,
+        _expectedLocation(elementB, 'ppp.Stream'));
+    _assertNoRecordedRelation(element, null, _expectedLocation(null, 'ppp;'));
+  }
+
   void test_isReferencedBy_TopLevelVariableElement() {
     addSource('/lib.dart', '''
 library lib;
 var V;
 ''');
     _indexTestUnit('''
 import 'lib.dart' show V; // imp
 import 'lib.dart' as pref;
 main() {
   pref.V = 5; // q
@@ skipping 370 matching lines @@
 
   RecordedRelation(this.element, this.relationship, this.location);
 
   @override
   String toString() {
     return 'RecordedRelation(element=$element; relationship=$relationship; '
         'location=$location; flags=' '${location.isQualified ? "Q" : ""}'
         '${location.isResolved ? "R" : ""})';
   }
 }