Code refactoring

0--key · 0--key · commit 77ffc22b5244 · 2017-02-10T18:34:55.000+02:00
diff --git a/tests/test_facade.py b/tests/test_facade.py
@@ -1,41 +1,54 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 import unittest
-from structural.facade import TestRunner
+from structural.facade import TestRunner, TC1, TC2, TC3
 
 
 class TestRunnerFacilities(unittest.TestCase):
 
     def setUp(self):
-        self.runner = TestRunner()
-        self.average_result = """###### In Test 1 ######
-Setting up
-Running test
-Tearing down
-Test Finished
-
-###### In Test 2 ######
-Setting up
-Running test
-Tearing down
-Test Finished
-
-###### In Test 3 ######
-Setting up
-Running test
-Tearing down
-Test Finished"""
-
-    def test_bunch_launch(self):
+        self.tc = TC1
+        self.average_result_tc1 = "###### In Test 1 ######\n" + \
+                                  "Setting up\n" + \
+                                  "Running test\n" + \
+                                  "Tearing down\n" + \
+                                  "Test Finished"
+
+    def test_tc1_output(self):
         import sys
         try:
             from io import StringIO
         except:
             from StringIO import StringIO
         out = StringIO()
         sys.stdout = out
-        self.runner.runAll()
+        self.tc.run(self)
         output = out.getvalue().strip()
-        self.assertEqual(output, self.average_result)
+        self.assertEqual(output, self.average_result_tc1)
+
+
+# ###### In Test 2 ######
+# Setting up
+# Running test
+# Tearing down
+# Test Finished
+
+# ###### In Test 3 ######
+# Setting up
+# Running test
+# Tearing down
+# Test Finished"""
+
+    # def test_bunch_launch(self):
+    #     import sys
+    #     try:
+    #         from io import StringIO
+    #     except:
+    #         from StringIO import StringIO
+    #     out = StringIO()
+    #     sys.stdout = out
+    #     self.runner.runAll()
+    #     output = out.getvalue().strip()
+    #     self.assertEqual(output, self.average_result)
 
 
