Capture the request time for performance data and support forced shutdown

pkg/analysis_server/test/src/plugin/plugin_manager_test.dart

 // Copyright (c) 2017, 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.
 
 import 'dart:async';
 import 'dart:io' as io;
 
 import 'package:analysis_server/src/plugin/notification_manager.dart';
 import 'package:analysis_server/src/plugin/plugin_manager.dart';
 import 'package:analyzer/context/context_root.dart';
@@ skipping 72 matching lines @@
       fail('Expected a StateError');
     } on StateError {
       // Expected.
     }
   }
 
   test_stop_notRunning() {
     expect(() => plugin.stop(), throwsA(new isInstanceOf<StateError>()));
   }
 
-  test_stop_running() {
+  test_stop_running() async {
     PluginSession session = new PluginSession(plugin);
     TestServerCommunicationChannel channel =
         new TestServerCommunicationChannel(session);
     plugin.currentSession = session;
-    plugin.stop();
+    await plugin.stop();
     expect(plugin.currentSession, isNull);
     expect(channel.sentRequests, hasLength(1));
     expect(channel.sentRequests[0].method, 'plugin.shutdown');
   }
 }
 
 @reflectiveTest
 class DiscoveredPluginInfoTest {
   MemoryResourceProvider resourceProvider;
   TestNotificationManager notificationManager;
@@ skipping 50 matching lines @@
       fail('Expected a StateError');
     } on StateError {
       // Expected.
     }
   }
 
   test_stop_notRunning() {
     expect(() => plugin.stop(), throwsA(new isInstanceOf<StateError>()));
   }
 
-  test_stop_running() {
+  test_stop_running() async {
     PluginSession session = new PluginSession(plugin);
     TestServerCommunicationChannel channel =
         new TestServerCommunicationChannel(session);
     plugin.currentSession = session;
-    plugin.stop();
+    await plugin.stop();
     expect(plugin.currentSession, isNull);
     expect(channel.sentRequests, hasLength(1));
     expect(channel.sentRequests[0].method, 'plugin.shutdown');
   }
 }
 
 @reflectiveTest
 class PluginManagerFromDiskTest extends PluginTestSupport {
   String byteStorePath = '/byteStore';
   PluginManager manager;
@@ skipping 299 matching lines @@
       fail('Expected a StateError to be thrown');
     } on StateError {
       // Expected behavior
     }
   }
 
   test_stop_notRunning() {
     expect(() => session.stop(), throwsA(new isInstanceOf<StateError>()));
   }
 
-  void test_stop_running() {
+  test_stop_running() async {
     TestServerCommunicationChannel channel =
         new TestServerCommunicationChannel(session);
-    session.stop();
+    await session.stop();
     expect(channel.sentRequests, hasLength(1));
     expect(channel.sentRequests[0].method, 'plugin.shutdown');
   }
 }
 
 /**
  * A class designed to be used as a superclass for test classes that define
  * tests that require plugins to be created on disk.
  */
 abstract class PluginTestSupport {
@@ skipping 156 matching lines @@
     notifications.add(notification);
   }
 
   @override
   noSuchMethod(Invocation invocation) {
     fail('Unexpected invocation of ${invocation.memberName}');
   }
 }
 
 class TestServerCommunicationChannel implements ServerCommunicationChannel {
+  final PluginSession session;
   int closeCount = 0;
   List<Request> sentRequests = <Request>[];
 
-  TestServerCommunicationChannel(PluginSession session) {
+  TestServerCommunicationChannel(this.session) {
     session.channel = this;
   }
 
   @override
   void close() {
     closeCount++;
   }
 
+  void kill() {
+    fail('Unexpected invocation of kill');
+  }
+
   @override
   void listen(void onResponse(Response response),
       void onNotification(Notification notification),
       {Function onError, void onDone()}) {
     fail('Unexpected invocation of listen');
   }
 
   @override
   void sendRequest(Request request) {
     sentRequests.add(request);
+    if (request.method == 'plugin.shutdown') {
+      session.handleOnDone();
+    }
   }
 }