Make pkg/stack_trace use stack chains.

pkg/scheduled_test/lib/scheduled_test.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.
 
 // TODO(nweiz): Add support for calling [schedule] while the schedule is already
 // running.
 // TODO(nweiz): Port the non-Pub-specific scheduled test libraries from Pub.
 /// A package for writing readable tests of asynchronous behavior.
 ///
 /// ## Installing ##
@@ skipping 173 matching lines @@
 /// If a single task might take a long time, you can also manually tell the
 /// [Schedule] that it's making progress by calling [Schedule.heartbeat], which
 /// will reset the timeout whenever it's called.
 ///
 /// [pub]: http://pub.dartlang.org
 /// [pkg]: http://pub.dartlang.org/packages/scheduled_test
 library scheduled_test;
 
 import 'dart:async';
 
+import 'package:stack_trace/stack_trace.dart';
 import 'package:unittest/unittest.dart' as unittest;
 
 import 'src/schedule.dart';
 import 'src/schedule_error.dart';
 
 export 'package:unittest/unittest.dart' hide
     test, solo_test, group, setUp, tearDown, completes, completion;
 
 export 'src/schedule.dart';
 export 'src/schedule_error.dart';
@@ skipping 19 matching lines @@
 /// Creates a new test case with the given description and body that will be the
 /// only test run in this file. This has the same semantics as
 /// [unittest.solo_test].
 void solo_test(String description, void body()) =>
   _test(description, body, unittest.solo_test);
 
 void _test(String description, void body(), Function testFn) {
   _ensureInitialized();
   _ensureSetUpForTopLevel();
   testFn(description, () {
-    var asyncDone = unittest.expectAsync0(() {});
-    return currentSchedule.run(() {
-      if (_setUpFn != null) _setUpFn();
-      body();
-    }).then((_) {
-      // If we got here, the test completed successfully so tell unittest so.
-      asyncDone();
-    }).catchError((e) {
+    var completer = new Completer();
+
+    Chain.capture(() {
+      _currentSchedule = new Schedule();
+      return currentSchedule.run(() {
+        if (_setUpFn != null) _setUpFn();
+        body();
+      }).then(completer.complete);
+    }, onError: (e, chain) {
       if (e is ScheduleError) {
         assert(e.schedule.errors.contains(e));
         assert(e.schedule == currentSchedule);
         unittest.registerException(e.schedule.errorString());
       } else {
-        unittest.registerException(e);
+        unittest.registerException(e, chain);
       }
+      completer.complete();
     });
+
+    return completer.future;
   });
 }
 
 /// Whether or not the tests currently being defined are in a group. This is
 /// only true when defining tests, not when executing them.
 bool _inGroup = false;
 
 /// Creates a new named group of tests. This has the same semantics as
 /// [unittest.group].
 void group(String description, void body()) {
@@ skipping 51 matching lines @@
 /// Registers callbacks for [unittest.setUp] and [unittest.tearDown] that set up
 /// and tear down the scheduled test infrastructure.
 void _setUpScheduledTest([void setUpFn()]) {
   if (!_inGroup) {
     _setUpForTopLevel = true;
     unittest.setUp(() {
       if (currentSchedule != null) {
         throw new StateError('There seems to be another scheduled test '
             'still running.');
       }
-      _currentSchedule = new Schedule();
       if (_setUpFn != null) {
         var parentFn = _setUpFn;
         _setUpFn = () { parentFn(); setUpFn(); };
       } else {
         _setUpFn = setUpFn;
       }
     });
 
     unittest.tearDown(() {
       _currentSchedule = null;
       _setUpFn = null;
     });
   } else {
     unittest.setUp(() {
-      if (currentSchedule == null) {
-        throw new StateError('No schedule allocated.');
-      } else if (_setUpFn != null) {
+      if (_setUpFn != null) {
         var parentFn = _setUpFn;
         _setUpFn = () { parentFn(); setUpFn(); };
       } else {
         _setUpFn = setUpFn;
       }
     });
   }
 }
 
 /// Ensures that the global configuration for `scheduled_test` has been
@@ skipping 20 matching lines @@
 /// [description] provides an optional description of the future, which is
 /// used when generating error messages.
 Future wrapFuture(Future future, [String description]) {
   if (currentSchedule == null) {
     throw new StateError("Unexpected call to wrapFuture with no current "
         "schedule.");
   }
 
   return currentSchedule.wrapFuture(future, description);
 }