Make "server.shutdown" exit the analysis server.

pkg/analysis_server/test/mocks.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 mocks;
 
 import 'dart:async';
 import 'dart:io';
 
 @MirrorsUsed(targets: 'mocks', override: '*')
@@ skipping 115 matching lines @@
 /**
  * A mock [ServerCommunicationChannel] for testing [AnalysisServer].
  */
 class MockServerChannel implements ServerCommunicationChannel {
   StreamController<Request> requestController = new StreamController<Request>();
   StreamController<Response> responseController = new StreamController<Response>();
   StreamController<Notification> notificationController = new StreamController<Notification>(sync: true);
 
   List<Response> responsesReceived = [];
   List<Notification> notificationsReceived = [];
+  bool _closed = false;
 
   MockServerChannel() {
   }
 
   @override
   void listen(void onRequest(Request request), {Function onError, void onDone()}) {
     requestController.stream.listen(onRequest, onError: onError, onDone: onDone);
   }
 
   @override
   void sendNotification(Notification notification) {
+    // Don't deliver notifications after the connection is closed.
+    if (_closed) {
+      return;
+    }
     notificationsReceived.add(notification);
     // Wrap send notification in future to simulate websocket
     // TODO(scheglov) ask Dan why and decide what to do
 //    new Future(() => notificationController.add(notification));
     notificationController.add(notification);
   }
 
   /**
    * Simulate request/response pair.
    */
   Future<Response> sendRequest(Request request) {
+    // No further requests should be sent after the connection is closed.
+    if (_closed) {
+      throw new Exception('sendRequest after connection closed');
+    }
     // Wrap send request in future to simulate websocket
     new Future(() => requestController.add(request));
     return waitForResponse(request);
   }
 
   @override
   void sendResponse(Response response) {
+    // Don't deliver responses after the connection is closed.
+    if (_closed) {
+      return;
+    }
     responsesReceived.add(response);
     // Wrap send response in future to simulate websocket
     new Future(() => responseController.add(response));
   }
 
   void expectMsgCount({responseCount: 0, notificationCount: 0}) {
     expect(responsesReceived, hasLength(responseCount));
     expect(notificationsReceived, hasLength(notificationCount));
   }
 
   Future<Response> waitForResponse(Request request) {
     String id = request.id;
     pumpEventQueue().then((_) {
       responseController.addError(new NoResponseException(request));
     });
     return responseController.stream.firstWhere((response) {
       return response.id == id;
     });
   }
+
+  @override
+  void close() {
+    _closed = true;
+  }
 }
 
 /**
  * A mock [AnalysisContext] for testing [AnalysisServer].
  */
 class MockAnalysisContext extends Mock implements AnalysisContext {
   noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }
 
 /**
@@ skipping 132 matching lines @@
   /**
    * Dependency list that will be returned by the next call to [computePackageMap].
    */
   Set<String> dependencies = new Set<String>();
 
   @override
   PackageMapInfo computePackageMap(resource.Folder folder) {
     return new PackageMapInfo(packageMap, dependencies);
   }
 }