diff --git a/tests/run-tests.py b/tests/run-tests.py index 1e71a27dee970ec9ef02c004f662b5a99182f69b_dGVzdHMvcnVuLXRlc3RzLnB5..81b12f69ef5b10d2c5a5870eeab6c4c3d07e1a6d_dGVzdHMvcnVuLXRlc3RzLnB5 100755 --- a/tests/run-tests.py +++ b/tests/run-tests.py @@ -94,4 +94,5 @@ try: # is pygments installed import pygments import pygments.lexers as lexers + import pygments.lexer as lexer import pygments.formatters as formatters @@ -97,7 +98,9 @@ import pygments.formatters as formatters + import pygments.token as token + import pygments.style as style pygmentspresent = True difflexer = lexers.DiffLexer() terminal256formatter = formatters.Terminal256Formatter() except ImportError: pass @@ -98,9 +101,41 @@ pygmentspresent = True difflexer = lexers.DiffLexer() terminal256formatter = formatters.Terminal256Formatter() except ImportError: pass +if pygmentspresent: + class TestRunnerStyle(style.Style): + default_style = "" + skipped = token.string_to_tokentype("Token.Generic.Skipped") + failed = token.string_to_tokentype("Token.Generic.Failed") + error = token.string_to_tokentype("Token.Generic.Error") + skippedname = token.string_to_tokentype("Token.Generic.SName") + failedname = token.string_to_tokentype("Token.Generic.FName") + styles = { + skipped: '#e5e5e5', + skippedname: '#00ffff', + failed: '#7f0000', + failedname: '#ff0000', + } + + class TestRunnerLexer(lexer.RegexLexer): + tokens = { + 'root': [ + (r'^Skipped', token.Generic.Skipped, 'skipped'), + (r'^Failed ', token.Generic.Failed, 'failed'), + (r'^ERROR: ', token.Generic.Failed, 'failed'), + ], + 'skipped': [ + (r'[\w-]+\.t', token.Generic.SName), + (r':.*', token.Generic.Skipped), + ], + 'failed': [ + (r'[\w-]+\.t', token.Generic.FName), + (r'(:| ).*', token.Generic.Failed), + ] + } + if sys.version_info > (3, 5, 0): PYTHON3 = True xrange = range # we use xrange in one place, and we'd rather not use range @@ -1598,7 +1633,14 @@ self.stream.write('t') else: if not self._options.nodiff: - self.stream.write('\nERROR: %s output changed\n' % test) + formatted = '\nERROR: %s output changed\n' % test + if self.color: + formatted = pygments.highlight( + formatted, + TestRunnerLexer(), + formatters.Terminal256Formatter( + style=TestRunnerStyle)) + self.stream.write(formatted) self.stream.write('!') self.stream.flush() @@ -2000,5 +2042,12 @@ if not self._runner.options.noskips: for test, msg in result.skipped: - self.stream.writeln('Skipped %s: %s' % (test.name, msg)) + formatted = 'Skipped %s: %s' % (test.name, msg) + if result.color: + formatted = pygments.highlight( + formatted, + TestRunnerLexer(), + formatters.Terminal256Formatter( + style=TestRunnerStyle)).strip("\n") + self.stream.writeln(formatted) for test, msg in result.failures: @@ -2004,5 +2053,12 @@ for test, msg in result.failures: - self.stream.writeln('Failed %s: %s' % (test.name, msg)) + formatted = 'Failed %s: %s' % (test.name, msg) + if result.color: + formatted = pygments.highlight( + formatted, + TestRunnerLexer(), + formatters.Terminal256Formatter( + style=TestRunnerStyle)).strip("\n") + self.stream.writeln(formatted) for test, msg in result.errors: self.stream.writeln('Errored %s: %s' % (test.name, msg)) diff --git a/tests/test-run-tests.t b/tests/test-run-tests.t index 1e71a27dee970ec9ef02c004f662b5a99182f69b_dGVzdHMvdGVzdC1ydW4tdGVzdHMudA==..81b12f69ef5b10d2c5a5870eeab6c4c3d07e1a6d_dGVzdHMvdGVzdC1ydW4tdGVzdHMudA== 100644 --- a/tests/test-run-tests.t +++ b/tests/test-run-tests.t @@ -131,6 +131,5 @@ \x1b[38;5;34m+ bar*baz (glob)\x1b[39m (esc) bar*bad (glob) \x1b[38;5;124m- bar*baz (glob)\x1b[39m (esc) - - ERROR: test-failure.t output changed + \x1b[38;5;88mERROR: \x1b[39m\x1b[38;5;9mtest-failure.t\x1b[39m\x1b[38;5;88m output changed\x1b[39m (esc) ! @@ -136,5 +135,5 @@ ! - Failed test-failure.t: output changed + \x1b[38;5;88mFailed \x1b[39m\x1b[38;5;9mtest-failure.t\x1b[39m\x1b[38;5;88m: output changed\x1b[39m (esc) # Ran 1 tests, 0 skipped, 1 failed. python hash seed: * (glob) [1]