#22 Create TestSuite from TestCase

src/no_op_test_case.py

@@ -0,0 +1,9 @@
+from test_case import TestCase
+
+
+class NoOpTestCase(TestCase):
+    def testMethod(self) -> None:
+        pass
+
+    def testMethod2(self) -> None:
+        pass

src/test_case.py

@@ -1,3 +1,6 @@
+from test_suite import TestSuite
+
+
 class TestCase:
 
     def __init__(self, name) -> None:
@@ -28,3 +31,15 @@ def run(self, result) -> None:
 
     def tearDown(self) -> None:
         pass
+
+    def getTestNames(self):
+        testNames = [testName for testName in dir(
+            self) if testName.startswith("test")]
+        return testNames
+
+    def asSuite(self):
+        suite = TestSuite()
+        testNames = self.getTestNames()
+        for testName in testNames:
+            suite.add(self.__class__(testName))
+        return suite

src/test_case_test.py

@@ -4,6 +4,7 @@
 from broken_setup import TestCaseWithBrokenSetup
 from test_suite import TestSuite
 from broken_teardown import TestCaseWithBrokenTearDown
+from no_op_test_case import NoOpTestCase
 
 
 class TestCaseTest(TestCase):
@@ -48,16 +49,20 @@ def testTearDownCalledEvenIfTestFails(self) -> None:
         test.run(self.result)
         assert ("setUp testBrokenMethod tearDown " == test.log)
 
+    def testCollectAllTestNames(self) -> None:
+        test = NoOpTestCase('testMethod')
+        assert (test.getTestNames() == ["testMethod", "testMethod2"])
 
-suite = TestSuite()
-suite.add(TestCaseTest("testTemplateMethod"))
-suite.add(TestCaseTest("testFailedResultFormatting"))
-suite.add(TestCaseTest("testFailedResult"))
-suite.add(TestCaseTest("testFailedSetUp"))
-suite.add(TestCaseTest("testSuite"))
-suite.add(TestCaseTest("testFailedTearDown"))
-suite.add(TestCaseTest("testTearDownCalledEvenIfTestFails"))
+    def testReturnsTestSuite(self) -> None:
+        suite = NoOpTestCase('testMethod').asSuite()
+        assert (len(suite.tests) == 2)
 
+        result = TestResult()
+        suite.run(result)
+        assert (result.summary() == "2 run, 0 failed")
+
+
+suite = TestCaseTest('anything').asSuite()
 result = TestResult()
 suite.run(result)
 print(result.summary())