summaryrefslogtreecommitdiff
path: root/python/subunit
diff options
context:
space:
mode:
Diffstat (limited to 'python/subunit')
-rw-r--r--python/subunit/__init__.py13
-rwxr-xr-xpython/subunit/run.py5
2 files changed, 16 insertions, 2 deletions
diff --git a/python/subunit/__init__.py b/python/subunit/__init__.py
index 73dbc17..6b65ae4 100644
--- a/python/subunit/__init__.py
+++ b/python/subunit/__init__.py
@@ -1111,3 +1111,16 @@ class TestResultStats(unittest.TestResult):
def wasSuccessful(self):
"""Tells whether or not this result was a success"""
return self.failed_tests == 0
+
+
+def get_default_formatter():
+ """Obtain the default formatter to write to.
+
+ :return: A file-like object.
+ """
+ formatter = os.getenv("SUBUNIT_FORMATTER")
+ if formatter:
+ return os.popen(formatter, "w")
+ else:
+ return sys.stdout
+
diff --git a/python/subunit/run.py b/python/subunit/run.py
index 2b90791..01c0b0e 100755
--- a/python/subunit/run.py
+++ b/python/subunit/run.py
@@ -22,7 +22,7 @@
import sys
-from subunit import TestProtocolClient
+from subunit import TestProtocolClient, get_default_formatter
class SubunitTestRunner(object):
@@ -41,6 +41,7 @@ if __name__ == '__main__':
from unittest import TestProgram
parser = optparse.OptionParser(__doc__)
args = parser.parse_args()[1]
- runner = SubunitTestRunner()
+ stream = get_default_formatter()
+ runner = SubunitTestRunner(stream)
program = TestProgram(module=None, argv=[sys.argv[0]] + args,
testRunner=runner)