[DAS] Fixes ambiguous extension access fix for method

[email protected]

Fixes #59543

Change-Id: I84c73c6315c401bc5f8d925f190d395354329102
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/396741
Reviewed-by: Samuel Rawlins <[email protected]>
Reviewed-by: Phil Quitslund <[email protected]>
Commit-Queue: Phil Quitslund <[email protected]>
Auto-Submit: Felipe Morschel <[email protected]>

Author: FMorschel

pkg/analysis_server/lib/src/services/correction/dart/add_extension_override.dart

@@ -18,8 +18,14 @@ class AddExtensionOverride extends MultiCorrectionProducer {
     var node = this.node;
     if (node is! SimpleIdentifier) return const [];
     var parent = node.parent;
-    if (parent is! PropertyAccess) return const [];
-    var target = parent.target;
+    Expression? target;
+    if (parent is MethodInvocation) {
+      target = parent.target;
+    } else if (parent is PropertyAccess) {
+      target = parent.target;
+    } else {
+      return const [];
+    }
     if (target == null) return const [];
     var dartFixContext = context.dartFixContext;
     if (dartFixContext == null) return const [];

pkg/analysis_server/test/src/services/correction/fix/add_extension_override_test.dart

@@ -20,6 +20,39 @@ class AddExtensionOverrideTest extends FixProcessorTest {
   @override
   FixKind get kind => DartFixKind.ADD_EXTENSION_OVERRIDE;
 
+  Future<void> test_method() async {
+    await resolveTestCode('''
+extension E on int {
+  void foo() {}
+}
+extension E2 on int {
+  void foo() {}
+}
+f() {
+  0.foo();
+}
+''');
+    await assertHasFix('''
+extension E on int {
+  void foo() {}
+}
+extension E2 on int {
+  void foo() {}
+}
+f() {
+  E(0).foo();
+}
+''', matchFixMessage: "Add an extension override for 'E'");
+
+    await assertHasFixesWithoutApplying(
+      expectedNumberOfFixesForKind: 2,
+      matchFixMessages: [
+        "Add an extension override for 'E'",
+        "Add an extension override for 'E2'",
+      ],
+    );
+  }
+
   Future<void> test_no_name() async {
     await resolveTestCode('''
 extension E on int {