Replace AnalysisServerListener with Stream.

pkg/analysis_server/test/domain_execution_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.domain.execution;
 
 import 'package:analyzer/file_system/physical_file_system.dart';
 import 'package:analysis_server/src/analysis_server.dart';
 import 'package:analysis_server/src/constants.dart';
 import 'package:analysis_server/src/domain_execution.dart';
 import 'package:analysis_server/src/protocol.dart';
 import 'mock_sdk.dart';
 import 'package:unittest/unittest.dart';
 
 import 'mocks.dart';
 import 'operation/operation_queue_test.dart';
 import 'package:typed_mock/typed_mock.dart';
 import 'package:analyzer/src/generated/engine.dart';
 import 'package:analyzer/src/generated/source.dart';
+import 'dart:async';
 
 main() {
   group('ExecutionDomainHandler', () {
     AnalysisServer server;
     ExecutionDomainHandler handler;
 
     setUp(() {
       server = new AnalysisServer(
           new MockServerChannel(),
           PhysicalResourceProvider.INSTANCE,
@@ skipping 94 matching lines @@
 //      test('both file and uri', () {
 //        Request request =
 //            new ExecutionMapUriParams('xxx', file: '', uri: '').toRequest('5');
 //        Response response = handler.handleRequest(request);
 //        expect(response, isResponseFailure('5'));
 //      });
 //    });
 
     group('setSubscriptions', () {
       test('failure - invalid service name', () {
-        expect(handler.launchDataListener, isNull);
+        expect(handler.onAnalysisSubscription, isNull);
 
         Request request = new Request('0', EXECUTION_SET_SUBSCRIPTIONS, {
           SUBSCRIPTIONS: ['noSuchService']
         });
         Response response = handler.handleRequest(request);
         expect(response, isResponseFailure('0'));
-        expect(handler.launchDataListener, isNull);
+        expect(handler.onAnalysisSubscription, isNull);
       });
 
       test('success - setting and clearing', () {
-        expect(handler.launchDataListener, isNull);
+        expect(handler.onAnalysisSubscription, isNull);
 
         Request request = new ExecutionSetSubscriptionsParams(
             [ExecutionService.LAUNCH_DATA]).toRequest('0');
         Response response = handler.handleRequest(request);
         expect(response, isResponseSuccess('0'));
-        expect(handler.launchDataListener, isNotNull);
+        expect(handler.onAnalysisSubscription, isNotNull);
 
         request = new ExecutionSetSubscriptionsParams([]).toRequest('0');
         response = handler.handleRequest(request);
         expect(response, isResponseSuccess('0'));
-        expect(handler.launchDataListener, isNull);
+        expect(handler.onAnalysisSubscription, isNull);
       });
     });
-  });
 
-  group('LaunchDataNotificationListener', () {
-    test('success - setting and clearing', () {
+    test('onAnalysisComplete - success - setting and clearing', () {
       Source source1 = new TestSource('/a.dart');
       Source source2 = new TestSource('/b.dart');
       Source source3 = new TestSource('/c.dart');
       Source source4 = new TestSource('/d.dart');
       Source source5 = new TestSource('/e.html');
       Source source6 = new TestSource('/f.html');
       Source source7 = new TestSource('/g.html');
       Source source8 = new TestSource('/h.dart');
       Source source9 = new TestSource('/i.dart');
 
       AnalysisContext context = new AnalysisContextMock();
       when(context.launchableClientLibrarySources)
           .thenReturn([source1, source2]);
       when(context.launchableServerLibrarySources)
           .thenReturn([source2, source3]);
       when(context.librarySources).thenReturn([source4]);
       when(context.getHtmlFilesReferencing(anyObject))
           .thenReturn([source5, source6]);
       when(context.htmlSources).thenReturn([source7]);
       when(context.getLibrariesReferencedFromHtml(anyObject))
           .thenReturn([source8, source9]);
 
       AnalysisServer server = new AnalysisServerMock();
       when(server.getAnalysisContexts()).thenReturn([context]);
+      when(server.isAnalysisComplete()).thenReturn(false);
+
+      StreamController controller = new StreamController.broadcast(sync: true);
+      when(server.onAnalysisComplete).thenReturn(controller.stream);
+
+      ExecutionDomainHandler handler = new ExecutionDomainHandler(server);
+      Request request = new ExecutionSetSubscriptionsParams(
+          [ExecutionService.LAUNCH_DATA]).toRequest('0');
+      Response response = handler.handleRequest(request);
+
       bool notificationSent = false;
       when(server.sendNotification(anyObject))
           .thenInvoke((Notification notification) {
         ExecutionLaunchDataParams params =
             new ExecutionLaunchDataParams.fromNotification(notification);
 
         List<ExecutableFile> executables = params.executables;
         expect(executables.length, equals(3));
         expect(
             executables[0],
@@ skipping 16 matching lines @@
         Map<String, List<String>> htmlToDart = params.htmlToDart;
         expect(htmlToDart.length, equals(1));
         List<String> dartFiles = htmlToDart[source7.fullName];
         expect(dartFiles, isNotNull);
         expect(dartFiles.length, equals(2));
         expect(dartFiles[0], equals(source8.fullName));
         expect(dartFiles[1], equals(source9.fullName));
 
         notificationSent = true;
       });
-      LaunchDataNotificationListener listener =
-          new LaunchDataNotificationListener(server);
-      listener.analysisComplete();
+      controller.add(null);
       expect(notificationSent, isTrue);
     });
   });
 }
 
 /**
  * A [Source] that knows it's [fullName].
  */
 class TestSource implements Source {
   String fullName;
@@ skipping 29 matching lines @@
 
   @override
   bool matches(item, Map matchState) {
     if (item is! ExecutableFile) {
       return false;
     }
     ExecutableFile file = item;
     return item.file == expectedFile && item.kind == expectedKind;
   }
 }