1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
# coding: utf-8
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
"""TestRunner and TestResult for gabbi-run."""
from unittest import TestResult
from unittest import TextTestResult
from unittest import TextTestRunner
import pytest
from gabbi import utils
class ConciseTestResult(TextTestResult):
"""A TextTestResult with simple but useful output.
If the output is a tty or GABBI_FORCE_COLOR is set in the
environment, output will be colorized.
"""
def __init__(self, stream, descriptions, verbosity):
super(ConciseTestResult, self).__init__(
stream, descriptions, verbosity)
self.colorize = utils.get_colorizer(stream)
def startTest(self, test):
super(TextTestResult, self).startTest(test)
if self.showAll:
self.stream.write('... ')
self.stream.flush()
def addSuccess(self, test):
super(TextTestResult, self).addSuccess(test)
if self.showAll:
self.stream.write(self.colorize('GREEN', '✓ '))
self.stream.writeln(self.getDescription(test))
def addFailure(self, test, err):
super(TextTestResult, self).addFailure(test, err)
if self.showAll:
self.stream.write(self.colorize('RED', '✗ '))
self.stream.writeln(self.getDescription(test))
def addError(self, test, err):
super(TextTestResult, self).addError(test, err)
if self.showAll:
self.stream.write(self.colorize('RED', 'E '))
self.stream.writeln(self.getDescription(test))
def addSkip(self, test, reason):
super(TextTestResult, self).addSkip(test, reason)
if self.showAll:
self.stream.write('- ')
self.stream.writeln(self.getDescription(test))
self.stream.writeln('\t[skipped] {0!r}'.format(reason))
def addExpectedFailure(self, test, err):
super(TextTestResult, self).addExpectedFailure(test, err)
if self.showAll:
self.stream.write('o ')
self.stream.writeln(self.getDescription(test))
self.stream.writeln('\t[expected failure]')
def addUnexpectedSuccess(self, test):
super(TextTestResult, self).addUnexpectedSuccess(test)
if self.showAll:
self.stream.write('! ')
self.stream.writeln(self.getDescription(test))
self.stream.writeln('\t[unexpected success]')
def getDescription(self, test):
# Chop the test method ('test_request') off the test.id().
name = test.id().rsplit('.', 1)[0]
desc = test.test_data.get('desc', None)
return ': '.join((name, desc)) if desc else name
def _exc_info_to_string(self, err, test):
"""Override exception to string handling
The default does too much. We don't want doctoring. We want
information!
"""
return err
def printErrorList(self, flavor, errors):
for test, err in errors:
# err[0] is the type of exception
# err[1] is the args of the exception
# err[3] is the traceback, not currently used
self.stream.writeln('%s: %s' % (flavor, self.getDescription(test)))
message = str(err[1])
for line in message.splitlines():
self.stream.writeln('\t%s' % line)
class PyTestResult(TestResult):
"""Wrap a test result to allow it to work with pytest.
The main behaviors here are:
* to turn what had been exceptions back into exceptions
* use pytest's skip and xfail methods
"""
def addFailure(self, test, err):
raise err[1]
def addError(self, test, err):
raise err[1]
def addSkip(self, test, reason):
pytest.skip(reason)
def addExpectedFailure(self, test, err):
pytest.xfail('%s' % err[1])
class ConciseTestRunner(TextTestRunner):
"""A TextTestRunner that uses ConciseTestResult for reporting results."""
resultclass = ConciseTestResult
|
