added log message count case in set_status tests

Author: Narek Mkhitaryan

tests/integration/folders/test_set_folder_status.py

@@ -14,6 +14,7 @@ class TestSetFolderStatus(TestCase):
     FOLDER_NAME = "test_folder"
     PROJECT_DESCRIPTION = "desc"
     PROJECT_TYPE = "Vector"
+    FOLDER_STATUSES = ["NotStarted", "InProgress", "Completed", "OnHold"]
 
     @classmethod
     def setUpClass(cls, *args, **kwargs):
@@ -34,7 +35,7 @@ def tearDownClass(cls) -> None:
     def test_set_folder_status(self):
         with self.assertLogs("sa", level="INFO") as cm:
             for index, status in enumerate(
-                ["NotStarted", "InProgress", "Completed", "OnHold"]
+                self.FOLDER_STATUSES
             ):
                 sa.set_folder_status(
                     project=self.PROJECT_NAME, folder=self.FOLDER_NAME, status=status
@@ -47,6 +48,7 @@ def test_set_folder_status(self):
                     == cm.output[index]
                 )
                 self.assertEqual(status, folder["status"])
+            self.assertEqual(len(cm.output), len(self.FOLDER_STATUSES))
 
     @patch("lib.infrastructure.services.folder.FolderService.update")
     def test_set_folder_status_fail(self, update_function):

tests/integration/projects/test_set_project_status.py

@@ -13,6 +13,7 @@ class TestSetProjectStatus(TestCase):
     PROJECT_NAME = "test_set_project_status"
     PROJECT_DESCRIPTION = "desc"
     PROJECT_TYPE = "Vector"
+    PROJECT_STATUSES = ["NotStarted", "InProgress", "Completed", "OnHold"]
 
     @classmethod
     def setUpClass(cls, *args, **kwargs):
@@ -30,7 +31,7 @@ def tearDownClass(cls) -> None:
     def test_set_project_status(self):
         with self.assertLogs("sa", level="INFO") as cm:
             for index, status in enumerate(
-                ["NotStarted", "InProgress", "Completed", "OnHold"]
+                self.PROJECT_STATUSES
             ):
                 sa.set_project_status(project=self.PROJECT_NAME, status=status)
                 project = sa.get_project_metadata(self.PROJECT_NAME)
@@ -39,6 +40,7 @@ def test_set_project_status(self):
                     == cm.output[index]
                 )
                 self.assertEqual(status, project["status"])
+            self.assertEqual(len(cm.output), len(self.PROJECT_STATUSES))
 
     @patch("lib.infrastructure.services.project.ProjectService.update")
     def test_set_project_status_fail(self, update_function):