mirror of
https://github.com/dart-lang/sdk
synced 2024-10-14 10:13:38 +00:00
b101a7d002
Change-Id: Ib33169c3e0ffc870915c189404074a1dea472546 Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/196548 Reviewed-by: Bob Nystrom <rnystrom@google.com> Commit-Queue: Leaf Petersen <leafp@google.com>
125 lines
2.8 KiB
Dart
125 lines
2.8 KiB
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.
|
|
|
|
// @dart = 2.9
|
|
|
|
import "dart:async";
|
|
import "dart:io";
|
|
import "package:expect/expect.dart";
|
|
import "package:async_helper/async_helper.dart";
|
|
|
|
class TestConsumer implements StreamConsumer<List<int>> {
|
|
final List expected;
|
|
final List received = [];
|
|
|
|
int addStreamCount = 0;
|
|
int expcetedAddStreamCount;
|
|
bool expectClose;
|
|
|
|
TestConsumer(this.expected,
|
|
{this.expectClose: true, this.expcetedAddStreamCount: -1}) {
|
|
if (expectClose) asyncStart();
|
|
}
|
|
|
|
Future addStream(Stream stream) {
|
|
addStreamCount++;
|
|
var sub = stream.listen((v) {
|
|
received.addAll(v);
|
|
});
|
|
sub.pause();
|
|
scheduleMicrotask(sub.resume);
|
|
return sub.asFuture();
|
|
}
|
|
|
|
void matches(List list) {
|
|
Expect.listEquals(list, received);
|
|
}
|
|
|
|
Future close() {
|
|
return new Future.value().then((_) {
|
|
if (expectClose) asyncEnd();
|
|
Expect.listEquals(expected, received);
|
|
if (expcetedAddStreamCount >= 0) {
|
|
Expect.equals(expcetedAddStreamCount, addStreamCount);
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
void testClose() {
|
|
var sink = new IOSink(new TestConsumer([], expcetedAddStreamCount: 0));
|
|
sink.close();
|
|
}
|
|
|
|
void testAddClose() {
|
|
var sink = new IOSink(new TestConsumer([0]));
|
|
sink.add([0]);
|
|
sink.close();
|
|
|
|
sink = new IOSink(new TestConsumer([0, 1, 2]));
|
|
sink.add([0]);
|
|
sink.add([1]);
|
|
sink.add([2]);
|
|
sink.close();
|
|
}
|
|
|
|
void testAddFlush() {
|
|
var consumer = new TestConsumer([0, 1, 2]);
|
|
var sink = new IOSink(consumer);
|
|
sink.add([0]);
|
|
sink.flush().then((s) {
|
|
consumer.matches([0]);
|
|
s.add([1]);
|
|
s.add([2]);
|
|
s.flush().then((s) {
|
|
consumer.matches([0, 1, 2]);
|
|
s.close();
|
|
});
|
|
});
|
|
}
|
|
|
|
void testAddStreamClose() {
|
|
{
|
|
var sink = new IOSink(new TestConsumer([0]));
|
|
var controller = new StreamController<List<int>>(sync: true);
|
|
sink.addStream(controller.stream).then((_) {
|
|
sink.close();
|
|
});
|
|
controller.add([0]);
|
|
controller.close();
|
|
}
|
|
{
|
|
var sink = new IOSink(new TestConsumer([0, 1, 2]));
|
|
var controller = new StreamController<List<int>>(sync: true);
|
|
sink.addStream(controller.stream).then((_) {
|
|
sink.close();
|
|
});
|
|
controller.add([0]);
|
|
controller.add([1]);
|
|
controller.add([2]);
|
|
controller.close();
|
|
}
|
|
}
|
|
|
|
void testAddStreamAddClose() {
|
|
{
|
|
var sink = new IOSink(new TestConsumer([0, 1]));
|
|
var controller = new StreamController<List<int>>(sync: true);
|
|
sink.addStream(controller.stream).then((_) {
|
|
sink.add([1]);
|
|
sink.close();
|
|
});
|
|
controller.add([0]);
|
|
controller.close();
|
|
}
|
|
}
|
|
|
|
void main() {
|
|
testClose();
|
|
testAddClose();
|
|
testAddFlush();
|
|
testAddStreamClose();
|
|
testAddStreamAddClose();
|
|
}
|