Fix isolate lifecycle handling in service library

runtime/observatory/test/test_helper.dart

 // Copyright (c) 2013, 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_helper;
 
 import 'dart:async';
 import 'dart:convert';
 import 'dart:io';
 import 'package:unittest/unittest.dart';
 import 'package:observatory/service_io.dart';
 
 // This invocation should set up the state being tested.
 const String _TESTEE_MODE_FLAG = "--testee-mode";
 
 class _TestLauncher {
   Process process;
   final List<String> args;
 
   _TestLauncher() : args = ['--enable-vm-service:0',
                             Platform.script.toFilePath(),
                             _TESTEE_MODE_FLAG] {}
 
-  Future<int> launch() {
+  Future<int> launch(bool pause_on_exit) {
     String dartExecutable = Platform.executable;
     var fullArgs = [];
+    if (pause_on_exit == true) {
+      fullArgs.add('--pause-isolates-on-exit');
+    }
     fullArgs.addAll(Platform.executableArguments);
     fullArgs.addAll(args);
-    print('** Launching $fullArgs');
+    print('** Launching $dartExecutable ${fullArgs.join(' ')}');
     return Process.start(dartExecutable, fullArgs).then((p) {
 
       Completer completer = new Completer();
       process = p;
       var portNumber;
       var blank;
       var first = true;
       process.stdout.transform(UTF8.decoder)
                     .transform(new LineSplitter()).listen((line) {
         if (line.startsWith('Observatory listening on http://')) {
@@ skipping 24 matching lines @@
     });
   }
 
   void requestExit() {
     print('** Requesting script to exit.');
     process.stdin.add([32, 13, 10]);
   }
 }
 
 typedef Future IsolateTest(Isolate isolate);
+typedef Future VMTest(VM vm);
 
 /// Runs [tests] in sequence, each of which should take an [Isolate] and
 /// return a [Future]. Code for setting up state can run before and/or
 /// concurrently with the tests. Uses [mainArgs] to determine whether
 /// to run tests or testee in this invokation of the script.
 void runIsolateTests(List<String> mainArgs,
                      List<IsolateTest> tests,
                      {void testeeBefore(),
-                      void testeeConcurrent()}) {
+                      void testeeConcurrent(),
+                      bool pause_on_exit}) {
   if (mainArgs.contains(_TESTEE_MODE_FLAG)) {
     if (testeeBefore != null) {
       testeeBefore();
     }
     print(''); // Print blank line to signal that we are ready.
     if (testeeConcurrent != null) {
       testeeConcurrent();
     }
     // Wait until signaled from spawning test.
     stdin.first.then((_) => exit(0));
   } else {
     var process = new _TestLauncher();
-    process.launch().then((port) {
+    process.launch(pause_on_exit).then((port) {
       String addr = 'ws://localhost:$port/ws';
+      var testIndex = 0;
+      var totalTests = tests.length - 1;
+      var name = Platform.script.pathSegments.last;
       new WebSocketVM(new WebSocketVMTarget(addr)).load()
           .then((VM vm) => vm.isolates.first.load())
-          .then((Isolate isolate) =>
-              Future.forEach(tests, (test) => test(isolate)))
-          .then((_) => exit(0));
+          .then((Isolate isolate) => Future.forEach(tests, (test) {
+            print('Running $name [$testIndex/$totalTests]');
+            testIndex++;
+            return test(isolate);
+          })).then((_) => exit(0));
     });
   }
 }
+
+
+// Cancel the subscription and complete the completer when finished processing
+// events.
+typedef void ServiceEventHandler(ServiceEvent event,
+                                 StreamSubscription subscription,
+                                 Completer completer);
+
+Future processServiceEvents(VM vm, ServiceEventHandler handler) {
+  Completer completer = new Completer();
+  var subscription;
+  subscription = vm.events.stream.listen((ServiceEvent event) {
+    handler(event, subscription, completer);
+  });
+  return completer.future;
+}
+
+
+/// Runs [tests] in sequence, each of which should take an [Isolate] and
+/// return a [Future]. Code for setting up state can run before and/or
+/// concurrently with the tests. Uses [mainArgs] to determine whether
+/// to run tests or testee in this invokation of the script.
+Future runVMTests(List<String> mainArgs,
+                  List<VMTest> tests,
+                  {Future testeeBefore(),
+                   Future testeeConcurrent(),
+                   bool pause_on_exit}) async {
+  if (mainArgs.contains(_TESTEE_MODE_FLAG)) {
+    if (testeeBefore != null) {
+      await testeeBefore();
+    }
+    print(''); // Print blank line to signal that we are ready.
+    if (testeeConcurrent != null) {
+      await testeeConcurrent();
+    }
+    // Wait until signaled from spawning test.
+    stdin.first.then((_) => exit(0));
+  } else {
+    var process = new _TestLauncher();
+    process.launch(pause_on_exit).then((port) async {
+      String addr = 'ws://localhost:$port/ws';
+      var testIndex = 0;
+      var totalTests = tests.length - 1;
+      var name = Platform.script.pathSegments.last;
+      new WebSocketVM(new WebSocketVMTarget(addr)).load()
+          .then((VM vm) => Future.forEach(tests, (test) {
+            print('Running $name [$testIndex/$totalTests]');
+            testIndex++;
+            return test(vm);
+          })).then((_) => exit(0));
+    });
+  }
+}