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 library unittest.invoker; | |
6 | |
7 import 'dart:async'; | |
8 | |
9 import 'package:stack_trace/stack_trace.dart'; | |
10 | |
11 import 'expect.dart'; | |
12 import 'live_test.dart'; | |
13 import 'live_test_controller.dart'; | |
14 import 'state.dart'; | |
15 import 'suite.dart'; | |
16 import 'test.dart'; | |
17 import 'utils.dart'; | |
18 | |
19 /// A test in this isolate. | |
20 class LocalTest implements Test { | |
21 /// The name of the test. | |
22 final String name; | |
23 | |
24 /// The test body. | |
25 final AsyncFunction _body; | |
26 | |
27 /// The callback used to clean up after the test. | |
28 /// | |
29 /// This is separated out from [_body] because it needs to run once the test's | |
30 /// asynchronous computation has finished, even if that's different from the | |
31 /// completion of the main body of the test. | |
32 final AsyncFunction _tearDown; | |
33 | |
34 LocalTest(this.name, body(), {tearDown()}) | |
35 : _body = body, | |
36 _tearDown = tearDown; | |
37 | |
38 /// Loads a single runnable instance of this test. | |
39 LiveTest load(Suite suite) { | |
40 var invoker = new Invoker._(suite, this); | |
41 return invoker.liveTest; | |
42 } | |
43 } | |
44 | |
45 /// The class responsible for managing the lifecycle of a single local test. | |
46 /// | |
47 /// The current invoker is accessible within the zone scope of the running test | |
48 /// using [Invoker.current]. It's used to track asynchronous callbacks and | |
49 /// report asynchronous errors. | |
50 class Invoker { | |
51 /// The live test being driven by the invoker. | |
52 /// | |
53 /// This provides a view into the state of the test being executed. | |
54 LiveTest get liveTest => _controller.liveTest; | |
55 LiveTestController _controller; | |
56 | |
57 /// The test being run. | |
58 LocalTest get _test => liveTest.test as LocalTest; | |
59 | |
60 /// Note that this is meaningless once [_onCompleteCompleter] is complete. | |
61 var _outstandingCallbacks = 0; | |
62 | |
63 /// The completer to complete once the test body finishes. | |
64 /// | |
65 /// This is distinct from [_controller.completer] because a tear-down may need | |
66 /// to run before the test is truly finished. | |
67 final _completer = new Completer(); | |
68 | |
69 /// The current invoker, or `null` if none is defined. | |
70 /// | |
71 /// An invoker is only set within the zone scope of a running test. | |
72 static Invoker get current { | |
73 // TODO(nweiz): Use a private symbol when dart2js supports it (issue 17526). | |
74 return Zone.current[#unittest.invoker]; | |
75 } | |
76 | |
77 Invoker._(Suite suite, LocalTest test) { | |
78 _controller = new LiveTestController(suite, test, _onRun); | |
79 } | |
80 | |
81 /// Tells the invoker that there's a callback running that it should wait for | |
82 /// before considering the test successful. | |
83 /// | |
84 /// Each call to [addOutstandingCallback] should be followed by a call to | |
85 /// [removeOutstandingCallback] once the callbak is no longer running. Note | |
86 /// that only successful tests wait for outstanding callbacks; as soon as a | |
87 /// test experiences an error, any further calls to [addOutstandingCallback] | |
88 /// or [removeOutstandingCallback] will do nothing. | |
89 void addOutstandingCallback() { | |
90 _outstandingCallbacks++; | |
91 } | |
92 | |
93 /// Tells the invoker that a callback declared with [addOutstandingCallback] | |
94 /// is no longer running. | |
95 void removeOutstandingCallback() { | |
96 _outstandingCallbacks--; | |
97 | |
98 if (_outstandingCallbacks != 0) return; | |
99 if (_completer.isCompleted) return; | |
100 | |
101 // The test must be passing if we get here, because if there were an error | |
102 // the completer would already be completed. | |
103 assert(liveTest.state.result == Result.success); | |
104 _completer.complete(); | |
105 } | |
106 | |
107 /// Notifies the invoker of an asynchronous error. | |
108 /// | |
109 /// Note that calling this explicitly is rarely necessary, since any | |
110 /// otherwise-uncaught errors will be forwarded to the invoker anyway. | |
111 void handleError(error, [StackTrace stackTrace]) { | |
112 if (stackTrace == null) stackTrace = new Chain.current(); | |
113 | |
114 var afterSuccess = liveTest.isComplete && | |
115 liveTest.state.result == Result.success; | |
116 | |
117 if (error is! TestFailure) { | |
118 _controller.setState(const State(Status.complete, Result.error)); | |
119 } else if (liveTest.state.result != Result.error) { | |
120 _controller.setState(const State(Status.complete, Result.failure)); | |
121 } | |
122 | |
123 _controller.addError(error, stackTrace); | |
124 | |
125 if (!_completer.isCompleted) _completer.complete(); | |
126 | |
127 // If a test was marked as success but then had an error, that indicates | |
128 // that it was poorly-written and could be flaky. | |
129 if (!afterSuccess) return; | |
130 handleError( | |
131 "This test failed after it had already completed. Make sure to use " | |
132 "[expectAsync]\n" | |
133 "or the [completes] matcher when testing async code.", | |
134 stackTrace); | |
135 } | |
136 | |
137 /// The method that's run when the test is started. | |
138 void _onRun() { | |
139 _controller.setState(const State(Status.running, Result.success)); | |
140 | |
141 Chain.capture(() { | |
142 runZoned(() { | |
143 // TODO(nweiz): Make the timeout configurable. | |
144 // TODO(nweiz): Reset this timer whenever the user's code interacts with | |
145 // the library. | |
146 var timer = new Timer(new Duration(seconds: 30), () { | |
147 if (liveTest.isComplete) return; | |
148 handleError( | |
149 new TimeoutException( | |
150 "Test timed out after 30 seconds.", | |
151 new Duration(seconds: 30))); | |
152 }); | |
153 | |
154 addOutstandingCallback(); | |
155 | |
156 // Run the test asynchronously so that the "running" state change has a | |
157 // chance to hit its event handler(s) before the test produces an error. | |
158 // If an error is emitted before the first state change is handled, we | |
159 // can end up with [onError] callbacks firing before the corresponding | |
160 // [onStateChange], which violates the timing guarantees. | |
161 new Future(_test._body) | |
162 .then((_) => removeOutstandingCallback()); | |
163 | |
164 // Explicitly handle an error here so that we can return the [Future]. | |
165 // If a [Future] returned from an error zone would throw an error | |
166 // through the zone boundary, it instead never completes, and we want to | |
167 // avoid that. | |
168 _completer.future.then((_) { | |
169 if (_test._tearDown == null) return null; | |
170 return new Future.sync(_test._tearDown); | |
171 }).catchError(Zone.current.handleUncaughtError).then((_) { | |
172 timer.cancel(); | |
173 _controller.setState( | |
174 new State(Status.complete, liveTest.state.result)); | |
175 | |
176 // Use [Timer.run] here to avoid starving the DOM or other | |
177 // non-microtask events. | |
178 Timer.run(_controller.completer.complete); | |
179 }); | |
180 }, zoneValues: {#unittest.invoker: this}, onError: handleError); | |
181 }); | |
182 } | |
183 } | |
OLD | NEW |