Implement searchMemberDeclarations() / searchTopLevelDeclarations().

pkg/analysis_server/test/services/search/search_engine2_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.search.search_engine2;
 
 import 'dart:async';
 
 import 'package:analysis_server/src/services/index2/index2.dart';
 import 'package:analysis_server/src/services/search/search_engine.dart';
@@ skipping 75 matching lines @@
 //    ClassElement elementC = findElement('C');
 //    var expected = [
 //      _expectId(elementA, MatchKind.DECLARATION, 'A extends T'),
 //      _expectId(elementB, MatchKind.DECLARATION, 'B extends A'),
 //      _expectId(elementC, MatchKind.DECLARATION, 'C implements B')
 //    ];
 //    return searchEngine.searchAllSubtypes(element).then((matches) {
 //      _assertMatches(matches, expected);
 //    });
 //  }
-//
-//  Future test_searchMemberDeclarations() {
-//    _indexTestUnit('''
-//class A {
-//  test() {}
-//}
-//class B {
-//  int test = 1;
-//  main() {
-//    int test = 2;
-//  }
-//}
-//''');
-//    ClassElement elementA = findElement('A');
-//    ClassElement elementB = findElement('B');
-//    var expected = [
-//      _expectId(elementA.methods[0], MatchKind.DECLARATION, 'test() {}'),
-//      _expectId(elementB.fields[0], MatchKind.DECLARATION, 'test = 1;')
-//    ];
-//    return searchEngine.searchMemberDeclarations('test').then((matches) {
-//      _assertMatches(matches, expected);
-//    });
-//  }
-//
+
+  test_searchMemberDeclarations() async {
+    _indexTestUnit('''
+class A {
+  test() {}
+}
+class B {
+  int test = 1;
+  main() {
+    int test = 2;
+  }
+}
+''');
+    ClassElement elementA = findElement('A');
+    ClassElement elementB = findElement('B');
+    var expected = [
+      _expectId(elementA.methods[0], MatchKind.DECLARATION, 'test() {}'),
+      _expectId(elementB.fields[0], MatchKind.DECLARATION, 'test = 1;')
+    ];
+    List<SearchMatch> matches =
+        await searchEngine.searchMemberDeclarations('test');
+    _assertMatches(matches, expected);
+  }
+
 //  Future test_searchMemberReferences() {
 //    _indexTestUnit('''
 //class A {
 //  var test; // A
 //  mainA() {
 //    test(); // a-inv-r-nq
 //    test = 1; // a-write-r-nq
 //    test += 2; // a-read-write-r-nq
 //    print(test); // a-read-r-nq
 //  }
@@ skipping 523 matching lines @@
 //    ClassElement elementC = findElement('C');
 //    var expected = [
 //      _expectId(elementA, MatchKind.REFERENCE, 'T {} // A'),
 //      _expectId(elementB, MatchKind.REFERENCE, 'T; // B'),
 //      _expectId(elementC, MatchKind.REFERENCE, 'T {} // C')
 //    ];
 //    return searchEngine.searchSubtypes(element).then((matches) {
 //      _assertMatches(matches, expected);
 //    });
 //  }
-//
-//  Future test_searchTopLevelDeclarations() {
-//    _indexTestUnit('''
-//class A {} // A
-//class B = Object with A;
-//typedef C();
-//D() {}
-//var E = null;
-//class NoMatchABCDE {}
-//''');
-//    Element topA = findElement('A');
-//    Element topB = findElement('B');
-//    Element topC = findElement('C');
-//    Element topD = findElement('D');
-//    Element topE = findElement('E');
-//    var expected = [
-//      _expectId(topA, MatchKind.DECLARATION, 'A {} // A'),
-//      _expectId(topB, MatchKind.DECLARATION, 'B ='),
-//      _expectId(topC, MatchKind.DECLARATION, 'C()'),
-//      _expectId(topD, MatchKind.DECLARATION, 'D() {}'),
-//      _expectId(topE, MatchKind.DECLARATION, 'E = null')
-//    ];
-//    return _verifyTopLevelDeclarations('^[A-E]\$', expected);
-//  }
+
+  test_searchTopLevelDeclarations() async {
+    _indexTestUnit('''
+class A {} // A
+class B = Object with A;
+typedef C();
+D() {}
+var E = null;
+class NoMatchABCDE {}
+''');
+    Element topA = findElement('A');
+    Element topB = findElement('B');
+    Element topC = findElement('C');
+    Element topD = findElement('D');
+    Element topE = findElement('E');
+    var expected = [
+      _expectId(topA, MatchKind.DECLARATION, 'A {} // A'),
+      _expectId(topB, MatchKind.DECLARATION, 'B ='),
+      _expectId(topC, MatchKind.DECLARATION, 'C()'),
+      _expectId(topD, MatchKind.DECLARATION, 'D() {}'),
+      _expectId(topE, MatchKind.DECLARATION, 'E = null')
+    ];
+    List<SearchMatch> matches =
+        await searchEngine.searchTopLevelDeclarations(r'^[A-E]$');
+    _assertMatches(matches, expected);
+  }
 
   ExpectedMatch _expectId(Element element, MatchKind kind, String search,
       {int length, bool isResolved: true, bool isQualified: false}) {
     int offset = findOffset(search);
     if (length == null) {
       length = getLeadingIdentifierLength(search);
     }
     return new ExpectedMatch(element, kind, offset, length,
         isResolved: isResolved, isQualified: isQualified);
   }
@@ skipping 12 matching lines @@
     resolveTestUnit(code);
     index.indexUnit(testUnit);
   }
 
   Future _verifyReferences(
       Element element, List<ExpectedMatch> expectedMatches) async {
     List<SearchMatch> matches = await searchEngine.searchReferences(element);
     _assertMatches(matches, expectedMatches);
   }
 
-//  Future _verifyTopLevelDeclarations(
-//      String pattern, List<ExpectedMatch> expectedMatches) {
-//    return searchEngine
-//        .searchTopLevelDeclarations(pattern)
-//        .then((List<SearchMatch> matches) {
-//      _assertMatches(matches, expectedMatches);
-//    });
-//  }
-
   static void _assertMatches(
       List<SearchMatch> matches, List<ExpectedMatch> expectedMatches) {
     expect(matches, unorderedEquals(expectedMatches));
   }
 }