Add Isolate.onExit.

tests/isolate/ondone_test.dart

Index: tests/isolate/ondone_test.dart
diff --git a/tests/isolate/ondone_test.dart b/tests/isolate/ondone_test.dart
new file mode 100644
index 0000000000000000000000000000000000000000..af990ad31e8e3d05cd950bcbe23519efadfed95e
--- /dev/null
+++ b/tests/isolate/ondone_test.dart
@@ -0,0 +1,94 @@
+// Copyright (c) 2014, the Dart project authors.  Please see the AUTHORS file
+// for details. All rights reserved. Use of this source is governed by a
+// BSD-style license that can be found in the LICENSE file.
+
+import "dart:isolate";
+import "dart:async";
+import "package:async_helper/async_helper.dart";
+
+void isomain(SendPort replyPort) {
+  RawReceivePort port = new RawReceivePort();
+  port.handler = (v) {
+    if (v == 0) {
+      // Shut down when receiving the 0 message.
+      port.close();
+    } else {
+      replyPort.send(v);
+    }
+  };
+  replyPort.send(port.sendPort);
+}
+
+
+void main() {
+  testExit();
+  testCancelExit();
+}
+
+void testExit() {
+  bool mayComplete = false;
+  asyncStart();
+  var completer = new Completer();  // Completed by first reply from isolate.
+  RawReceivePort reply = new RawReceivePort(completer.complete);
+  RawReceivePort onExitPort;
+  onExitPort = new RawReceivePort((_) {
+    reply.close();
+    onExitPort.close();
+    if (!mayComplete) throw "COMPLETED EARLY";
+    asyncEnd();
+  });
+  Isolate.spawn(isomain, reply.sendPort).then((Isolate isolate) {
+    isolate.addOnExitListener(onExitPort.sendPort);
+    return completer.future;
+  }).then((echoPort) {
+    int counter = 4;
+    reply.handler = (v) {
+      if (v != counter) throw "WRONG REPLY";
+      if (v == 0) throw "REPLY INSTEAD OF SHUTDOWN";
+      counter--;
+      mayComplete = (counter == 0);
+      echoPort.send(counter);
+    };
+    echoPort.send(counter);
+  });
+}
+
+
+void testCancelExit() {
+  bool mayComplete = false;
+  asyncStart();
+  var completer = new Completer();  // Completed by first reply from isolate.
+  RawReceivePort reply = new RawReceivePort(completer.complete);
+  RawReceivePort onExitPort2 = new RawReceivePort((_) {
+    throw "RECEIVED EXIT MESSAGE";
+  });
+  RawReceivePort onExitPort1;
+  onExitPort1 = new RawReceivePort((_) {
+    reply.close();
+    onExitPort1.close();
+    if (!mayComplete) throw "COMPLETED EARLY";
+    new Timer(const Duration(milliseconds: 0), () {
+      onExitPort2.close();
+      asyncEnd();
+    });
+  });
+  Isolate.spawn(isomain, reply.sendPort).then((Isolate isolate) {
+    isolate.addOnExitListener(onExitPort2.sendPort);
+    isolate.addOnExitListener(onExitPort1.sendPort);
+    return completer.future.then((echoPort) {
+      int counter = 4;
+      reply.handler = (v) {
+        if (v != counter) throw "WRONG REPLY";
+        if (v == 0) throw "REPLY INSTEAD OF SHUTDOWN";
+        counter--;
+        mayComplete = (counter == 0);
+        if (counter == 1) {
+          // Remove listener 2, keep listener 1.
+          isolate.removeOnExitListener(onExitPort2.sendPort);
+        }
+        echoPort.send(counter);
+      };
+      echoPort.send(counter);
+    });
+  });
+}