Quick Fix for returning Future from 'async' functions.

pkg/analysis_server/test/services/correction/fix_test.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.
 
 library test.services.correction.fix;
 
 import 'package:analysis_server/src/protocol.dart' hide AnalysisError;
 import 'package:analysis_server/src/services/correction/fix.dart';
 import 'package:analyzer/file_system/file_system.dart';
 import 'package:analyzer/source/package_map_resolver.dart';
@@ skipping 1718 matching lines @@
   print(0)
 }
 ''');
     assertHasFix(FixKind.INSERT_SEMICOLON, '''
 main() {
   print(0);
 }
 ''');
   }
 
+  void test_illegalAsyncReturnType_asyncLibrary_import() {
+    errorFilter = (AnalysisError error) {
+      return error.errorCode == StaticTypeWarningCode.ILLEGAL_ASYNC_RETURN_TYPE;
+    };
+    resolveTestUnit('''
+library main;
+int main() async {
+}
+''');
+    assertHasFix(FixKind.REPLACE_RETURN_TYPE_FUTURE, '''
+library main;
+import 'dart:async';
+Future<int> main() async {
+}
+''');
+  }
+
+  void test_illegalAsyncReturnType_asyncLibrary_usePrefix() {
+    errorFilter = (AnalysisError error) {
+      return error.errorCode == StaticTypeWarningCode.ILLEGAL_ASYNC_RETURN_TYPE;
+    };
+    resolveTestUnit('''
+import 'dart:async' as al;
+int main() async {
+}
+''');
+    assertHasFix(FixKind.REPLACE_RETURN_TYPE_FUTURE, '''
+import 'dart:async' as al;
+al.Future<int> main() async {
+}
+''');
+  }
+
+  void test_illegalAsyncReturnType_complexTypeName() {
+    errorFilter = (AnalysisError error) {
+      return error.errorCode == StaticTypeWarningCode.ILLEGAL_ASYNC_RETURN_TYPE;
+    };
+    resolveTestUnit('''
+import 'dart:async';
+List<int> main() async {
+}
+''');
+    assertHasFix(FixKind.REPLACE_RETURN_TYPE_FUTURE, '''
+import 'dart:async';
+Future<List<int>> main() async {
+}
+''');
+  }
+
+  void test_illegalAsyncReturnType_void() {

[Brian Wilkerson, 2015/03/09 21:05:35]

Perhaps a test case containing:

import 'dart:async';
main() async {
}

to test the absence of a return type.

[scheglov, 2015/03/09 21:34:37]

Well, we don't report any error in this case.
So, we don't need a fix :-)

+    errorFilter = (AnalysisError error) {
+      return error.errorCode == StaticTypeWarningCode.ILLEGAL_ASYNC_RETURN_TYPE;
+    };
+    resolveTestUnit('''
+import 'dart:async';
+void main() async {
+}
+''');
+    assertHasFix(FixKind.REPLACE_RETURN_TYPE_FUTURE, '''
+import 'dart:async';
+Future main() async {
+}
+''');
+  }
+
   void test_importLibraryPackage_withClass() {
     _configureMyPkg('''
 library my_lib;
 class Test {}
 ''');
     // try to find a fix
     resolveTestUnit('''
 main() {
   Test test = null;
 }
@@ skipping 1332 matching lines @@
 
   List<Position> _findResultPositions(List<String> searchStrings) {
     List<Position> positions = <Position>[];
     for (String search in searchStrings) {
       int offset = resultCode.indexOf(search);
       positions.add(new Position(testFile, offset));
     }
     return positions;
   }
 }