Add plugin support for getAssists

pkg/analysis_server/test/edit/assists_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.edit.assists;
+import 'dart:async';
 
 import 'package:analysis_server/plugin/protocol/protocol.dart';
 import 'package:analysis_server/src/edit/edit_domain.dart';
+import 'package:analysis_server/src/plugin/plugin_manager.dart';
+import 'package:analyzer_plugin/protocol/protocol.dart' as plugin;
+import 'package:analyzer_plugin/protocol/protocol_generated.dart' as plugin;
+import 'package:analyzer_plugin/src/protocol/protocol_internal.dart' as plugin;
 import 'package:plugin/manager.dart';
 import 'package:test/test.dart';
 import 'package:test_reflective_loader/test_reflective_loader.dart';
 
 import '../analysis_abstract.dart';
 
 main() {
   defineReflectiveSuite(() {
     defineReflectiveTests(AssistsTest);
   });
@@ skipping 11 matching lines @@
   prepareAssistsAt(int offset, int length) async {
     Request request =
         new EditGetAssistsParams(testFile, offset, length).toRequest('0');
     Response response = await waitResponse(request);
     var result = new EditGetAssistsResult.fromResponse(response);
     changes = result.assists;
   }
 
   @override
   void setUp() {
+    enableNewAnalysisDriver = true;
     super.setUp();
     createProject();
     ExtensionManager manager = new ExtensionManager();
     manager.processPlugins([server.serverPlugin]);
     handler = new EditDomainHandler(server);
   }
 
+  test_fromPlugins() async {
+    PluginInfo info = new PluginInfo('a', 'b', 'c', null, null);
+    String message = 'From a plugin';
+    plugin.PrioritizedSourceChange change = new plugin.PrioritizedSourceChange(
+        5,
+        new plugin.SourceChange(message, edits: <plugin.SourceFileEdit>[
+          new plugin.SourceFileEdit('', 0,
+              edits: <plugin.SourceEdit>[new plugin.SourceEdit(0, 0, 'x')])
+        ]));
+    plugin.EditGetAssistsResult result = new plugin.EditGetAssistsResult(
+        <plugin.PrioritizedSourceChange>[change]);
+    pluginManager.broadcastResults = <PluginInfo, Future<plugin.Response>>{
+      info: new Future.value(result.toResponse('-'))
+    };
+
+    addTestFile('main() {}');
+    await waitForTasksFinished();
+    await prepareAssists('in(');
+    _assertHasChange(message, 'xmain() {}');
+  }
+
   test_removeTypeAnnotation() async {
     addTestFile('''
 main() {
   int v = 1;
 }
 ''');
     await waitForTasksFinished();
     await prepareAssists('v =');
     _assertHasChange(
         'Remove type annotation',
@@ skipping 50 matching lines @@
       if (change.message == message) {
         String resultCode =
             SourceEdit.applySequence(testCode, change.edits[0].edits);
         expect(resultCode, expectedCode);
         return;
       }
     }
     fail("Expected to find |$message| in\n" + changes.join('\n'));
   }
 }