Issue 60683. Fix for UnimplementedError from ElementImpl2.enclosingElement2, when GenericFunctionTypeElementImpl2.

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

Author: scheglov

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

@@ -1947,10 +1947,6 @@ abstract class ElementImpl2 implements Element {
   @override
   String get displayName => name3 ?? '<unnamed>';
 
-  @override
-  // TODO(augmentations): implement enclosingElement2
-  Element? get enclosingElement2 => throw UnimplementedError();
-
   @override
   List<Fragment> get fragments {
     return [
@@ -4075,6 +4071,9 @@ class GenericFunctionTypeElementImpl2 extends FunctionTypedElementImpl2
   @override
   String? get documentationComment => _wrappedElement.documentationComment;
 
+  @override
+  Element? get enclosingElement2 => firstFragment.enclosingFragment?.element;
+
   @override
   GenericFunctionTypeFragmentImpl get firstFragment => _wrappedElement;
 

pkg/analyzer/lib/src/test_utilities/find_node.dart

@@ -101,6 +101,8 @@ class FindNode {
   FunctionTypedFormalParameter get singleFunctionTypedFormalParameter =>
       _single();
 
+  GenericFunctionType get singleGenericFunctionType => _single();
+
   GenericTypeAlias get singleGenericTypeAlias => _single();
 
   GuardedPattern get singleGuardedPattern => _single();

pkg/analyzer/test/src/dart/resolution/generic_function_type_test.dart

@@ -3,6 +3,7 @@
 // BSD-style license that can be found in the LICENSE file.
 
 import 'package:analyzer/dart/ast/ast.dart';
+import 'package:test/expect.dart';
 import 'package:test_reflective_loader/test_reflective_loader.dart';
 
 import 'context_collection_resolution.dart';
@@ -47,6 +48,18 @@ int Function(int a)? y;
 ''');
   }
 
+  test_element_enclosingElement2() async {
+    await assertNoErrorsInCode('''
+void f(
+  void Function() a,
+) {}
+''');
+
+    var node = findNode.singleGenericFunctionType;
+    var element = node.declaredFragment!.element;
+    expect(element.enclosingElement2, same(result.libraryElement2));
+  }
+
   test_metadata_typeParameter() async {
     await assertNoErrorsInCode(r'''
 const a = 42;