Implement search for local elements.

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 274 matching lines @@
 ''');
     FunctionElement element = findElement('test');
     Element mainElement = findElement('main');
     var expected = [
       _expectId(mainElement, MatchKind.INVOCATION, 'test();'),
       _expectId(mainElement, MatchKind.REFERENCE, 'test);')
     ];
     await _verifyReferences(element, expected);
   }
 
+  test_searchReferences_FunctionElement_local() async {
+    _indexTestUnit('''
+main() {
+  test() {}
+  test();
+  print(test);
+}
+''');
+    FunctionElement element = findElement('test');
+    Element mainElement = findElement('main');
+    var expected = [
+      _expectId(mainElement, MatchKind.INVOCATION, 'test();'),
+      _expectId(mainElement, MatchKind.REFERENCE, 'test);')
+    ];
+    await _verifyReferences(element, expected);
+  }
+
   test_searchReferences_FunctionTypeAliasElement() async {
     _indexTestUnit('''
 typedef Test();
 main() {
   Test a;
   Test b;
 }
 ''');
     FunctionTypeAliasElement element = findElement('Test');
     Element aElement = findElement('a');
@@ skipping 27 matching lines @@
 //}
 //''');
 //    ImportElement element = testLibraryElement.imports[0];
 //    Element mainElement = findElement('main');
 //    var kind = MatchKind.REFERENCE;
 //    var expected = [
 //      _expectId(mainElement, kind, 'math.PI);', length: 'math.'.length)
 //    ];
 //    return _verifyReferences(element, expected);
 //  }
-//
-//  Future test_searchReferences_LabelElement() {
-//    _indexTestUnit('''
-//main() {
-//label:
-//  while (true) {
-//    if (true) {
-//      break label; // 1
-//    }
-//    break label; // 2
-//  }
-//}
-//''');
-//    LabelElement element = findElement('label');
-//    Element mainElement = findElement('main');
-//    var expected = [
-//      _expectId(mainElement, MatchKind.REFERENCE, 'label; // 1'),
-//      _expectId(mainElement, MatchKind.REFERENCE, 'label; // 2')
-//    ];
-//    return _verifyReferences(element, expected);
-//  }
-//
+
+  test_searchReferences_LabelElement() async {
+    _indexTestUnit('''
+main() {
+label:
+  while (true) {
+    if (true) {
+      break label; // 1
+    }
+    break label; // 2
+  }
+}
+''');
+    LabelElement element = findElement('label');
+    Element mainElement = findElement('main');
+    var expected = [
+      _expectId(mainElement, MatchKind.REFERENCE, 'label; // 1'),
+      _expectId(mainElement, MatchKind.REFERENCE, 'label; // 2')
+    ];
+    await _verifyReferences(element, expected);
+  }
+
 //  Future test_searchReferences_LibraryElement() {
 //    var codeA = 'part of lib; // A';
 //    var codeB = 'part of lib; // B';
 //    addSource('/unitA.dart', codeA);
 //    addSource('/unitB.dart', codeB);
 //    _indexTestUnit('''
 //library lib;
 //part 'unitA.dart';
 //part 'unitB.dart';
 //''');
 //    LibraryElement element = testLibraryElement;
 //    CompilationUnitElement elementA = element.parts[0];
 //    CompilationUnitElement elementB = element.parts[1];
 //    index.index(context, elementA.computeNode());
 //    index.index(context, elementB.computeNode());
 //    var expected = [
 //      new ExpectedMatch(elementA, MatchKind.REFERENCE,
 //          codeA.indexOf('lib; // A'), 'lib'.length),
 //      new ExpectedMatch(elementB, MatchKind.REFERENCE,
 //          codeB.indexOf('lib; // B'), 'lib'.length),
 //    ];
 //    return _verifyReferences(element, expected);
 //  }
-//
-//  Future test_searchReferences_LocalVariableElement() {
-//    _indexTestUnit('''
-//main() {
-//  var v;
-//  v = 1;
-//  v += 2;
-//  print(v);
-//  v();
-//}
-//''');
-//    LocalVariableElement element = findElement('v');
-//    Element mainElement = findElement('main');
-//    var expected = [
-//      _expectId(mainElement, MatchKind.WRITE, 'v = 1;'),
-//      _expectId(mainElement, MatchKind.READ_WRITE, 'v += 2;'),
-//      _expectId(mainElement, MatchKind.READ, 'v);'),
-//      _expectId(mainElement, MatchKind.INVOCATION, 'v();')
-//    ];
-//    return _verifyReferences(element, expected);
-//  }
 
+  test_searchReferences_LocalVariableElement() async {
+    _indexTestUnit('''
+main() {
+  var v;
+  v = 1;
+  v += 2;
+  print(v);
+  v();
+}
+''');
+    LocalVariableElement element = findElement('v');
+    Element mainElement = findElement('main');
+    var expected = [
+      _expectId(mainElement, MatchKind.WRITE, 'v = 1;'),
+      _expectId(mainElement, MatchKind.READ_WRITE, 'v += 2;'),
+      _expectId(mainElement, MatchKind.READ, 'v);'),
+      _expectId(mainElement, MatchKind.INVOCATION, 'v();')
+    ];
+    await _verifyReferences(element, expected);
+  }
+
   test_searchReferences_MethodElement() async {
     _indexTestUnit('''
 class A {
   m() {}
   main() {
     m(); // 1
     this.m(); // 2
     print(m); // 3
     print(this.m); // 4
   }
@@ skipping 20 matching lines @@
 }
 ''');
     MethodMember method = findNodeElementAtString('m(); // ref');
     Element mainElement = findElement('main');
     var expected = [
       _expectIdQ(mainElement, MatchKind.INVOCATION, 'm(); // ref')
     ];
     await _verifyReferences(method, expected);
   }
 
-//  Future test_searchReferences_ParameterElement() {
-//    _indexTestUnit('''
-//foo({p}) {
-//  p = 1;
-//  p += 2;
-//  print(p);
-//  p();
-//}
-//main() {
-//  foo(p: 42);
-//}
-//''');
-//    ParameterElement element = findElement('p');
-//    Element fooElement = findElement('foo');
-//    Element mainElement = findElement('main');
-//    var expected = [
-//      _expectId(fooElement, MatchKind.WRITE, 'p = 1;'),
-//      _expectId(fooElement, MatchKind.READ_WRITE, 'p += 2;'),
-//      _expectId(fooElement, MatchKind.READ, 'p);'),
-//      _expectId(fooElement, MatchKind.INVOCATION, 'p();'),
-//      _expectId(mainElement, MatchKind.REFERENCE, 'p: 42')
-//    ];
-//    return _verifyReferences(element, expected);
-//  }
-//
+  test_searchReferences_ParameterElement() async {
+    _indexTestUnit('''
+foo({p}) {
+  p = 1;
+  p += 2;
+  print(p);
+  p();
+}
+main() {
+  foo(p: 42);
+}
+''');
+    ParameterElement element = findElement('p');
+    Element fooElement = findElement('foo');
+    Element mainElement = findElement('main');
+    var expected = [
+      _expectId(fooElement, MatchKind.WRITE, 'p = 1;'),
+      _expectId(fooElement, MatchKind.READ_WRITE, 'p += 2;'),
+      _expectId(fooElement, MatchKind.READ, 'p);'),
+      _expectId(fooElement, MatchKind.INVOCATION, 'p();'),
+      _expectIdQ(mainElement, MatchKind.REFERENCE, 'p: 42')
+    ];
+    await _verifyReferences(element, expected);
+  }
+
 //  Future test_searchReferences_PrefixElement() {
 //    _indexTestUnit('''
 //import 'dart:async' as ppp;
 //main() {
 //  ppp.Future a;
 //  ppp.Stream b;
 //}
 //''');
 //    PrefixElement element = findNodeElementAtString('ppp;');
 //    Element elementA = findElement('a');
@@ skipping 72 matching lines @@
       _expectIdQ(main, MatchKind.WRITE, 'V = 1; // q'),
       _expectIdQ(main, MatchKind.READ, 'V); // q'),
       _expectIdQ(main, MatchKind.INVOCATION, 'V(); // q'),
       _expectId(main, MatchKind.WRITE, 'V = 1; // nq'),
       _expectId(main, MatchKind.READ, 'V); // nq'),
       _expectId(main, MatchKind.INVOCATION, 'V(); // nq'),
     ];
     await _verifyReferences(variable, expected);
   }
 
-//  Future test_searchReferences_TypeParameterElement() {
-//    _indexTestUnit('''
-//class A<T> {
-//  main(T a, T b) {}
-//}
-//''');
-//    TypeParameterElement element = findElement('T');
-//    Element aElement = findElement('a');
-//    Element bElement = findElement('b');
-//    var expected = [
-//      _expectId(aElement, MatchKind.REFERENCE, 'T a'),
-//      _expectId(bElement, MatchKind.REFERENCE, 'T b')
-//    ];
-//    return _verifyReferences(element, expected);
-//  }
+  test_searchReferences_TypeParameterElement() async {
+    _indexTestUnit('''
+class A<T> {
+  main(T a, T b) {}
+}
+''');
+    TypeParameterElement element = findElement('T');
+    Element aElement = findElement('a');
+    Element bElement = findElement('b');
+    var expected = [
+      _expectId(aElement, MatchKind.REFERENCE, 'T a'),
+      _expectId(bElement, MatchKind.REFERENCE, 'T b')
+    ];
+    await _verifyReferences(element, expected);
+  }
 
 //  Future test_searchSubtypes() {
 //    _indexTestUnit('''
 //class T {}
 //class A extends T {} // A
 //class B = Object with T; // B
 //class C implements T {} // C
 //''');
 //    ClassElement element = findElement('T');
 //    ClassElement elementA = findElement('A');
@@ skipping 71 matching lines @@
 //        .then((List<SearchMatch> matches) {
 //      _assertMatches(matches, expectedMatches);
 //    });
 //  }
 
   static void _assertMatches(
       List<SearchMatch> matches, List<ExpectedMatch> expectedMatches) {
     expect(matches, unorderedEquals(expectedMatches));
   }
 }