Optimize top-level element declarations search.

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 47 matching lines @@
           expectedRelationship == recordedRelation.relationship) &&
       (expectedLocation == null ||
           _equalsLocation(recordedRelation.location, expectedLocation));
 }
 
 
 @reflectiveTest
 class DartUnitContributorTest extends AbstractSingleUnitTest {
   IndexStore store = new MockIndexStore();
   List<RecordedRelation> recordedRelations = <RecordedRelation>[];
+  List<Element> recordedTopElements = <Element>[];
 
   CompilationUnitElement importedUnit({int index: 0}) {
     List<ImportElement> imports = testLibraryElement.imports;
     return imports[index].importedLibrary.definingCompilationUnit;
   }
 
   void setUp() {
     super.setUp();
     when(store.aboutToIndexDart(context, anyObject)).thenReturn(true);
     when(
         store.recordRelationship(
             anyObject,
             anyObject,
             anyObject)).thenInvoke(
                 (Element element, Relationship relationship, Location location) {
       recordedRelations.add(
           new RecordedRelation(element, relationship, location));
     });
+    when(store.recordTopDeclaration(anyObject)).thenInvoke((Element element) {
+      recordedTopElements.add(element);
+    });
   }
 
   void test_definesClass() {
     _indexTestUnit('class A {}');
     // prepare elements
     ClassElement classElement = findElement("A");
     // verify
-    _assertDefinesTopLevelElement(
-        IndexConstants.DEFINES,
-        _expectedLocation(classElement, 'A {}'));
+    _assertDefinesTopLevelElement(classElement);
   }
 
   void test_definesClassAlias() {
     _indexTestUnit('''
 class Mix {}
 class MyClass = Object with Mix;''');
     // prepare elements
     Element classElement = findElement("MyClass");
     // verify
-    _assertDefinesTopLevelElement(
-        IndexConstants.DEFINES,
-        _expectedLocation(classElement, 'MyClass ='));
+    _assertDefinesTopLevelElement(classElement);
   }
 
   void test_definesClassEnum() {
     _indexTestUnit('enum MyEnum {A, B, c}');
     // prepare elements
     ClassElement classElement = findElement("MyEnum");
     // verify
-    _assertDefinesTopLevelElement(
-        IndexConstants.DEFINES,
-        _expectedLocation(classElement, 'MyEnum {'));
+    _assertDefinesTopLevelElement(classElement);
   }
 
   void test_definesFunction() {
     _indexTestUnit('myFunction() {}');
     // prepare elements
     FunctionElement functionElement = findElement("myFunction");
     // verify
-    _assertDefinesTopLevelElement(
-        IndexConstants.DEFINES,
-        _expectedLocation(functionElement, 'myFunction() {}'));
+    _assertDefinesTopLevelElement(functionElement);
   }
 
 
   void test_definesFunctionType() {
     _indexTestUnit('typedef MyFunction(int p);');
     // prepare elements
     FunctionTypeAliasElement typeAliasElement = findElement("MyFunction");
     // verify
-    _assertDefinesTopLevelElement(
-        IndexConstants.DEFINES,
-        _expectedLocation(typeAliasElement, 'MyFunction(int p);'));
+    _assertDefinesTopLevelElement(typeAliasElement);
   }
 
   void test_definesVariable() {
     _indexTestUnit('var myVar = 42;');
     // prepare elements
     VariableElement varElement = findElement("myVar");
     // verify
-    _assertDefinesTopLevelElement(
-        IndexConstants.DEFINES,
-        _expectedLocation(varElement, 'myVar = 42;'));
+    _assertDefinesTopLevelElement(varElement);
   }
 
   void test_forIn() {
     _indexTestUnit('''
 main() {
   for (var v in []) {
   }
 }''');
     // prepare elements
     Element mainElement = findElement("main");
@@ skipping 1484 matching lines @@
 
   void test_nullUnit() {
     indexDartUnit(store, context, null);
   }
 
   void test_nullUnitElement() {
     CompilationUnit unit = new CompilationUnit(null, null, [], [], null);
     indexDartUnit(store, context, unit);
   }
 
-  void _assertDefinesTopLevelElement(Relationship relationship,
-      ExpectedLocation expectedLocation) {
-    _assertRecordedRelation(testLibraryElement, relationship, expectedLocation);
-    _assertRecordedRelation(
-        UniverseElement.INSTANCE,
-        relationship,
-        expectedLocation);
+  void _assertDefinesTopLevelElement(Element element) {
+    ExpectedLocation location = new ExpectedLocation(element, element.nameOffset,
+        element.name.length, false, true);
+    _assertRecordedRelation(testLibraryElement, IndexConstants.DEFINES, location);
+    expect(recordedTopElements, contains(element));
   }
 
   /**
    * Asserts that [recordedRelations] has no item with the specified properties.
    */
   void _assertNoRecordedRelation(Element element, Relationship relationship,
       ExpectedLocation location) {
     for (RecordedRelation recordedRelation in recordedRelations) {
       if (_equalsRecordedRelation(
           recordedRelation,
@@ skipping 98 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" : ""})';
   }
 }