Elements. Migrate a few summary tests.

Change-Id: I52791ad27681de7b1507925c9598b824b1ebb7c1
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/396568
Reviewed-by: Brian Wilkerson <[email protected]>
Commit-Queue: Konstantin Shcheglov <[email protected]>

Author: scheglov

pkg/analyzer/analyzer_use_new_elements.txt

@@ -227,12 +227,8 @@ test/src/dart/resolution/super_formal_parameter_test.dart
 test/src/services/top_level_declarations_test.dart
 test/src/summary/element_text.dart
 test/src/summary/elements/class_test.dart
-test/src/summary/elements/const_test.dart
 test/src/summary/elements/enum_test.dart
 test/src/summary/elements/extension_type_test.dart
-test/src/summary/elements/library_export_test.dart
-test/src/summary/elements/library_fragment_test.dart
-test/src/summary/elements/library_import_test.dart
 test/src/summary/elements/metadata_test.dart
 test/src/summary/elements/since_sdk_version_test.dart
 test/src/summary/elements/top_level_function_test.dart

pkg/analyzer/lib/src/dart/element/element.dart

@@ -799,7 +799,7 @@ class CompilationUnitElementImpl extends UriReferencedElementImpl
   @override
   List<ExtensionElement2> get accessibleExtensions2 {
     return scope.accessibleExtensions
-        .map((element) => element.augmentation as ExtensionElement2)
+        .map((element) => element.augmented as ExtensionElement2)
         .toList();
   }
 
@@ -9186,8 +9186,10 @@ class PrefixElementImpl2 extends ElementImpl2 implements PrefixElement2 {
   String? get name3 => firstFragment.name2;
 
   @override
-  // TODO(scheglov): implement scope
-  Scope get scope => throw UnimplementedError();
+  Scope get scope {
+    var libraryFragment = firstFragment.libraryFragment;
+    return LibraryFragmentScope(libraryFragment);
+  }
 
   @override
   T? accept2<T>(ElementVisitor2<T> visitor) {

pkg/analyzer/test/src/summary/elements/const_test.dart

@@ -663,14 +663,14 @@ library
 ''');
     var x = library.definingCompilationUnit.topLevelVariables[0];
     var xExpr = x.constantInitializer as InstanceCreationExpression;
-    var xType = xExpr.constructorName.staticElement!.returnType;
+    var xType = xExpr.constructorName.element!.returnType;
     _assertTypeStr(
       xType,
       'C<int>',
     );
     var y = library.definingCompilationUnit.topLevelVariables[0];
     var yExpr = y.constantInitializer as InstanceCreationExpression;
-    var yType = yExpr.constructorName.staticElement!.returnType;
+    var yType = yExpr.constructorName.element!.returnType;
     _assertTypeStr(yType, 'C<int>');
   }
 

pkg/analyzer/test/src/summary/elements/library_export_test.dart

@@ -827,8 +827,11 @@ library
           firstFragment: <testLibraryFragment>::@class::B::@constructor::new
           superConstructor: package:test/foo.dart::<fragment>::@class::A::@constructor::new#element
 ''');
-    var typeA = library.definingCompilationUnit.getClass('B')!.supertype!;
-    expect(typeA.element.source.shortName, 'foo.dart');
+
+    var typeA = library.getClass2('B')!.supertype!;
+    var fragmentA = typeA.element3.firstFragment;
+    var sourceA = fragmentA.libraryFragment.source;
+    expect(sourceA.shortName, 'foo.dart');
   }
 
   test_exportImport_configurations_useFirst() async {
@@ -896,8 +899,11 @@ library
           firstFragment: <testLibraryFragment>::@class::B::@constructor::new
           superConstructor: package:test/foo_io.dart::<fragment>::@class::A::@constructor::new#element
 ''');
-    var typeA = library.definingCompilationUnit.getClass('B')!.supertype!;
-    expect(typeA.element.source.shortName, 'foo_io.dart');
+
+    var typeA = library.getClass2('B')!.supertype!;
+    var fragmentA = typeA.element3.firstFragment;
+    var sourceA = fragmentA.libraryFragment.source;
+    expect(sourceA.shortName, 'foo_io.dart');
   }
 
   test_exportImport_configurations_useSecond() async {
@@ -965,8 +971,11 @@ library
           firstFragment: <testLibraryFragment>::@class::B::@constructor::new
           superConstructor: package:test/foo_html.dart::<fragment>::@class::A::@constructor::new#element
 ''');
-    var typeA = library.definingCompilationUnit.getClass('B')!.supertype!;
-    expect(typeA.element.source.shortName, 'foo_html.dart');
+
+    var typeA = library.getClass2('B')!.supertype!;
+    var fragmentA = typeA.element3.firstFragment;
+    var sourceA = fragmentA.libraryFragment.source;
+    expect(sourceA.shortName, 'foo_html.dart');
   }
 
   test_exports() async {