commit edde03955b15c6a38c58fe2637dbc52439cb8686
author pq <pquitslund@google.com> Wed Jan 08 15:02:23 2025 -0800
committer Commit Queue <dart-scoped@luci-project-accounts.iam.gserviceaccount.com> Wed Jan 08 15:02:23 2025 -0800
tree 7d14909fcbdb9e56b91726ee6a28cb3e69d26f6a
parent 31d8adf1f3b9375e5a7af9fc81f8d28738d878e7

[element model] migrate `element_visitors_test`

Change-Id: I5eb336b52380de3ad49d0dd9bced1f0dc7fc1cb8
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/403720
Commit-Queue: Phil Quitslund <pquitslund@google.com>
Auto-Submit: Phil Quitslund <pquitslund@google.com>
Commit-Queue: Konstantin Shcheglov <scheglov@google.com>
Reviewed-by: Konstantin Shcheglov <scheglov@google.com>

pkg/analysis_server/analyzer_use_new_elements.txt

diff --git a/pkg/analysis_server/analyzer_use_new_elements.txt b/pkg/analysis_server/analyzer_use_new_elements.txt
index 13cef93..37a52e3 100644
--- a/pkg/analysis_server/analyzer_use_new_elements.txt
+++ b/pkg/analysis_server/analyzer_use_new_elements.txt
@@ -13,4 +13,3 @@
 lib/src/services/refactoring/legacy/rename_import.dart
 lib/src/services/search/element_visitors.dart
 test/services/refactoring/legacy/abstract_rename.dart
-test/services/search/element_visitors_test.dart

pkg/analysis_server/lib/src/services/search/element_visitors.dart

diff --git a/pkg/analysis_server/lib/src/services/search/element_visitors.dart b/pkg/analysis_server/lib/src/services/search/element_visitors.dart
index 5299084..f213e11 100644
--- a/pkg/analysis_server/lib/src/services/search/element_visitors.dart
+++ b/pkg/analysis_server/lib/src/services/search/element_visitors.dart
@@ -6,6 +6,7 @@
 import 'package:analyzer/dart/element/element2.dart';
 import 'package:analyzer/dart/element/visitor.dart';
 import 'package:analyzer/dart/element/visitor2.dart';
+import 'package:analyzer/src/utilities/extensions/element.dart';
 
 /// Return the [Element] that is either [root], or one of its direct or
 /// indirect children, and has the given [nameOffset].
@@ -22,6 +23,11 @@
   return null;
 }
 
+/// Return the [Element2] that is either [root], or one of its direct or
+/// indirect children, and has the given [nameOffset].
+Element2? findElementByNameOffset2(Element2? root, int nameOffset) =>
+    findElementByNameOffset(root.asElement, nameOffset).asElement2;
+
 /// Uses [processor] to visit all of the children of [element].
 /// If [processor] returns `true`, then children of a child are visited too.
 void visitChildren(Element element, BoolElementProcessor processor) {

pkg/analysis_server/test/services/search/element_visitors_test.dart

diff --git a/pkg/analysis_server/test/services/search/element_visitors_test.dart b/pkg/analysis_server/test/services/search/element_visitors_test.dart
index c1a2a98..4e69aac3 100644
--- a/pkg/analysis_server/test/services/search/element_visitors_test.dart
+++ b/pkg/analysis_server/test/services/search/element_visitors_test.dart
@@ -3,8 +3,7 @@
 // BSD-style license that can be found in the LICENSE file.
 
 import 'package:analysis_server/src/services/search/element_visitors.dart';
-import 'package:analyzer/dart/element/element.dart';
-import 'package:analyzer/src/utilities/extensions/element.dart';
+import 'package:analyzer/dart/element/element2.dart';
 import 'package:test/test.dart';
 import 'package:test_reflective_loader/test_reflective_loader.dart';
 
@@ -18,7 +17,7 @@
 
 @reflectiveTest
 class FindElementByNameOffsetTest extends AbstractSingleUnitTest {
-  Element get testUnitElement => testLibraryElement.asElement;
+  Element2 get testUnitElement => testLibraryElement;
 
   Future<void> test_class() async {
     await resolveTestCode(r'''
@@ -43,13 +42,13 @@
 class AAA {}
 class BBB {}
 ''');
-    expect(findElementByNameOffset(null, 0), isNull);
+    expect(findElementByNameOffset2(null, 0), isNull);
 
-    expect(findElementByNameOffset(testUnitElement, 0), isNull);
-    expect(findElementByNameOffset(testUnitElement, 1), isNull);
+    expect(findElementByNameOffset2(testUnitElement, 0), isNull);
+    expect(findElementByNameOffset2(testUnitElement, 1), isNull);
 
-    expect(findElementByNameOffset(testUnitElement, 5), isNull);
-    expect(findElementByNameOffset(testUnitElement, 7), isNull);
+    expect(findElementByNameOffset2(testUnitElement, 5), isNull);
+    expect(findElementByNameOffset2(testUnitElement, 7), isNull);
   }
 
   Future<void> test_topLevelVariable() async {
@@ -63,8 +62,8 @@
   }
 
   void _assertElement(int nameOffset, ElementKind kind, String name) {
-    var element = findElementByNameOffset(testUnitElement, nameOffset)!;
+    var element = findElementByNameOffset2(testUnitElement, nameOffset)!;
     expect(element.kind, kind);
-    expect(element.name, name);
+    expect(element.name3, name);
   }
 }