Add JUnit XML style output to test command.

Fred Ross · Fred Ross · commit e3413fdbc295 · 2014-11-18T09:39:02.000-08:00
diff --git a/setup.py b/setup.py
@@ -34,6 +34,17 @@ def run_test_suite():
     unittest.TextTestRunner().run(suite)
     os.chdir(original_cwd)
 
+def run_test_suite_with_junit_output():
+    try:
+        import unittest2 as unittest
+    except ImportError:
+        import unittest
+    import xmlrunner
+    original_cwd = os.path.abspath(os.getcwd())
+    os.chdir('tests')
+    suite = unittest.defaultTestLoader.discover('.')
+    xmlrunner.XMLTestRunner(output='../test-reports').run(suite)
+    os.chdir(original_cwd)
 
 class CoverageCommand(Command):
     """setup.py command to run code coverage of the test suite."""
@@ -73,6 +84,20 @@ def finalize_options(self):
     def run(self):
         run_test_suite()
 
+class JunitXmlTestCommand(Command):
+    """setup.py command to run the whole test suite."""
+    description = "Run test full test suite with JUnit-formatted output."
+    user_options = []
+
+    def initialize_options(self):
+        pass
+
+    def finalize_options(self):
+        pass
+
+    def run(self):
+        run_test_suite_with_junit_output()
+
 
 class DistCommand(Command):
     """setup.py command to create .spl files for modular input and search
@@ -175,6 +200,7 @@ def exclude(path):
 
     cmdclass={'coverage': CoverageCommand,
               'test': TestCommand,
+              'testjunit': JunitXmlTestCommand,
               'dist': DistCommand},
 
     description="The Splunk Software Development Kit for Python.",
@@ -202,4 +228,4 @@ def exclude(path):
         "Topic :: Software Development :: Libraries :: Python Modules",
         "Topic :: Software Development :: Libraries :: Application Frameworks",
     ],
-)
+)
