Improve live printing for --test-repositories test

tests/test.py

@@ -465,16 +465,18 @@ class TestContainer(object):
             A file-like object used for diagnostic output that provides .write()
             and .flush()
         """
+        # TODO multi-threading
+        stream.write("%s ... " % path)
+        stream.flush()
 
-        cls._write(stream, "%s ... " % path)
-
+        success = False
+        try:
             if re.match('https?://', path, re.I) is not None:
                 # Download the repository
                 try:
                     with urlopen(path) as f:
                         source = f.read().decode("utf-8", 'replace')
                 except Exception as e:
-                cls._write(stream, 'failed (%s)\n' % str_cls(e))
                     yield cls._fail("Downloading %s failed" % path, e)
                     return
             else:
@@ -482,7 +484,6 @@ class TestContainer(object):
                     with _open(path) as f:
                         source = f.read().decode('utf-8', 'replace')
                 except Exception as e:
-                cls._write(stream, 'failed (%s)\n' % str_cls(e))
                     yield cls._fail("Opening %s failed" % path, e)
                     return
 
@@ -507,13 +508,20 @@ class TestContainer(object):
                 yield cls._fail("Unrecognized schema version %s in %s"
                                 % (schema, path))
                 return
+
+            success = True
+
             # Do not generate 1000 failing tests for not yet updated repos
             if schema != '3.0.0':
-            cls._write(stream, "skipping (schema version %s)\n"
+                stream.write("skipping (schema version %s)"
                              % data['schema_version'])
                 return
-
-        cls._write(stream, 'done\n')
+            else:
+                stream.write("done")
+        finally:
+            if not success:
+                stream.write("failed")
+            stream.write("\n")
 
         # `path` is for output during tests only
         yield cls._test_repository_keys, (path, data)
@@ -544,21 +552,6 @@ class TestContainer(object):
 
         return cls._test_error, args
 
-    @classmethod
-    def _write(cls, stream, string):
-        """
-        Writes dianostic output to a file-like object.
-
-        :param stream:
-            Must have the methods .write() and .flush()
-
-        :param string:
-            The string to write - a newline will NOT be appended
-        """
-
-        stream.write(string)
-        stream.flush()
-
 
 class DefaultChannelTests(TestContainer, unittest.TestCase):
     maxDiff = None
@@ -589,7 +582,7 @@ class DefaultChannelTests(TestContainer, unittest.TestCase):
             # when run with "--test-repositories" parameter.
             return
 
-        cls._write(stream, "Fetching remote repositories:\n")
+        stream.write("Fetching remote repositories:\n")
 
         for repository in cls.j['repositories']:
             if repository.startswith('.'):
@@ -600,7 +593,8 @@ class DefaultChannelTests(TestContainer, unittest.TestCase):
             for test in cls._include_tests(repository, stream):
                 yield test
 
-        cls._write(stream, '\n')
+        stream.write('\n')
+        stream.flush()
 
 
 class DefaultRepositoryTests(TestContainer, unittest.TestCase):