| OLD | NEW |
| 1 // Copyright (c) 2013, the Dart project authors. Please see the AUTHORS file | 1 // Copyright (c) 2013, the Dart project authors. Please see the AUTHORS file |
| 2 // for details. All rights reserved. Use of this source code is governed by a | 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. | 3 // BSD-style license that can be found in the LICENSE file. |
| 4 | 4 |
| 5 library test_utils; | 5 library test_utils; |
| 6 | 6 |
| 7 import 'dart:async'; | 7 import 'dart:async'; |
| 8 | 8 |
| 9 import 'package:scheduled_test/src/utils.dart'; | 9 import 'package:scheduled_test/src/utils.dart'; |
| 10 import 'package:scheduled_test/src/mock_clock.dart' as mock_clock; | 10 import 'package:scheduled_test/src/mock_clock.dart' as mock_clock; |
| 11 | 11 |
| 12 export 'package:scheduled_test/src/utils.dart'; | 12 export 'package:scheduled_test/src/utils.dart'; |
| 13 | 13 |
| 14 /// Wraps [input] to provide a timeout. If [input] completes before | 14 /// Wraps [input] to provide a timeout. If [input] completes before |
| 15 /// [milliseconds] have passed, then the return value completes in the same way. | 15 /// [milliseconds] have passed, then the return value completes in the same way. |
| 16 /// However, if [milliseconds] pass before [input] has completed, [onTimeout] is | 16 /// However, if [milliseconds] pass before [input] has completed, [onTimeout] is |
| 17 /// run and its result is passed to [input] (with chaining, if it returns a | 17 /// run and its result is passed to [input] (with chaining, if it returns a |
| 18 /// [Future]). | 18 /// [Future]). |
| 19 /// | 19 /// |
| 20 /// Note that timing out will not cancel the asynchronous operation behind | 20 /// Note that timing out will not cancel the asynchronous operation behind |
| 21 /// [input]. | 21 /// [input]. |
| 22 Future timeout(Future input, int milliseconds, onTimeout()) { | 22 Future timeout(Future input, int milliseconds, onTimeout()) { |
| 23 bool completed = false; | |
| 24 var completer = new Completer(); | 23 var completer = new Completer(); |
| 25 var timer = new Timer(new Duration(milliseconds: milliseconds), () { | 24 var timer = new Timer(new Duration(milliseconds: milliseconds), () { |
| 26 completed = true; | 25 chainToCompleter(new Future.of(onTimeout), completer); |
| 27 chainToCompleter(new Future.immediate(null).then((_) => onTimeout()), | |
| 28 completer); | |
| 29 }); | 26 }); |
| 30 input.then((value) { | 27 input.then((value) { |
| 31 if (completed) return; | 28 if (completer.isCompleted) return; |
| 32 timer.cancel(); | 29 timer.cancel(); |
| 33 completer.complete(value); | 30 completer.complete(value); |
| 34 }).catchError((e) { | 31 }).catchError((e) { |
| 35 if (completed) return; | 32 if (completer.isCompleted) return; |
| 36 timer.cancel(); | 33 timer.cancel(); |
| 37 completer.completeError(e.error, e.stackTrace); | 34 completer.completeError(e.error, e.stackTrace); |
| 38 }); | 35 }); |
| 39 return completer.future; | 36 return completer.future; |
| 40 } | 37 } |
| 41 | 38 |
| 42 /// Returns a [Future] that will complete in [milliseconds]. | 39 /// Returns a [Future] that will complete in [milliseconds]. |
| 43 Future sleep(int milliseconds) { | 40 Future sleep(int milliseconds) { |
| 44 var completer = new Completer(); | 41 var completer = new Completer(); |
| 45 mock_clock.newTimer(new Duration(milliseconds: milliseconds), () { | 42 mock_clock.newTimer(new Duration(milliseconds: milliseconds), () { |
| 46 completer.complete(); | 43 completer.complete(); |
| 47 }); | 44 }); |
| 48 return completer.future; | 45 return completer.future; |
| 49 } | 46 } |
| OLD | NEW |