#2 Configure a setUp() method

kaiosilveira · Mar 18, 2023 · 7bc9bd9 · 7bc9bd9
2 parents 94b0034 + 1de7bc9
commit 7bc9bd9
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 5 deletions.
diff --git a/src/test_case.py b/src/test_case.py
@@ -3,6 +3,10 @@ class TestCase:
     def __init__(self, name) -> None:
         self.name = name
 
+    def setUp(self) -> None:
+        pass
+
     def run(self) -> None:
+        self.setUp()
         method = getattr(self, self.name)
         method()
diff --git a/src/test_case_test.py b/src/test_case_test.py
@@ -3,11 +3,17 @@
 
 
 class TestCaseTest(TestCase):
+    def setUp(self) -> None:
+        self.test = WasRun("testMethod")
+
     def testRunning(self) -> None:
-        test = WasRun("testMethod")
-        assert (not test.wasRun)
-        test.run()
-        assert (test.wasRun)
+        self.test.run()
+        assert (self.test.wasRun)
+
+    def testSetup(self) -> None:
+        self.test.run()
+        assert (self.test.wasSetUp)
 
 
 TestCaseTest("testRunning").run()
+TestCaseTest("testSetup").run()
diff --git a/src/was_run.py b/src/was_run.py
@@ -3,8 +3,11 @@
 
 class WasRun(TestCase):
     def __init__(self, name) -> None:
-        self.wasRun = None
         TestCase.__init__(self, name)
 
+    def setUp(self) -> None:
+        self.wasRun = None
+        self.wasSetUp = 1
+
     def testMethod(self) -> None:
         self.wasRun = 1