Big merge from experimental to bleeding edge.

tests/lib/async/stream_controller_test.dart

Index: tests/lib/async/stream_controller_test.dart
diff --git a/tests/lib/async/stream_controller_test.dart b/tests/lib/async/stream_controller_test.dart
new file mode 100644
index 0000000000000000000000000000000000000000..e06ec4ad512c92d2775001bac7ae6624c92d5c14
--- /dev/null
+++ b/tests/lib/async/stream_controller_test.dart
@@ -0,0 +1,359 @@
+// Copyright (c) 2011, 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.
+
+// Test the basic StreamController and StreamController.singleSubscription.
+import 'dart:async';
+import 'event_helper.dart';
+
+testController() {
+  // Test normal flow.
+  var c = new StreamController();
+  Events expectedEvents = new Events()
+      ..add(42)
+      ..add("dibs")
+      ..error("error!")
+      ..error("error too!")
+      ..close();
+  Events actualEvents = new Events.capture(c);
+  expectedEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test automatic unsubscription on error.
+  c = new StreamController();
+  expectedEvents = new Events()..add(42)..error("error");
+  actualEvents = new Events.capture(c, unsubscribeOnError: true);
+  Events sentEvents =
+      new Events()..add(42)..error("error")..add("Are you there?");
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test manual unsubscription.
+  c = new StreamController();
+  expectedEvents = new Events()..add(42)..error("error")..add(37);
+  actualEvents = new Events.capture(c, unsubscribeOnError: false);
+  expectedEvents.replay(c);
+  actualEvents.subscription.cancel();
+  c.add("Are you there");  // Not sent to actualEvents.
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test filter.
+  c = new StreamController();
+  expectedEvents = new Events()
+    ..add("a string")..add("another string")..close();
+  sentEvents = new Events()
+    ..add("a string")..add(42)..add("another string")..close();
+  actualEvents = new Events.capture(c.where((v) => v is String));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test map.
+  c = new StreamController();
+  expectedEvents = new Events()..add("abab")..error("error")..close();
+  sentEvents = new Events()..add("ab")..error("error")..close();
+  actualEvents = new Events.capture(c.mappedBy((v) => "$v$v"));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test handleError.
+  c = new StreamController();
+  expectedEvents = new Events()..add("ab")..error("[foo]");
+  sentEvents = new Events()..add("ab")..error("foo")..add("ab")..close();
+  actualEvents = new Events.capture(c.handleError((v) {
+        if (v.error is String) {
+          return new AsyncError("[${v.error}]",
+                                "other stack");
+        }
+      }), unsubscribeOnError: true);
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // reduce is tested asynchronously and therefore not in this file.
+
+  // Test expand
+  c = new StreamController();
+  sentEvents = new Events()..add(3)..add(2)..add(4)..close();
+  expectedEvents = new Events()..add(1)..add(2)..add(3)
+                               ..add(1)..add(2)
+                               ..add(1)..add(2)..add(3)..add(4)
+                               ..close();
+  actualEvents = new Events.capture(c.expand((v) {
+    var l = [];
+    for (int i = 0; i < v; i++) l.add(i + 1);
+    return l;
+  }));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test transform.
+  c = new StreamController();
+  sentEvents = new Events()..add("a")..error(42)..add("b")..close();
+  expectedEvents =
+      new Events()..error("a")..add(42)..error("b")..add("foo")..close();
+  actualEvents = new Events.capture(c.transform(new StreamTransformer.from(
+      onData: (v, s) { s.signalError(new AsyncError(v)); },
+      onError: (e, s) { s.add(e.error); },
+      onDone: (s) {
+        s.add("foo");
+        s.close();
+      })));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test multiple filters.
+  c = new StreamController();
+  sentEvents = new Events()..add(42)
+                           ..add("snugglefluffy")
+                           ..add(7)
+                           ..add("42")
+                           ..error("not FormatException")  // Unsubscribes.
+                           ..close();
+  expectedEvents = new Events()..add(42)..error("not FormatException");
+  actualEvents = new Events.capture(
+      c.where((v) => v is String)
+       .mappedBy((v) => int.parse(v))
+       .handleError((v) {
+          if (v.error is! FormatException) return v;
+        })
+       .where((v) => v > 10),
+      unsubscribeOnError: true);
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test subscription changes while firing.
+  c = new StreamController();
+  var sink = c.sink;
+  var stream = c.stream;
+  var counter = 0;
+  var subscription = stream.listen(null);
+  subscription.onData((data) {
+    counter += data;
+    subscription.cancel();
+    stream.listen((data) {
+      counter += 10 * data;
+    });
+    var subscription2 = stream.listen(null);
+    subscription2.onData((data) {
+      counter += 100 * data;
+      if (data == 4) subscription2.cancel();
+    });
+  });
+  sink.add(1); // seen by stream 1
+  sink.add(2); // seen by stream 10 and 100
+  sink.add(3); // -"-
+  sink.add(4); // -"-
+  sink.add(5); // seen by stream 10
+  Expect.equals(1 + 20 + 200 + 30 + 300 + 40 + 400 + 50, counter);
+}
+
+testSingleController() {
+  // Test normal flow.
+  var c = new StreamController.singleSubscription();
+  Events expectedEvents = new Events()
+      ..add(42)
+      ..add("dibs")
+      ..error("error!")
+      ..error("error too!")
+      ..close();
+  Events actualEvents = new Events.capture(c);
+  expectedEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test automatic unsubscription on error.
+  c = new StreamController.singleSubscription();
+  expectedEvents = new Events()..add(42)..error("error");
+  actualEvents = new Events.capture(c, unsubscribeOnError: true);
+  Events sentEvents =
+      new Events()..add(42)..error("error")..add("Are you there?");
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test manual unsubscription.
+  c = new StreamController.singleSubscription();
+  expectedEvents = new Events()..add(42)..error("error")..add(37);
+  actualEvents = new Events.capture(c, unsubscribeOnError: false);
+  expectedEvents.replay(c);
+  actualEvents.subscription.cancel();
+  c.add("Are you there");  // Not sent to actualEvents.
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test filter.
+  c = new StreamController.singleSubscription();
+  expectedEvents = new Events()
+    ..add("a string")..add("another string")..close();
+  sentEvents = new Events()
+    ..add("a string")..add(42)..add("another string")..close();
+  actualEvents = new Events.capture(c.where((v) => v is String));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test map.
+  c = new StreamController.singleSubscription();
+  expectedEvents = new Events()..add("abab")..error("error")..close();
+  sentEvents = new Events()..add("ab")..error("error")..close();
+  actualEvents = new Events.capture(c.mappedBy((v) => "$v$v"));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test handleError.
+  c = new StreamController.singleSubscription();
+  expectedEvents = new Events()..add("ab")..error("[foo]");
+  sentEvents = new Events()..add("ab")..error("foo")..add("ab")..close();
+  actualEvents = new Events.capture(c.handleError((v) {
+        if (v.error is String) {
+          return new AsyncError("[${v.error}]",
+                                "other stack");
+        }
+      }), unsubscribeOnError: true);
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // reduce is tested asynchronously and therefore not in this file.
+
+  // Test expand
+  c = new StreamController.singleSubscription();
+  sentEvents = new Events()..add(3)..add(2)..add(4)..close();
+  expectedEvents = new Events()..add(1)..add(2)..add(3)
+                               ..add(1)..add(2)
+                               ..add(1)..add(2)..add(3)..add(4)
+                               ..close();
+  actualEvents = new Events.capture(c.expand((v) {
+    var l = [];
+    for (int i = 0; i < v; i++) l.add(i + 1);
+    return l;
+  }));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // pipe is tested asynchronously and therefore not in this file.
+  c = new StreamController.singleSubscription();
+  var list = <int>[];
+  c.pipeInto(new CollectionSink<int>(list))
+   .then(() { Expect.listEquals(<int>[1,2,9,3,9], list); });
+  c.add(1);
+  c.add(2);
+  c.add(9);
+  c.add(3);
+  c.add(9);
+  c.close();
+
+  // Test transform.
+  c = new StreamController.singleSubscription();
+  sentEvents = new Events()..add("a")..error(42)..add("b")..close();
+  expectedEvents =
+      new Events()..error("a")..add(42)..error("b")..add("foo")..close();
+  actualEvents = new Events.capture(c.transform(new StreamTransformer.from(
+      onData: (v, s) { s.signalError(new AsyncError(v)); },
+      onError: (e, s) { s.add(e.error); },
+      onDone: (s) {
+        s.add("foo");
+        s.close();
+      })));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test multiple filters.
+  c = new StreamController.singleSubscription();
+  sentEvents = new Events()..add(42)
+                           ..add("snugglefluffy")
+                           ..add(7)
+                           ..add("42")
+                           ..error("not FormatException")  // Unsubscribes.
+                           ..close();
+  expectedEvents = new Events()..add(42)..error("not FormatException");
+  actualEvents = new Events.capture(
+      c.where((v) => v is String)
+       .mappedBy((v) => int.parse(v))
+       .handleError((v) {
+          if (v.error is! FormatException) return v;
+        })
+       .where((v) => v > 10),
+      unsubscribeOnError: true);
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  // Test that only one subscription is allowed.
+  c = new StreamController.singleSubscription();
+  var sink = c.sink;
+  var stream = c.stream;
+  var counter = 0;
+  var subscription = stream.listen((data) { counter += data; });
+  Expect.throws(() => stream.listen(null), (e) => e is StateError);
+  sink.add(1);
+  Expect.equals(1, counter);
+  c.close();
+}
+
+testExtraMethods() {
+  Events sentEvents = new Events()..add(1)..add(2)..add(3)..close();
+
+  var c = new StreamController();
+  Events expectedEvents = new Events()..add(3)..close();
+  Events actualEvents = new Events.capture(c.skip(2));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  c = new StreamController();
+  expectedEvents = new Events()..close();
+  actualEvents = new Events.capture(c.skip(3));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  c = new StreamController();
+  expectedEvents = new Events()..close();
+  actualEvents = new Events.capture(c.skip(7));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  c = new StreamController();
+  expectedEvents = sentEvents;
+  actualEvents = new Events.capture(c.skip(0));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+
+  c = new StreamController();
+  expectedEvents = new Events()..add(3)..close();
+  actualEvents = new Events.capture(c.skipWhile((x) => x <= 2));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+
+  c = new StreamController();
+  expectedEvents = new Events()..add(1)..add(2)..close();
+  actualEvents = new Events.capture(c.take(2));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+
+  c = new StreamController();
+  expectedEvents = new Events()..add(1)..add(2)..close();
+  actualEvents = new Events.capture(c.takeWhile((x) => x <= 2));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  c = new StreamController();
+  sentEvents = new Events()
+      ..add(1)..add(1)..add(2)..add(1)..add(2)..add(2)..add(2)..close();
+  expectedEvents = new Events()
+      ..add(1)..add(2)..add(1)..add(2)..close();
+  actualEvents = new Events.capture(c.distinct());
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+
+  c = new StreamController();
+  sentEvents = new Events()
+      ..add(5)..add(6)..add(4)..add(6)..add(8)..add(3)..add(4)..add(1)..close();
+  expectedEvents = new Events()
+      ..add(5)..add(4)..add(3)..add(1)..close();
+  // Use 'distinct' as a filter with access to the previously emitted event.
+  actualEvents = new Events.capture(c.distinct((a, b) => a < b));
+  sentEvents.replay(c);
+  Expect.listEquals(expectedEvents.events, actualEvents.events);
+}
+
+main() {
+  testController();
+  testSingleController();
+  testExtraMethods();
+}