Updating Py 3 runner so that it continues to work if unittest WritelnDecorator disappear

gregmalcolm · gregmalcolm · commit ec0f96e32d86 · 2010-08-04T23:52:50.000-04:00
diff --git a/python 3/runner/mountain.py b/python 3/runner/mountain.py
@@ -6,10 +6,11 @@
 
 from . import path_to_enlightenment
 from .sensei import Sensei
+from .writeln_decorator import WritelnDecorator
 
 class Mountain:
     def __init__(self):
-        self.stream = unittest._WritelnDecorator(sys.stdout)
+        self.stream = WritelnDecorator(sys.stdout)
         self.tests = path_to_enlightenment.koans()
         self.lesson = Sensei(self.stream)
     
diff --git a/python 3/runner/runner_tests/test_sensei.py b/python 3/runner/runner_tests/test_sensei.py
@@ -8,6 +8,7 @@
 from libs.mock import *
 
 from runner.sensei import Sensei
+from runner.writeln_decorator import WritelnDecorator
 from runner.mockable_test_result import MockableTestResult
 
 class AboutParrots:
@@ -83,7 +84,7 @@ class AboutFreemasons:
 class TestSensei(unittest.TestCase):
 
     def setUp(self):
-        self.sensei = Sensei(unittest._WritelnDecorator(sys.stdout))
+        self.sensei = Sensei(WritelnDecorator(sys.stdout))
         self.sensei.stream.writeln = Mock()
         
     def test_that_it_delegates_testing_to_test_cases(self):
