Tests decorator: save case name

Save the case name in the "unittest.TestCase" object when executing
test cases through the "run_sub_tests_with_dataset" decorator.
This allows dumping directories with names specific to the test case
that can be used for debugging as showed in
test_updater_key_rotations.py

Signed-off-by: Martin Vrachev <mvrachev@vmware.com>

Author: Martin Vrachev

tests/test_updater_key_rotations.py

@@ -40,7 +40,6 @@ class TestUpdaterKeyRotations(unittest.TestCase):
     def setUpClass(cls) -> None:
         cls.sim: RepositorySimulator
         cls.metadata_dir: str
-        cls.subtest_count = 0
         # pylint: disable-next=consider-using-with
         cls.temp_dir = tempfile.TemporaryDirectory()
 
@@ -57,16 +56,14 @@ def tearDownClass(cls) -> None:
         cls.temp_dir.cleanup()
 
     def setup_subtest(self) -> None:
-        self.subtest_count += 1
-
         # Setup repository for subtest: make sure no roots have been published
         self.sim = RepositorySimulator()
         self.sim.signed_roots.clear()
         self.sim.root.version = 0
 
         if self.dump_dir is not None:
             # create subtest dumpdir
-            name = f"{self.id().split('.')[-1]}-{self.subtest_count}"
+            name = f"{self.id().split('.')[-1]}-{self.case_name}"
             self.sim.dump_dir = os.path.join(self.dump_dir, name)
             os.mkdir(self.sim.dump_dir)
 

tests/utils.py

@@ -57,6 +57,8 @@ def real_decorator(
         def wrapper(test_cls: unittest.TestCase) -> None:
             for case, data in dataset.items():
                 with test_cls.subTest(case=case):
+                    # Save case name for future reference
+                    test_cls.case_name = case.replace(" ", "_")
                     function(test_cls, data)
         return wrapper
     return real_decorator