Search for extension getter/method/setter using FindElement.

scheglov · commit-bot@chromium.org · commit bd2b0504f1fa · 2019-07-23T22:36:02.000Z
R=brianwilkerson@google.com, pquitslund@google.com Change-Id: Ifa4cb6ac3ab43ae473807f19488871678247f974 Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/110134 Reviewed-by: Phil Quitslund <pquitslund@google.com> Reviewed-by: Brian Wilkerson <brianwilkerson@google.com> Commit-Queue: Konstantin Shcheglov <scheglov@google.com>
diff --git a/pkg/analyzer/lib/src/test_utilities/find_element.dart b/pkg/analyzer/lib/src/test_utilities/find_element.dart
@@ -171,6 +171,13 @@ class FindElement {
       findIn(enum_.accessors);
     }
 
+    for (var extension_ in unitElement.extensions) {
+      if (of != null && extension_.name != of) {
+        continue;
+      }
+      findIn(extension_.accessors);
+    }
+
     for (var class_ in unitElement.types) {
       if (of != null && class_.name != of) {
         continue;
@@ -275,6 +282,13 @@ class FindElement {
       }
     }
 
+    for (var extension_ in unitElement.extensions) {
+      if (of != null && extension_.name != of) {
+        continue;
+      }
+      findIn(extension_.methods);
+    }
+
     for (var class_ in unitElement.types) {
       if (of != null && class_.name != of) {
         continue;
@@ -372,6 +386,13 @@ class FindElement {
       }
     }
 
+    for (var extension_ in unitElement.extensions) {
+      if (of != null && extension_.name != of) {
+        continue;
+      }
+      findIn(extension_.accessors);
+    }
+
     for (var class_ in unitElement.types) {
       if (of != null && class_.name != of) {
         continue;
