Add ui.output_results().

3 files changed, 55 insertions, 0 deletions

diff --git a/testrepository/ui/__init__.py b/testrepository/ui/__init__.py
index 4226b74..74bd349 100644
--- a/testrepository/ui/__init__.py
+++ b/testrepository/ui/__init__.py
@@ -73,6 +73,25 @@ class AbstractUI(object):
         """Helper for iter_streams which subclasses should implement."""
         raise NotImplementedError(self._iter_streams)
 
+    def output_values(self, values):
+        """Show values to the user.
+
+        :param values: An iterable of (label, value).
+        """
+        raise NotImplementedError(self.output_values)
+
+    def output_results(self, suite_or_test):
+        """Show suite_or_test to the user by 'running' it.
+
+        This expects the run to be fast/cheap.
+
+        :param suite_or_test: A suite or test to show to the user. This should
+            obey the 'TestCase' protocol - it should have a method run(result)
+            that causes all the tests contained in the object to be handed to
+            the result object.
+        """
+        raise NotImplementedError(self.output_results)
+
     def set_command(self, cmd):
         """Inform the UI what command it is running.
 
diff --git a/testrepository/ui/cli.py b/testrepository/ui/cli.py
index 2a704d9..d5f630c 100644
--- a/testrepository/ui/cli.py
+++ b/testrepository/ui/cli.py
@@ -17,8 +17,33 @@
 from optparse import OptionParser
 import os
 
+import testtools
+
 from testrepository import ui
 
+class CLITestResult(testtools.TestResult):
+    """A TestResult for the CLI."""
+
+    def __init__(self, stream):
+        """Construct a CLITestResult writing to stream."""
+        super(CLITestResult, self).__init__()
+        self.stream = stream
+        self.sep1 = '=' * 70 + '\n'
+        self.sep2 = '-' * 70 + '\n'
+
+    def _show_list(self, label, error_list):
+        for test, output in error_list:
+            self.stream.write(self.sep1)
+            self.stream.write("%s: %s\n" % (label, test.id()))
+            self.stream.write(self.sep2)
+            self.stream.write(output)
+
+    def stopTestRun(self):
+        self._show_list('ERROR', self.errors)
+        self._show_list('FAIL', self.failures)
+        super(CLITestResult, self).stopTestRun()
+
+
 class UI(ui.AbstractUI):
     """A command line user interface."""
 
@@ -44,6 +69,14 @@ class UI(ui.AbstractUI):
             outputs.append('%s: %s' % (label, value))
         self._stdout.write('%s\n' % ' '.join(outputs))
 
+    def output_results(self, suite_or_test):
+        result = CLITestResult(self._stdout)
+        result.startTestRun()
+        try:
+            suite_or_test.run(result)
+        finally:
+            result.stopTestRun()
+
     def set_command(self, cmd):
         ui.AbstractUI.set_command(self, cmd)
         parser = OptionParser()
diff --git a/testrepository/ui/model.py b/testrepository/ui/model.py
index a287a1b..dd7d9f6 100644
--- a/testrepository/ui/model.py
+++ b/testrepository/ui/model.py
@@ -56,3 +56,6 @@ class UI(ui.AbstractUI):
 
     def output_values(self, values):
         self.outputs.append(('values', values))
+
+    def output_results(self, suite_or_test):
+        self.outputs.append(('results', suite_or_test))