OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2015, the Dart project authors. Please see the AUTHORS file |
| 2 // for details. All rights reserved. Use of this source code is governed by a |
| 3 // BSD-style license that can be found in the LICENSE file. |
| 4 |
| 5 import "dart:async"; |
| 6 |
| 7 import "package:async/async.dart" show StreamCompleter; |
| 8 import "package:test/test.dart"; |
| 9 |
| 10 import "utils.dart"; |
| 11 |
| 12 main() { |
| 13 test("a stream is linked before listening", () async { |
| 14 var completer = new StreamCompleter(); |
| 15 completer.setSourceStream(createStream()); |
| 16 expect(completer.stream.toList(), completion([1, 2, 3, 4])); |
| 17 }); |
| 18 |
| 19 test("listened to before a stream is linked", () async { |
| 20 var completer = new StreamCompleter(); |
| 21 var done = completer.stream.toList(); |
| 22 await flushMicrotasks(); |
| 23 completer.setSourceStream(createStream()); |
| 24 expect(done, completion([1, 2, 3, 4])); |
| 25 }); |
| 26 |
| 27 test("cancel before linking a stream doesn't listen on stream", () async { |
| 28 var completer = new StreamCompleter(); |
| 29 var subscription = completer.stream.listen(null); |
| 30 subscription.pause(); // Should be ignored. |
| 31 subscription.cancel(); |
| 32 completer.setSourceStream(new UnusableStream()); // Doesn't throw. |
| 33 }); |
| 34 |
| 35 test("listen and pause before linking stream", () async { |
| 36 var controller = new StreamCompleter(); |
| 37 var events = []; |
| 38 var subscription = controller.stream.listen(events.add); |
| 39 var done = subscription.asFuture(); |
| 40 subscription.pause(); |
| 41 var sourceController = new StreamController(); |
| 42 sourceController..add(1)..add(2)..add(3)..add(4); |
| 43 controller.setSourceStream(sourceController.stream); |
| 44 await flushMicrotasks(); |
| 45 expect(sourceController.hasListener, isTrue); |
| 46 expect(sourceController.isPaused, isTrue); |
| 47 expect(events, []); |
| 48 subscription.resume(); |
| 49 await flushMicrotasks(); |
| 50 expect(sourceController.hasListener, isTrue); |
| 51 expect(sourceController.isPaused, isFalse); |
| 52 expect(events, [1, 2, 3, 4]); |
| 53 sourceController.close(); |
| 54 await done; |
| 55 expect(events, [1, 2, 3, 4]); |
| 56 }); |
| 57 |
| 58 test("pause more than once", () async { |
| 59 var completer = new StreamCompleter(); |
| 60 var events = []; |
| 61 var subscription = completer.stream.listen(events.add); |
| 62 var done = subscription.asFuture(); |
| 63 subscription.pause(); |
| 64 subscription.pause(); |
| 65 subscription.pause(); |
| 66 completer.setSourceStream(createStream()); |
| 67 for (int i = 0; i < 3; i++) { |
| 68 await flushMicrotasks(); |
| 69 expect(events, []); |
| 70 subscription.resume(); |
| 71 } |
| 72 await done; |
| 73 expect(events, [1, 2, 3, 4]); |
| 74 }); |
| 75 |
| 76 test("cancel new stream before source is done", () async { |
| 77 var completer = new StreamCompleter(); |
| 78 var lastEvent = -1; |
| 79 var controller = new StreamController(); |
| 80 var subscription; |
| 81 subscription = completer.stream.listen( |
| 82 (value) { |
| 83 expect(value, lessThan(3)); |
| 84 lastEvent = value; |
| 85 if (value == 2) { |
| 86 subscription.cancel(); |
| 87 } |
| 88 }, |
| 89 onError: unreachable("error"), |
| 90 onDone: unreachable("done"), |
| 91 cancelOnError: true); |
| 92 completer.setSourceStream(controller.stream); |
| 93 expect(controller.hasListener, isTrue); |
| 94 |
| 95 await flushMicrotasks(); |
| 96 expect(controller.hasListener, isTrue); |
| 97 controller.add(1); |
| 98 |
| 99 await flushMicrotasks(); |
| 100 expect(lastEvent, 1); |
| 101 expect(controller.hasListener, isTrue); |
| 102 controller.add(2); |
| 103 |
| 104 await flushMicrotasks(); |
| 105 expect(lastEvent, 2); |
| 106 expect(controller.hasListener, isFalse); |
| 107 }); |
| 108 |
| 109 test("complete with setEmpty before listening", () async { |
| 110 var completer = new StreamCompleter(); |
| 111 completer.setEmpty(); |
| 112 var done = new Completer(); |
| 113 completer.stream.listen( |
| 114 unreachable("data"), |
| 115 onError: unreachable("error"), |
| 116 onDone: done.complete); |
| 117 await done.future; |
| 118 }); |
| 119 |
| 120 test("complete with setEmpty after listening", () async { |
| 121 var completer = new StreamCompleter(); |
| 122 var done = new Completer(); |
| 123 completer.stream.listen( |
| 124 unreachable("data"), |
| 125 onError: unreachable("error"), |
| 126 onDone: done.complete); |
| 127 completer.setEmpty(); |
| 128 await done.future; |
| 129 }); |
| 130 |
| 131 test("source stream isn't listened to until completer stream is", () async { |
| 132 var completer = new StreamCompleter(); |
| 133 var controller; |
| 134 controller = new StreamController(onListen: () { |
| 135 scheduleMicrotask(controller.close); |
| 136 }); |
| 137 |
| 138 completer.setSourceStream(controller.stream); |
| 139 await flushMicrotasks(); |
| 140 expect(controller.hasListener, isFalse); |
| 141 var subscription = completer.stream.listen(null); |
| 142 expect(controller.hasListener, isTrue); |
| 143 await subscription.asFuture(); |
| 144 }); |
| 145 |
| 146 test("cancelOnError true when listening before linking stream", () async { |
| 147 var completer = new StreamCompleter(); |
| 148 var lastEvent = -1; |
| 149 var controller = new StreamController(); |
| 150 completer.stream.listen( |
| 151 (value) { |
| 152 expect(value, lessThan(3)); |
| 153 lastEvent = value; |
| 154 }, |
| 155 onError: (value) { |
| 156 expect(value, "3"); |
| 157 lastEvent = value; |
| 158 }, |
| 159 onDone: unreachable("done"), |
| 160 cancelOnError: true); |
| 161 completer.setSourceStream(controller.stream); |
| 162 expect(controller.hasListener, isTrue); |
| 163 |
| 164 await flushMicrotasks(); |
| 165 expect(controller.hasListener, isTrue); |
| 166 controller.add(1); |
| 167 |
| 168 await flushMicrotasks(); |
| 169 expect(lastEvent, 1); |
| 170 expect(controller.hasListener, isTrue); |
| 171 controller.add(2); |
| 172 |
| 173 await flushMicrotasks(); |
| 174 expect(lastEvent, 2); |
| 175 expect(controller.hasListener, isTrue); |
| 176 controller.addError("3"); |
| 177 |
| 178 await flushMicrotasks(); |
| 179 expect(lastEvent, "3"); |
| 180 expect(controller.hasListener, isFalse); |
| 181 }); |
| 182 |
| 183 test("cancelOnError true when listening after linking stream", () async { |
| 184 var completer = new StreamCompleter(); |
| 185 var lastEvent = -1; |
| 186 var controller = new StreamController(); |
| 187 completer.setSourceStream(controller.stream); |
| 188 controller.add(1); |
| 189 expect(controller.hasListener, isFalse); |
| 190 |
| 191 completer.stream.listen( |
| 192 (value) { |
| 193 expect(value, lessThan(3)); |
| 194 lastEvent = value; |
| 195 }, |
| 196 onError: (value) { |
| 197 expect(value, "3"); |
| 198 lastEvent = value; |
| 199 }, |
| 200 onDone: unreachable("done"), |
| 201 cancelOnError: true); |
| 202 |
| 203 expect(controller.hasListener, isTrue); |
| 204 |
| 205 await flushMicrotasks(); |
| 206 expect(lastEvent, 1); |
| 207 expect(controller.hasListener, isTrue); |
| 208 controller.add(2); |
| 209 |
| 210 await flushMicrotasks(); |
| 211 expect(lastEvent, 2); |
| 212 expect(controller.hasListener, isTrue); |
| 213 controller.addError("3"); |
| 214 |
| 215 await flushMicrotasks(); |
| 216 expect(controller.hasListener, isFalse); |
| 217 }); |
| 218 |
| 219 test("linking a stream after setSourceStream before listen", () async { |
| 220 var completer = new StreamCompleter(); |
| 221 completer.setSourceStream(createStream()); |
| 222 expect(() => completer.setSourceStream(createStream()), throwsStateError); |
| 223 expect(() => completer.setEmpty(), throwsStateError); |
| 224 await completer.stream.toList(); |
| 225 // Still fails after source is done |
| 226 expect(() => completer.setSourceStream(createStream()), throwsStateError); |
| 227 expect(() => completer.setEmpty(), throwsStateError); |
| 228 }); |
| 229 |
| 230 test("linking a stream after setSourceStream after listen", () async { |
| 231 var completer = new StreamCompleter(); |
| 232 var list = completer.stream.toList(); |
| 233 completer.setSourceStream(createStream()); |
| 234 expect(() => completer.setSourceStream(createStream()), throwsStateError); |
| 235 expect(() => completer.setEmpty(), throwsStateError); |
| 236 await list; |
| 237 // Still fails after source is done. |
| 238 expect(() => completer.setSourceStream(createStream()), throwsStateError); |
| 239 expect(() => completer.setEmpty(), throwsStateError); |
| 240 }); |
| 241 |
| 242 test("linking a stream after setEmpty before listen", () async { |
| 243 var completer = new StreamCompleter(); |
| 244 completer.setEmpty(); |
| 245 expect(() => completer.setSourceStream(createStream()), throwsStateError); |
| 246 expect(() => completer.setEmpty(), throwsStateError); |
| 247 await completer.stream.toList(); |
| 248 // Still fails after source is done |
| 249 expect(() => completer.setSourceStream(createStream()), throwsStateError); |
| 250 expect(() => completer.setEmpty(), throwsStateError); |
| 251 }); |
| 252 |
| 253 test("linking a stream after setEmpty() after listen", () async { |
| 254 var completer = new StreamCompleter(); |
| 255 var list = completer.stream.toList(); |
| 256 completer.setEmpty(); |
| 257 expect(() => completer.setSourceStream(createStream()), throwsStateError); |
| 258 expect(() => completer.setEmpty(), throwsStateError); |
| 259 await list; |
| 260 // Still fails after source is done. |
| 261 expect(() => completer.setSourceStream(createStream()), throwsStateError); |
| 262 expect(() => completer.setEmpty(), throwsStateError); |
| 263 }); |
| 264 |
| 265 test("listening more than once after setting stream", () async { |
| 266 var completer = new StreamCompleter(); |
| 267 completer.setSourceStream(createStream()); |
| 268 var list = completer.stream.toList(); |
| 269 expect(() => completer.stream.toList(), throwsStateError); |
| 270 await list; |
| 271 expect(() => completer.stream.toList(), throwsStateError); |
| 272 }); |
| 273 |
| 274 test("listening more than once before setting stream", () async { |
| 275 var completer = new StreamCompleter(); |
| 276 completer.stream.toList(); |
| 277 expect(() => completer.stream.toList(), throwsStateError); |
| 278 }); |
| 279 |
| 280 test("setting onData etc. before and after setting stream", () async { |
| 281 var completer = new StreamCompleter(); |
| 282 var controller = new StreamController(); |
| 283 var subscription = completer.stream.listen(null); |
| 284 var lastEvent = 0; |
| 285 subscription.onData((value) => lastEvent = value); |
| 286 subscription.onError((value) => lastEvent = "$value"); |
| 287 subscription.onDone(() => lastEvent = -1); |
| 288 completer.setSourceStream(controller.stream); |
| 289 await flushMicrotasks(); |
| 290 controller.add(1); |
| 291 await flushMicrotasks(); |
| 292 expect(lastEvent, 1); |
| 293 controller.addError(2); |
| 294 await flushMicrotasks(); |
| 295 expect(lastEvent, "2"); |
| 296 subscription.onData((value) => lastEvent = -value); |
| 297 subscription.onError((value) => lastEvent = "${-value}"); |
| 298 controller.add(1); |
| 299 await flushMicrotasks(); |
| 300 expect(lastEvent, -1); |
| 301 controller.addError(2); |
| 302 await flushMicrotasks(); |
| 303 expect(lastEvent, "-2"); |
| 304 controller.close(); |
| 305 await flushMicrotasks(); |
| 306 expect(lastEvent, -1); |
| 307 }); |
| 308 |
| 309 test("pause w/ resume future accross setting stream", () async { |
| 310 var completer = new StreamCompleter(); |
| 311 var resume = new Completer(); |
| 312 var subscription = completer.stream.listen(unreachable("data")); |
| 313 subscription.pause(resume.future); |
| 314 await flushMicrotasks(); |
| 315 completer.setSourceStream(createStream()); |
| 316 await flushMicrotasks(); |
| 317 resume.complete(); |
| 318 var events = []; |
| 319 subscription.onData(events.add); |
| 320 await subscription.asFuture(); |
| 321 expect(events, [1, 2, 3, 4]); |
| 322 }); |
| 323 |
| 324 test("asFuture with error accross setting stream", () async { |
| 325 var completer = new StreamCompleter(); |
| 326 var controller = new StreamController(); |
| 327 var subscription = completer.stream.listen(unreachable("data"), |
| 328 cancelOnError: false); |
| 329 var done = subscription.asFuture(); |
| 330 expect(controller.hasListener, isFalse); |
| 331 completer.setSourceStream(controller.stream); |
| 332 await flushMicrotasks(); |
| 333 expect(controller.hasListener, isTrue); |
| 334 controller.addError(42); |
| 335 await done.then(unreachable("data"), onError: (error) { |
| 336 expect(error, 42); |
| 337 }); |
| 338 expect(controller.hasListener, isFalse); |
| 339 }); |
| 340 } |
| 341 |
| 342 Stream<int> createStream() async* { |
| 343 yield 1; |
| 344 await flushMicrotasks(); |
| 345 yield 2; |
| 346 await flushMicrotasks(); |
| 347 yield 3; |
| 348 await flushMicrotasks(); |
| 349 yield 4; |
| 350 } |
OLD | NEW |