More tests for creating undefined methods and functions.

pkg/analysis_services/lib/src/correction/fix.dart

 // Copyright (c) 2014, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 // This code was auto-generated, is not intended to be edited, and is subject to
 // significant change. Please see the README file for more information.
 
 library services.src.correction.fix;
 
 import 'package:analysis_services/correction/change.dart';
@@ skipping 149 matching lines @@
       _addFix_importLibrary_withType();
       _addFix_importLibrary_withTopLevelVariable();
     }
     if (errorCode == StaticTypeWarningCode.INSTANCE_ACCESS_TO_STATIC_MEMBER) {
       _addFix_useStaticAccess_method();
       _addFix_useStaticAccess_property();
     }
     if (errorCode == StaticTypeWarningCode.INVOCATION_OF_NON_FUNCTION) {
       _addFix_removeParentheses_inGetterInvocation();
     }
-//    if (identical(errorCode, StaticTypeWarningCode.UNDEFINED_FUNCTION)) {
-//      _addFix_importLibrary_withFunction();
-//      _addFix_undefinedFunction_useSimilar();
-//      _addFix_undefinedFunction_create();
-//    }
+    if (errorCode == StaticTypeWarningCode.UNDEFINED_FUNCTION) {
+      _addFix_importLibrary_withFunction();
+      _addFix_undefinedFunction_useSimilar();
+      _addFix_undefinedFunction_create();
+    }
     if (errorCode == StaticTypeWarningCode.UNDEFINED_GETTER) {
       _addFix_createFunction_forFunctionType();
     }
     if (errorCode == HintCode.UNDEFINED_METHOD ||
         errorCode == StaticTypeWarningCode.UNDEFINED_METHOD) {
       _addFix_undefinedMethod_useSimilar();
       _addFix_undefinedMethod_create();
       _addFix_undefinedFunction_create();
     }
     // done
@@ skipping 399 matching lines @@
   }
 
   void
       _addFix_createMissingOverrides(List<ExecutableElement> missingOverrides) {
     // sort by name
     missingOverrides.sort((Element firstElement, Element secondElement) {
       return compareStrings(
           firstElement.displayName,
           secondElement.displayName);
     });
-    // TODO
     ClassDeclaration targetClass = node.parent as ClassDeclaration;
     int insertOffset = targetClass.end - 1;
     SourceBuilder sb = new SourceBuilder(file, insertOffset);
     // add elements
     bool isFirst = true;
     for (ExecutableElement missingOverride in missingOverrides) {
       if (!isFirst || !targetClass.members.isEmpty) {
         sb.append(eol);
       }
       _addFix_createMissingOverrides_single(sb, targetClass, missingOverride);
@@ skipping 444 matching lines @@
 //        if (closestName != null) {
 //          _addFix(
 //              FixKind.CHANGE_TO,
 //              [closestName]);
 //        }
 //      }
 //    }
   }
 
   void _addFix_undefinedFunction_create() {
-    // TODO(scheglov) implement
-//    // should be the name of the invocation
-//    if (node is SimpleIdentifier && node.parent is MethodInvocation) {
-//    } else {
-//      return;
-//    }
-//    String name = (node as SimpleIdentifier).name;
-//    MethodInvocation invocation = node.parent as MethodInvocation;
-//    // function invocation has no target
-//    Expression target = invocation.realTarget;
-//    if (target != null) {
-//      return;
-//    }
-//    // prepare environment
-//    int insertOffset;
-//    String sourcePrefix;
-//    AstNode enclosingMember =
-//        node.getAncestor((node) => node is CompilationUnitMember);
-//    insertOffset = enclosingMember.end;
-//    sourcePrefix = "${eol}${eol}";
-//    // build method source
-//    SourceBuilder sb = new SourceBuilder.con1(insertOffset);
-//    {
-//      sb.append(sourcePrefix);
-//      // may be return type
-//      {
-//        DartType type =
-//            _addFix_undefinedMethod_create_getReturnType(invocation);
-//        if (type != null) {
-//          String typeSource = utils.getTypeSource2(type);
-//          if (typeSource != "dynamic") {
-//            sb.startPosition("RETURN_TYPE");
-//            sb.append(typeSource);
-//            sb.endPosition();
-//            sb.append(" ");
-//          }
-//        }
-//      }
-//      // append name
-//      {
-//        sb.startPosition("NAME");
-//        sb.append(name);
-//        sb.endPosition();
-//      }
-//      _addFix_undefinedMethod_create_parameters(sb, invocation.argumentList);
-//      sb.append(") {${eol}}");
-//    }
-//    // insert source
-//    _addInsertEdit(insertOffset, sb.toString());
-//    // add linked positions
-//    _addLinkedPosition("NAME", sb, SourceRangeFactory.rangeNode(node));
-//    _addLinkedPositions(sb);
-//    // add proposal
-//    _addFix(FixKind.CREATE_FUNCTION, [name]);
+    // should be the name of the invocation
+    if (node is SimpleIdentifier && node.parent is MethodInvocation) {
+    } else {
+      return;
+    }
+    String name = (node as SimpleIdentifier).name;
+    MethodInvocation invocation = node.parent as MethodInvocation;
+    // function invocation has no target
+    Expression target = invocation.realTarget;
+    if (target != null) {
+      return;
+    }
+    // prepare environment
+    int insertOffset;
+    String sourcePrefix;
+    AstNode enclosingMember =
+        node.getAncestor((node) => node is CompilationUnitMember);
+    insertOffset = enclosingMember.end;
+    sourcePrefix = "${eol}${eol}";
+    // build method source
+    SourceBuilder sb = new SourceBuilder(file, insertOffset);
+    {
+      sb.append(sourcePrefix);
+      // append return type
+      {
+        DartType type = _inferReturnType(invocation);
+        _appendType(sb, type, 'RETURN_TYPE');
+      }
+      // append name
+      {
+        sb.startPosition("NAME");
+        sb.append(name);
+        sb.endPosition();
+      }
+      _addFix_undefinedMethod_create_parameters(sb, invocation.argumentList);
+      sb.append(") {${eol}}");
+    }
+    // insert source
+    _insertBuilder(sb);
+    _addLinkedPosition3('NAME', sb, rf.rangeNode(node));
+    // add proposal
+    _addFix(FixKind.CREATE_FUNCTION, [name]);
   }
 
   void _addFix_undefinedFunction_useSimilar() {
     // TODO(scheglov) implement
 //    if (node is SimpleIdentifier) {
 //      String name = (node as SimpleIdentifier).name;
 //      QuickFixProcessorImpl_ClosestElementFinder finder =
 //          new QuickFixProcessorImpl_ClosestElementFinder(
 //              name,
 //              new Predicate_QuickFixProcessorImpl_addFix_undefinedFunction_useSimilar());
@@ skipping 31 matching lines @@
       String sourceSuffix;
       bool staticModifier = false;
       Expression target = invocation.realTarget;
       if (target == null) {
         targetSource = source;
         ClassMember enclosingMember =
             node.getAncestor((node) => node is ClassMember);
         staticModifier = _inStaticContext();
         prefix = utils.getNodePrefix(enclosingMember);
         insertOffset = enclosingMember.end;
-        sourcePrefix = "${eol}${prefix}${eol}";
+        sourcePrefix = "${eol}${eol}";
         sourceSuffix = "";
       } else {
         // prepare target interface type
         DartType targetType = target.bestType;
         if (targetType is! InterfaceType) {
           return;
         }
         ClassElement targetElement = targetType.element as ClassElement;
         targetSource = targetElement.source;
         // may be static
@@ skipping 176 matching lines @@
    */
   void _addInsertEdit(int offset, String text) {
     Edit edit = new Edit(offset, 0, text);
     edits.add(edit);
   }
 
   /**
    * Adds a single linked position to [groupId].
    */
   void _addLinkedPosition(String groupId, SourceRange range) {
-    _addLinkedPosition2(
-        groupId,
-        new Position(file, range.offset, range.length));
+    Position position = new Position(file, range.offset, range.length);
+    LinkedPositionGroup group = _getLinkedPosition(groupId);
+    group.addPosition(position);
   }
 
   /**
-   * Adds a single linked position to [groupId].
-   */
-  void _addLinkedPosition2(String groupId, Position position) {
-    LinkedPositionGroup group = linkedPositionGroups[groupId];
-    if (group == null) {
-      group = new LinkedPositionGroup(groupId);
-      linkedPositionGroups[groupId] = group;
-    }
-    group.add(position);
-  }
-
-  /**
    * Adds a single linked position to [groupId].
    */
   void _addLinkedPosition3(String groupId, SourceBuilder sb,
       SourceRange range) {
     if (sb.offset < range.offset) {
       int delta = sb.length;
       range = range.getTranslated(delta);
     }
     _addLinkedPosition(groupId, range);
   }
@@ skipping 258 matching lines @@
         }
         String valueSource =
             sourceContent.substring(valueRange.offset, valueRange.end);
         defaultSourceMap[parameter] = valueSource;
       }
     }
     return defaultSourceMap;
   }
 
   /**
+   * Returns an existing or just added [LinkedPositionGroup] with [groupId].
+   */
+  LinkedPositionGroup _getLinkedPosition(String groupId) {
+    LinkedPositionGroup group = linkedPositionGroups[groupId];
+    if (group == null) {
+      group = new LinkedPositionGroup(groupId);
+      linkedPositionGroups[groupId] = group;
+    }
+    return group;
+  }
+
+  /**
    * Returns `true` if [node] is in static context.
    */
   bool _inStaticContext() {
     // constructor initializer cannot reference "this"
     if (node.getAncestor((node) => node is ConstructorInitializer) != null) {
       return true;
     }
     // field initializer cannot reference "this"
     if (node.getAncestor((node) => node is FieldDeclaration) != null) {
       return true;
@@ skipping 144 matching lines @@
 //  }
 
   /**
    * Inserts the given [SourceBuilder] at its offset.
    */
   void _insertBuilder(SourceBuilder builder) {
     String text = builder.toString();
     _addInsertEdit(builder.offset, text);
     // add linked positions
     builder.linkedPositionGroups.forEach((LinkedPositionGroup group) {
+      LinkedPositionGroup fixGroup = _getLinkedPosition(group.id);
       group.positions.forEach((Position position) {
-        _addLinkedPosition2(group.id, position);
+        fixGroup.addPosition(position);
+      });
+      group.proposals.forEach((String proposal) {
+        fixGroup.addProposal(proposal);
       });
     });
   }
 
   _ConstructorLocation
       _prepareNewConstructorLocation(ClassDeclaration classDeclaration) {
     List<ClassMember> members = classDeclaration.members;
     // find the last field/constructor
     ClassMember lastFieldOrConstructor = null;
     for (ClassMember member in members) {
@@ skipping 80 matching lines @@
         return parent.prefix == node;
       }
     }
     return false;
   }
 }
 
 
 /**
  * Describes the location for a newly created [ConstructorDeclaration].
- *
- * TODO(scheglov) rename
  */
 class _ConstructorLocation {
   final String _prefix;
   final int _offset;
   final String _suffix;
 
   _ConstructorLocation(this._prefix, this._offset, this._suffix);
 }