Add stream output support for the UI.

3 files changed, 20 insertions, 0 deletions

diff --git a/testrepository/ui/__init__.py b/testrepository/ui/__init__.py
index eac93a1..d6d4339 100644
--- a/testrepository/ui/__init__.py
+++ b/testrepository/ui/__init__.py
@@ -102,6 +102,17 @@ class AbstractUI(object):
         """
         raise NotImplementedError(self.output_results)
 
+    def output_stream(self, stream):
+        """Show a byte stream to the user.
+
+        This is not currently typed, but in future a MIME type may be
+        permitted.
+
+        :param stream: A file like object that can be read from. The UI will
+        not close the file.
+        """
+        raise NotImplementedError(self.output_results)
+
     def output_table(self, table):
         """Show a table to the user.
 
diff --git a/testrepository/ui/cli.py b/testrepository/ui/cli.py
index 0ad4cd3..3f2ff81 100644
--- a/testrepository/ui/cli.py
+++ b/testrepository/ui/cli.py
@@ -77,6 +77,12 @@ class UI(ui.AbstractUI):
         finally:
             result.stopTestRun()
 
+    def output_stream(self, stream):
+        contents = stream.read(65536)
+        while contents:
+            self._stdout.write(contents)
+            contents = stream.read(65536)
+
     def output_table(self, table):
         # stringify
         contents = []
diff --git a/testrepository/ui/model.py b/testrepository/ui/model.py
index 267cdd5..494c4e5 100644
--- a/testrepository/ui/model.py
+++ b/testrepository/ui/model.py
@@ -81,6 +81,9 @@ class UI(ui.AbstractUI):
     def output_results(self, suite_or_test):
         self.outputs.append(('results', suite_or_test))
 
+    def output_stream(self, stream):
+        self.outputs.append(('stream', stream))
+
     def output_table(self, table):
         self.outputs.append(('table', table))