Implementation of the 'search.findMemberDeclarations' API.

pkg/analysis_server/test/search/member_declarations_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.search.member_declarations;
+
+import 'dart:async';
+
+import 'package:analysis_server/src/computer/element.dart';
+import 'package:analysis_server/src/constants.dart';
+import 'package:analysis_server/src/protocol.dart';
+import 'package:analysis_server/src/search/search_result.dart';
+import 'package:analysis_testing/reflective_tests.dart';
+import 'package:unittest/unittest.dart';
+
+import 'abstract_search_domain.dart';
+
+
+main() {
+  groupSep = ' | ';
+  group('findMemberDeclarations', () {
+    runReflectiveTests(MemberDeclarationsTest);
+  });
+}
+
+
+@ReflectiveTestCase()
+class MemberDeclarationsTest extends AbstractSearchDomainTest {
+  void assertHasDeclaration(ElementKind kind, String className) {
+    result = findTopLevelResult(kind, className);
+    if (result == null) {
+      fail('Not found: kind=$kind in="$className"\nin\n' + results.join('\n'));
+    }
+  }
+
+  void assertNoDeclaration(ElementKind kind, String className) {
+    result = findTopLevelResult(kind, className);
+    if (result != null) {
+      fail('Unexpected: kind=$kind in="$className"\nin\n' + results.join('\n'));
+    }
+  }
+
+  Future findMemberDeclarations(String name) {
+    return waitForTasksFinished().then((_) {
+      Request request = new Request('0', SEARCH_FIND_MEMBER_DECLARATIONS);
+      request.setParameter(NAME, name);
+      Response response = handleSuccessfulRequest(request);
+      searchId = response.getResult(ID);
+      results.clear();
+      return waitForSearchResults();
+    });
+  }
+
+  SearchResult findTopLevelResult(ElementKind kind, String enclosingClass) {
+    for (SearchResult result in results) {
+      Element element = result.path[0];
+      Element clazz = result.path[1];
+      if (element.kind == kind && clazz.name == enclosingClass) {
+        return result;
+      }
+    }
+    return null;
+  }
+
+  test_methodField() {
+    addTestFile('''
+class A {
+  foo() {}
+  bar() {}
+}
+class B {
+  int foo;
+}
+''');
+    return findMemberDeclarations('foo').then((_) {
+      expect(results, hasLength(2));
+      assertHasDeclaration(ElementKind.METHOD, 'A');
+      assertHasDeclaration(ElementKind.FIELD, 'B');
+    });
+  }
+
+  test_methodGetter() {
+    addTestFile('''
+class A {
+  foo() {}
+  bar() {}
+}
+class B {
+  get foo => null;
+}
+''');
+    return findMemberDeclarations('foo').then((_) {
+      expect(results, hasLength(2));
+      assertHasDeclaration(ElementKind.METHOD, 'A');
+      assertHasDeclaration(ElementKind.GETTER, 'B');
+    });
+  }
+
+  test_methodGetterSetter() {
+    addTestFile('''
+class A {
+  foo() {}
+  bar() {}
+}
+class B {
+  get foo => null;
+  set foo(x) {}
+}
+''');
+    return findMemberDeclarations('foo').then((_) {
+      expect(results, hasLength(3));
+      assertHasDeclaration(ElementKind.METHOD, 'A');
+      assertHasDeclaration(ElementKind.GETTER, 'B');
+      assertHasDeclaration(ElementKind.SETTER, 'B');
+    });
+  }
+
+  test_methodMethod() {
+    addTestFile('''
+class A {
+  foo() {}
+  bar() {}
+}
+class B {
+  foo() {}
+}
+''');
+    return findMemberDeclarations('foo').then((_) {
+      expect(results, hasLength(2));
+      assertHasDeclaration(ElementKind.METHOD, 'A');
+      assertHasDeclaration(ElementKind.METHOD, 'B');
+    });
+  }
+
+  test_methodSetter() {
+    addTestFile('''
+class A {
+  foo() {}
+  bar() {}
+}
+class B {
+  set foo(x) {}
+}
+''');
+    return findMemberDeclarations('foo').then((_) {
+      expect(results, hasLength(2));
+      assertHasDeclaration(ElementKind.METHOD, 'A');
+      assertHasDeclaration(ElementKind.SETTER, 'B');
+    });
+  }
+}