| 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.console_reporter; | |
| 6 | |
| 7 import 'dart:async'; | |
| 8 import 'dart:io'; | |
| 9 | |
| 10 import 'engine.dart'; | |
| 11 import 'io.dart'; | |
| 12 import 'live_test.dart'; | |
| 13 import 'state.dart'; | |
| 14 import 'suite.dart'; | |
| 15 import 'utils.dart'; | |
| 16 | |
| 17 /// The terminal escape for green text, or the empty string if this is Windows | |
| 18 /// or not outputting to a terminal. | |
| 19 final _green = getSpecial('\u001b[32m'); | |
| 20 | |
| 21 /// The terminal escape for red text, or the empty string if this is Windows or | |
| 22 /// not outputting to a terminal. | |
| 23 final _red = getSpecial('\u001b[31m'); | |
| 24 | |
| 25 /// The terminal escape for removing test coloring, or the empty string if this | |
| 26 /// is Windows or not outputting to a terminal. | |
| 27 final _noColor = getSpecial('\u001b[0m'); | |
| 28 | |
| 29 /// The maximum console line length. | |
| 30 /// | |
| 31 /// Lines longer than this will be cropped. | |
| 32 const _lineLength = 100; | |
| 33 | |
| 34 /// A reporter that prints test results to the console in a single | |
| 35 /// continuously-updating line. | |
| 36 class ConsoleReporter { | |
| 37 /// The engine used to run the tests. | |
| 38 final Engine _engine; | |
| 39 | |
| 40 /// Whether multiple test suites are being run. | |
| 41 final bool _multipleSuites; | |
| 42 | |
| 43 /// A stopwatch that tracks the duration of the full run. | |
| 44 final _stopwatch = new Stopwatch(); | |
| 45 | |
| 46 /// The set of tests that have completed and been marked as passing. | |
| 47 final _passed = new Set<LiveTest>(); | |
| 48 | |
| 49 /// The set of tests that have completed and been marked as failing or error. | |
| 50 final _failed = new Set<LiveTest>(); | |
| 51 | |
| 52 /// The size of [_passed] last time a progress notification was printed. | |
| 53 int _lastProgressPassed; | |
| 54 | |
| 55 /// The size of [_failed] last time a progress notification was printed. | |
| 56 int _lastProgressFailed; | |
| 57 | |
| 58 /// The message printed for the last progress notification. | |
| 59 String _lastProgressMessage; | |
| 60 | |
| 61 /// Creates a [ConsoleReporter] that will run all tests in [suites]. | |
| 62 ConsoleReporter(Iterable<Suite> suites) | |
| 63 : _multipleSuites = suites.length > 1, | |
| 64 _engine = new Engine(suites) { | |
| 65 | |
| 66 _engine.onTestStarted.listen((liveTest) { | |
| 67 _progressLine(_description(liveTest)); | |
| 68 liveTest.onStateChange.listen((state) { | |
| 69 if (state.status != Status.complete) return; | |
| 70 if (state.result == Result.success) { | |
| 71 _passed.add(liveTest); | |
| 72 } else { | |
| 73 _passed.remove(liveTest); | |
| 74 _failed.add(liveTest); | |
| 75 } | |
| 76 _progressLine(_description(liveTest)); | |
| 77 }); | |
| 78 | |
| 79 liveTest.onError.listen((error) { | |
| 80 if (liveTest.state.status != Status.complete) return; | |
| 81 | |
| 82 _progressLine(_description(liveTest)); | |
| 83 print(''); | |
| 84 print(indent(error.error.toString())); | |
| 85 print(indent(terseChain(error.stackTrace).toString())); | |
| 86 }); | |
| 87 }); | |
| 88 } | |
| 89 | |
| 90 /// Runs all tests in all provided suites. | |
| 91 /// | |
| 92 /// This returns `true` if all tests succeed, and `false` otherwise. It will | |
| 93 /// only return once all tests have finished running. | |
| 94 Future<bool> run() { | |
| 95 if (_stopwatch.isRunning) { | |
| 96 throw new StateError("ConsoleReporter.run() may not be called more than " | |
| 97 "once."); | |
| 98 } | |
| 99 | |
| 100 if (_engine.liveTests.isEmpty) { | |
| 101 print("No tests ran."); | |
| 102 return new Future.value(true); | |
| 103 } | |
| 104 | |
| 105 _stopwatch.start(); | |
| 106 return _engine.run().then((success) { | |
| 107 if (success) { | |
| 108 _progressLine("All tests passed!"); | |
| 109 print(''); | |
| 110 } else { | |
| 111 _progressLine('Some tests failed.', color: _red); | |
| 112 print(''); | |
| 113 } | |
| 114 | |
| 115 return success; | |
| 116 }); | |
| 117 } | |
| 118 | |
| 119 /// Signals that the caller is done with any test output and the reporter | |
| 120 /// should release any resources it has allocated. | |
| 121 Future close() => _engine.close(); | |
| 122 | |
| 123 /// Prints a line representing the current state of the tests. | |
| 124 /// | |
| 125 /// [message] goes after the progress report, and may be truncated to fit the | |
| 126 /// entire line within [_lineLength]. If [color] is passed, it's used as the | |
| 127 /// color for [message]. | |
| 128 void _progressLine(String message, {String color}) { | |
| 129 // Print nothing if nothing has changed since the last progress line. | |
| 130 if (_passed.length == _lastProgressPassed && | |
| 131 _failed.length == _lastProgressFailed && | |
| 132 message == _lastProgressMessage) { | |
| 133 return; | |
| 134 } | |
| 135 | |
| 136 _lastProgressPassed = _passed.length; | |
| 137 _lastProgressFailed = _failed.length; | |
| 138 _lastProgressMessage = message; | |
| 139 | |
| 140 if (color == null) color = ''; | |
| 141 var duration = _stopwatch.elapsed; | |
| 142 var buffer = new StringBuffer(); | |
| 143 | |
| 144 // \r moves back to the beginning of the current line. | |
| 145 buffer.write('\r${_timeString(duration)} '); | |
| 146 buffer.write(_green); | |
| 147 buffer.write('+'); | |
| 148 buffer.write(_passed.length); | |
| 149 buffer.write(_noColor); | |
| 150 | |
| 151 if (_failed.isNotEmpty) { | |
| 152 buffer.write(_red); | |
| 153 buffer.write(' -'); | |
| 154 buffer.write(_failed.length); | |
| 155 buffer.write(_noColor); | |
| 156 } | |
| 157 | |
| 158 buffer.write(': '); | |
| 159 buffer.write(color); | |
| 160 | |
| 161 // Ensure the line fits within [_lineLength]. [buffer] includes the color | |
| 162 // escape sequences too. Because these sequences are not visible characters, | |
| 163 // we make sure they are not counted towards the limit. | |
| 164 var nonVisible = 1 + _green.length + _noColor.length + color.length + | |
| 165 (_failed.isEmpty ? 0 : _red.length + _noColor.length); | |
| 166 var length = buffer.length - nonVisible; | |
| 167 buffer.write(_truncate(message, _lineLength - length)); | |
| 168 buffer.write(_noColor); | |
| 169 | |
| 170 // Pad the rest of the line so that it looks erased. | |
| 171 length = buffer.length - nonVisible - _noColor.length; | |
| 172 buffer.write(' ' * (_lineLength - length)); | |
| 173 stdout.write(buffer.toString()); | |
| 174 } | |
| 175 | |
| 176 /// Returns a representation of [duration] as `MM:SS`. | |
| 177 String _timeString(Duration duration) { | |
| 178 return "${duration.inMinutes.toString().padLeft(2, '0')}:" | |
| 179 "${(duration.inSeconds % 60).toString().padLeft(2, '0')}"; | |
| 180 } | |
| 181 | |
| 182 /// Truncates [text] to fit within [maxLength]. | |
| 183 /// | |
| 184 /// This will try to truncate along word boundaries and preserve words both at | |
| 185 /// the beginning and the end of [text]. | |
| 186 String _truncate(String text, int maxLength) { | |
| 187 // Return the full message if it fits. | |
| 188 if (text.length <= maxLength) return text; | |
| 189 | |
| 190 // If we can fit the first and last three words, do so. | |
| 191 var words = text.split(' '); | |
| 192 if (words.length > 1) { | |
| 193 var i = words.length; | |
| 194 var length = words.first.length + 4; | |
| 195 do { | |
| 196 i--; | |
| 197 length += 1 + words[i].length; | |
| 198 } while (length <= maxLength && i > 0); | |
| 199 if (length > maxLength || i == 0) i++; | |
| 200 if (i < words.length - 4) { | |
| 201 // Require at least 3 words at the end. | |
| 202 var buffer = new StringBuffer(); | |
| 203 buffer.write(words.first); | |
| 204 buffer.write(' ...'); | |
| 205 for ( ; i < words.length; i++) { | |
| 206 buffer.write(' '); | |
| 207 buffer.write(words[i]); | |
| 208 } | |
| 209 return buffer.toString(); | |
| 210 } | |
| 211 } | |
| 212 | |
| 213 // Otherwise truncate to return the trailing text, but attempt to start at | |
| 214 // the beginning of a word. | |
| 215 var result = text.substring(text.length - maxLength + 4); | |
| 216 var firstSpace = result.indexOf(' '); | |
| 217 if (firstSpace > 0) { | |
| 218 result = result.substring(firstSpace); | |
| 219 } | |
| 220 return '...$result'; | |
| 221 } | |
| 222 | |
| 223 /// Returns a description of [liveTest]. | |
| 224 /// | |
| 225 /// This differs from the test's own description in that it may also include | |
| 226 /// the suite's name. | |
| 227 String _description(LiveTest liveTest) { | |
| 228 if (_multipleSuites) return "${liveTest.suite.name}: ${liveTest.test.name}"; | |
| 229 return liveTest.test.name; | |
| 230 } | |
| 231 } | |
| OLD | NEW |