OLD | NEW |
| (Empty) |
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 | |
3 // BSD-style license that can be found in the LICENSE file. | |
4 | |
5 library barback.test.transformer.mock; | |
6 | |
7 import 'dart:async'; | |
8 | |
9 import 'package:barback/barback.dart'; | |
10 import 'package:barback/src/utils.dart'; | |
11 import 'package:scheduled_test/scheduled_test.dart'; | |
12 | |
13 /// The abstract base class for transformers used to test barback. | |
14 /// | |
15 /// This adds the ability to pause and resume different components of the | |
16 /// transformers, and to tell whether they're running, when they start running, | |
17 /// and how many times they've run. | |
18 /// | |
19 /// Transformers extending this should override [doIsPrimary] and [doApply] | |
20 /// rather than [isPrimary] and [apply], and they should use [getInput] and | |
21 /// [getPrimary] rather than [transform.getInput] and [transform.primaryInput]. | |
22 abstract class MockTransformer extends Transformer { | |
23 /// The number of times the transformer has been applied. | |
24 /// | |
25 /// This is scheduled. The Future will complete at the point in the schedule | |
26 /// that this is called. | |
27 Future<int> get numRuns => schedule(() => _numRuns); | |
28 var _numRuns = 0; | |
29 | |
30 /// The number of currently running transforms. | |
31 int _runningTransforms = 0; | |
32 | |
33 /// A completer for pausing the transformer before it finishes running | |
34 /// [apply]. | |
35 Completer _apply; | |
36 | |
37 /// Completers for pausing the transformer before it finishes running | |
38 /// [isPrimary]. | |
39 final _isPrimary = new Map<AssetId, Completer>(); | |
40 | |
41 /// Completers for pausing the transformer before it finishes getting inputs | |
42 /// the [Transform]. | |
43 final _getInput = new Map<AssetId, Completer>(); | |
44 | |
45 /// Completer for pausing the transformer before it accesses [primaryInput]. | |
46 Completer _primaryInput; | |
47 | |
48 /// A completer that completes once this transformer begins running. | |
49 /// | |
50 /// Once this transformer finishes running, this is reset to a new completer, | |
51 /// so it can be used multiple times. | |
52 var _started = new Completer(); | |
53 | |
54 /// `true` if any transforms are currently running. | |
55 /// | |
56 /// This is scheduled. The Future will complete at the point in the schedule | |
57 /// that this is called. | |
58 Future<bool> get isRunning => schedule(() => _runningTransforms > 0); | |
59 | |
60 /// If this is set to `true`, the transformer will consume its primary input | |
61 /// during [apply]. | |
62 bool consumePrimary = false; | |
63 | |
64 /// Pauses the schedule until this transformer begins running. | |
65 void waitUntilStarted() { | |
66 schedule(() => _started.future, "wait until $this starts"); | |
67 } | |
68 | |
69 /// Causes the transformer to pause after running [apply] but before the | |
70 /// returned Future completes. | |
71 /// | |
72 /// This can be resumed by calling [resumeApply]. This operation is scheduled. | |
73 void pauseApply() { | |
74 schedule(() { | |
75 _apply = new Completer(); | |
76 }, "pause apply for $this"); | |
77 } | |
78 | |
79 /// Resumes the transformer's [apply] call after [pauseApply] was called. | |
80 /// | |
81 /// This operation is scheduled. | |
82 void resumeApply() { | |
83 schedule(() { | |
84 _apply.complete(); | |
85 _apply = null; | |
86 }, "resume apply for $this"); | |
87 } | |
88 | |
89 /// Causes the transformer to pause after running [isPrimary] on the asset | |
90 /// with the given [name], but before the returned Future completes. | |
91 /// | |
92 /// This can be resumed by calling [resumeIsPrimary]. This operation is | |
93 /// scheduled. | |
94 void pauseIsPrimary(String name) { | |
95 schedule(() { | |
96 _isPrimary[new AssetId.parse(name)] = new Completer(); | |
97 }, "pause isPrimary($name) for $this"); | |
98 } | |
99 | |
100 /// Resumes the transformer's [isPrimary] call on the asset with the given | |
101 /// [name] after [pauseIsPrimary] was called. | |
102 /// | |
103 /// This operation is scheduled. | |
104 void resumeIsPrimary(String name) { | |
105 schedule(() { | |
106 _isPrimary.remove(new AssetId.parse(name)).complete(); | |
107 }, "resume isPrimary($name) for $this"); | |
108 } | |
109 | |
110 /// Causes the transformer to pause while loading the secondary input with | |
111 /// the given [name]. | |
112 /// | |
113 /// This can be resumed by calling [resumeGetInput]. This operation is | |
114 /// scheduled. | |
115 void pauseGetInput(String name) { | |
116 schedule(() { | |
117 _getInput[new AssetId.parse(name)] = new Completer(); | |
118 }, "pause getInput($name) for $this"); | |
119 } | |
120 | |
121 /// Resumes the transformer's loading of the input with the given [name] after | |
122 /// [pauseGetInput] was called. | |
123 /// | |
124 /// This operation is scheduled. | |
125 void resumeGetInput(String name) { | |
126 schedule(() { | |
127 _getInput.remove(new AssetId.parse(name)).complete(); | |
128 }, "resume getInput($name) for $this"); | |
129 } | |
130 | |
131 /// Causes the transformer to pause before accessing [primaryInput]. | |
132 /// | |
133 /// This can be resumed by calling [resumeGetPrimary]. This operation is | |
134 /// scheduled. | |
135 void pausePrimaryInput() { | |
136 schedule(() { | |
137 _primaryInput = new Completer(); | |
138 }, "pause primaryInput for $this"); | |
139 } | |
140 | |
141 /// Resumes the transformer's invocation of [primaryInput] after | |
142 /// [pauseGetPrimary] was called. | |
143 /// | |
144 /// This operation is scheduled. | |
145 void resumePrimaryInput() { | |
146 schedule(() { | |
147 _primaryInput.complete(); | |
148 _primaryInput = null; | |
149 }, "resume getPrimary() for $this"); | |
150 } | |
151 | |
152 /// Like [Transform.getInput], but respects [pauseGetInput]. | |
153 /// | |
154 /// This is intended for use by subclasses of [MockTransformer]. | |
155 Future<Asset> getInput(Transform transform, AssetId id) { | |
156 return newFuture(() { | |
157 if (_getInput.containsKey(id)) return _getInput[id].future; | |
158 }).then((_) => transform.getInput(id)); | |
159 } | |
160 | |
161 /// Like [Transform.primaryInput], but respects [pauseGetPrimary]. | |
162 /// | |
163 /// This is intended for use by subclasses of [MockTransformer]. | |
164 Future<Asset> getPrimary(Transform transform) { | |
165 return newFuture(() { | |
166 if (_primaryInput != null) return _primaryInput.future; | |
167 }).then((_) => transform.primaryInput); | |
168 } | |
169 | |
170 Future<bool> isPrimary(AssetId id) { | |
171 return newFuture(() => doIsPrimary(id)).then((result) { | |
172 return newFuture(() { | |
173 if (_isPrimary.containsKey(id)) { | |
174 return _isPrimary[id].future; | |
175 } | |
176 }).then((_) => result); | |
177 }); | |
178 } | |
179 | |
180 Future apply(Transform transform) { | |
181 _numRuns++; | |
182 if (_runningTransforms == 0) _started.complete(); | |
183 _runningTransforms++; | |
184 if (consumePrimary) transform.consumePrimary(); | |
185 return newFuture(() => doApply(transform)).then((_) { | |
186 if (_apply != null) return _apply.future; | |
187 }).whenComplete(() { | |
188 _runningTransforms--; | |
189 if (_runningTransforms == 0) _started = new Completer(); | |
190 }); | |
191 } | |
192 | |
193 /// The wrapped version of [isPrimary] for subclasses to override. | |
194 /// | |
195 /// This may return a `Future<bool>` or, if it's entirely synchronous, a | |
196 /// `bool`. | |
197 doIsPrimary(AssetId id); | |
198 | |
199 /// The wrapped version of [doApply] for subclasses to override. | |
200 /// | |
201 /// If this does asynchronous work, it should return a [Future] that completes | |
202 /// once it's finished. | |
203 doApply(Transform transform); | |
204 } | |
OLD | NEW |