chore: Run formatter

intuitem · Feb 28, 2024 · 309f582 · 309f582
1 parent f0e62af
commit 309f582
Show file tree

Hide file tree

Showing 36 changed files with 2,416 additions and 805 deletions.
diff --git a/backend/app_tests/api/test_api_assets.py b/backend/app_tests/api/test_api_assets.py
@@ -79,7 +79,12 @@ def test_delete_assets(self):
 
 
 @pytest.mark.django_db
-@pytest.mark.parametrize("test", GROUPS_PERMISSIONS.keys(), ids=[GROUPS_PERMISSIONS[key]["name"] for key in GROUPS_PERMISSIONS.keys()], indirect=True)
+@pytest.mark.parametrize(
+    "test",
+    GROUPS_PERMISSIONS.keys(),
+    ids=[GROUPS_PERMISSIONS[key]["name"] for key in GROUPS_PERMISSIONS.keys()],
+    indirect=True,
+)
 class TestAssetsAuthenticated:
     """Perform tests on Assets API endpoint with authentication"""
 
@@ -97,7 +102,10 @@ def test_get_assets(self, test):
                 "type": ASSET_TYPE[0],
                 "folder": test.folder,
             },
-            {"folder": {"id": str(test.folder.id), "str": test.folder.name}, "type": ASSET_TYPE[1]},
+            {
+                "folder": {"id": str(test.folder.id), "str": test.folder.name},
+                "type": ASSET_TYPE[1],
+            },
             user_group=test.user_group,
         )
 
@@ -116,7 +124,10 @@ def test_create_assets(self, test):
                 "parent_assets": [],
                 "folder": str(test.folder.id),
             },
-            {"folder": {"id": str(test.folder.id), "str": test.folder.name}, "type": ASSET_TYPE[1]},
+            {
+                "folder": {"id": str(test.folder.id), "str": test.folder.name},
+                "type": ASSET_TYPE[1],
+            },
             user_group=test.user_group,
         )
 
@@ -174,7 +185,10 @@ def test_update_assets(self, test):
                 "type": ASSET_TYPE2[0],
                 "folder": str(folder.id),
             },
-            {"folder": {"id": str(test.folder.id), "str": test.folder.name}, "type": ASSET_TYPE[1]},
+            {
+                "folder": {"id": str(test.folder.id), "str": test.folder.name},
+                "type": ASSET_TYPE[1],
+            },
             user_group=test.user_group,
         )
 

diff --git a/backend/app_tests/api/test_api_compliance_assessments.py b/backend/app_tests/api/test_api_compliance_assessments.py
@@ -97,17 +97,20 @@ def test_delete_compliance_assessments(self, authenticated_client):
 
 
 @pytest.mark.django_db
-@pytest.mark.parametrize("test", GROUPS_PERMISSIONS.keys(), ids=[GROUPS_PERMISSIONS[key]["name"] for key in GROUPS_PERMISSIONS.keys()], indirect=True)
+@pytest.mark.parametrize(
+    "test",
+    GROUPS_PERMISSIONS.keys(),
+    ids=[GROUPS_PERMISSIONS[key]["name"] for key in GROUPS_PERMISSIONS.keys()],
+    indirect=True,
+)
 class TestComplianceAssessmentsAuthenticated:
     """Perform tests on ComplianceAssessments API endpoint with authentication"""
 
     def test_get_compliance_assessments(self, test):
         """test to get compliance assessments from the API with authentication"""
 
         EndpointTestsQueries.Auth.import_object(test.admin_client, "Framework")
-        project = Project.objects.create(
-            name="test", folder=test.folder
-        )
+        project = Project.objects.create(name="test", folder=test.folder)
 
         EndpointTestsQueries.Auth.get_object(
             test.client,
@@ -134,9 +137,7 @@ def test_create_compliance_assessments(self, test):
         """test to create compliance assessments with the API with authentication"""
 
         EndpointTestsQueries.Auth.import_object(test.admin_client, "Framework")
-        project = Project.objects.create(
-            name="test", folder=test.folder
-        )
+        project = Project.objects.create(name="test", folder=test.folder)
 
         EndpointTestsQueries.Auth.create_object(
             test.client,
@@ -165,9 +166,7 @@ def test_update_compliance_assessments(self, test):
         EndpointTestsQueries.Auth.import_object(test.admin_client, "Documents")
         EndpointTestsQueries.Auth.import_object(test.admin_client, "Framework")
         EndpointTestsQueries.Auth.import_object(test.admin_client, "Framework2")
-        project = Project.objects.create(
-            name="test", folder=test.folder
-        )
+        project = Project.objects.create(name="test", folder=test.folder)
         project2 = Project.objects.create(
             name="test2", folder=Folder.objects.create(name="test2")
         )
@@ -204,9 +203,7 @@ def test_delete_compliance_assessments(self, test):
         """test to delete compliance assessments with the API with authentication"""
 
         EndpointTestsQueries.Auth.import_object(test.admin_client, "Framework")
-        project = Project.objects.create(
-            name="test", folder=test.folder
-        )
+        project = Project.objects.create(name="test", folder=test.folder)
 
         EndpointTestsQueries.Auth.delete_object(
             test.client,

diff --git a/backend/app_tests/api/test_api_evidences.py b/backend/app_tests/api/test_api_evidences.py
@@ -92,14 +92,21 @@ def test_delete_evidences(self):
 
 
 @pytest.mark.django_db
-@pytest.mark.parametrize("test", GROUPS_PERMISSIONS.keys(), ids=[GROUPS_PERMISSIONS[key]["name"] for key in GROUPS_PERMISSIONS.keys()], indirect=True)
+@pytest.mark.parametrize(
+    "test",
+    GROUPS_PERMISSIONS.keys(),
+    ids=[GROUPS_PERMISSIONS[key]["name"] for key in GROUPS_PERMISSIONS.keys()],
+    indirect=True,
+)
 class TestEvidencesAuthenticated:
     """Perform tests on Evidences API endpoint with authentication"""
 
     def test_get_evidences(self, test):
         """test to get evidences from the API with authentication"""
 
-        security_measure = SecurityMeasure.objects.create(name="test", folder=test.folder)
+        security_measure = SecurityMeasure.objects.create(
+            name="test", folder=test.folder
+        )
 
         EndpointTestsQueries.Auth.get_object(
             test.client,
@@ -127,7 +134,9 @@ def test_get_evidences(self, test):
     def test_create_evidences(self, test):
         """test to create evidences with the API with authentication"""
 
-        security_measure = SecurityMeasure.objects.create(name="test", folder=test.folder)
+        security_measure = SecurityMeasure.objects.create(
+            name="test", folder=test.folder
+        )
 
         with open(
             path.join(path.dirname(path.dirname(__file__)), EVIDENCE_ATTACHMENT), "rb"
@@ -162,7 +171,9 @@ def test_update_evidences(self, test):
         """test to update evidences with the API with authentication"""
 
         folder = Folder.objects.create(name="test2")
-        security_measure = SecurityMeasure.objects.create(name="test", folder=test.folder)
+        security_measure = SecurityMeasure.objects.create(
+            name="test", folder=test.folder
+        )
         security_measure2 = SecurityMeasure.objects.create(name="test2", folder=folder)
 
         with open(

diff --git a/backend/app_tests/api/test_api_folders.py b/backend/app_tests/api/test_api_folders.py
@@ -62,7 +62,12 @@ def test_delete_folders(self):
 
 
 @pytest.mark.django_db
-@pytest.mark.parametrize("test", GROUPS_PERMISSIONS.keys(), ids=[GROUPS_PERMISSIONS[key]["name"] for key in GROUPS_PERMISSIONS.keys()], indirect=True)
+@pytest.mark.parametrize(
+    "test",
+    GROUPS_PERMISSIONS.keys(),
+    ids=[GROUPS_PERMISSIONS[key]["name"] for key in GROUPS_PERMISSIONS.keys()],
+    indirect=True,
+)
 class TestFoldersAuthenticated:
     """Perform tests on Folders API endpoint with authentication"""
 
@@ -74,7 +79,7 @@ def test_get_folders(self, test):
             "Folders",
             Folder,
             {
-                "name": FOLDER_NAME, 
+                "name": FOLDER_NAME,
                 "description": FOLDER_DESCRIPTION,
                 "parent_folder": test.folder,
             },

diff --git a/backend/app_tests/api/test_api_libraries.py b/backend/app_tests/api/test_api_libraries.py
@@ -44,7 +44,12 @@ def test_delete_risk_matrix(self, authenticated_client):
 
 
 @pytest.mark.django_db
-@pytest.mark.parametrize("test", GROUPS_PERMISSIONS.keys(), ids=[GROUPS_PERMISSIONS[key]["name"] for key in GROUPS_PERMISSIONS.keys()], indirect=True)
+@pytest.mark.parametrize(
+    "test",
+    GROUPS_PERMISSIONS.keys(),
+    ids=[GROUPS_PERMISSIONS[key]["name"] for key in GROUPS_PERMISSIONS.keys()],
+    indirect=True,
+)
 class TestLibrariesAuthenticated:
     """Perform tests on Libraries API endpoint with authentication"""
 
@@ -58,7 +63,6 @@ def test_get_libraries(self, test):
     def test_import_frameworks(self, test):
         """test to import frameworks with the API with authentication"""
 
-
         # Uses the API endpoint to get library details with the admin client
         lib_detail_response = test.admin_client.get(
             EndpointTestsUtils.get_object_urn("Framework")
@@ -68,17 +72,28 @@ def test_import_frameworks(self, test):
         assert (
             Framework.objects.all().count() == 0
         ), "libraries are already imported in the database"
-        EndpointTestsQueries.Auth.get_object(test.client, "Frameworks", user_group=test.user_group)
-
-        EndpointTestsQueries.Auth.import_object(test.client, "Framework", user_group=test.user_group)
+        EndpointTestsQueries.Auth.get_object(
+            test.client, "Frameworks", user_group=test.user_group
+        )
 
-        expect = {"BI-UG-ADM": True, "BI-UG-GAD": False, "BI-UG-GVA": False, "BI-UG-DMA": False, 
-            "BI-UG-ANA": False, "BI-UG-VAL": False, "BI-UG-AUD": False}
+        EndpointTestsQueries.Auth.import_object(
+            test.client, "Framework", user_group=test.user_group
+        )
 
-        assert (
-            Framework.objects.all().count() == (1 if expect[test['user_group']] else 0)
+        expect = {
+            "BI-UG-ADM": True,
+            "BI-UG-GAD": False,
+            "BI-UG-GVA": False,
+            "BI-UG-DMA": False,
+            "BI-UG-ANA": False,
+            "BI-UG-VAL": False,
+            "BI-UG-AUD": False,
+        }
+
+        assert Framework.objects.all().count() == (
+            1 if expect[test["user_group"]] else 0
         ), "frameworks are not correctly imported in the database"
-        if expect[test['user_group']]:
+        if expect[test["user_group"]]:
             # Uses the API endpoint to assert that the library was properly imported
             EndpointTestsQueries.Auth.get_object(
                 test.client,
@@ -100,14 +115,27 @@ def test_delete_frameworks(self, test):
         assert (
             Framework.objects.all().count() == 1
         ), "frameworks for deletion are not correctly imported in the database"
-        expect = {"BI-UG-ADM": True, "BI-UG-GAD": False, "BI-UG-GVA": False, "BI-UG-DMA": False, 
-            "BI-UG-ANA": False, "BI-UG-VAL": False, "BI-UG-AUD": False}
-        should_work = expect[test['user_group']]
-        if should_work: # this if should be removed, but it is not working as expected, todo
+        expect = {
+            "BI-UG-ADM": True,
+            "BI-UG-GAD": False,
+            "BI-UG-GVA": False,
+            "BI-UG-DMA": False,
+            "BI-UG-ANA": False,
+            "BI-UG-VAL": False,
+            "BI-UG-AUD": False,
+        }
+        should_work = expect[test["user_group"]]
+        if (
+            should_work
+        ):  # this if should be removed, but it is not working as expected, todo
             EndpointTestsQueries.Auth.delete_object(
-                test.client, "Frameworks", Framework, user_group=test.user_group, fails=not(should_work)
+                test.client,
+                "Frameworks",
+                Framework,
+                user_group=test.user_group,
+                fails=not (should_work),
             )
-        if not should_work: # remove object
+        if not should_work:  # remove object
             EndpointTestsQueries.Auth.delete_object(
                 test.admin_client, "Frameworks", Framework
             )
@@ -124,18 +152,29 @@ def test_import_risk_matrix(self, test):
         assert (
             RiskMatrix.objects.all().count() == 0
         ), "libraries are already imported in the database"
-        EndpointTestsQueries.Auth.get_object(test.client, "Risk matrices", user_group=test.user_group)
-
-        EndpointTestsQueries.Auth.import_object(test.client, "Risk matrix", user_group=test.user_group)
+        EndpointTestsQueries.Auth.get_object(
+            test.client, "Risk matrices", user_group=test.user_group
+        )
 
-        # Uses the API endpoint to assert that the library was properly imported
-        expect = {"BI-UG-ADM": True, "BI-UG-GAD": False, "BI-UG-GVA": False, "BI-UG-DMA": False, 
-            "BI-UG-ANA": False, "BI-UG-VAL": False, "BI-UG-AUD": False}
+        EndpointTestsQueries.Auth.import_object(
+            test.client, "Risk matrix", user_group=test.user_group
+        )
 
-        assert (
-            RiskMatrix.objects.all().count() == (1 if expect[test['user_group']] else 0)
+        # Uses the API endpoint to assert that the library was properly imported
+        expect = {
+            "BI-UG-ADM": True,
+            "BI-UG-GAD": False,
+            "BI-UG-GVA": False,
+            "BI-UG-DMA": False,
+            "BI-UG-ANA": False,
+            "BI-UG-VAL": False,
+            "BI-UG-AUD": False,
+        }
+
+        assert RiskMatrix.objects.all().count() == (
+            1 if expect[test["user_group"]] else 0
         ), "Risk matrices are not correctly imported in the database"
-        if expect[test['user_group']]:
+        if expect[test["user_group"]]:
             EndpointTestsQueries.Auth.get_object(
                 test.client,
                 "Risk matrices",
@@ -154,15 +193,27 @@ def test_delete_matrix(self, test):
         """test to delete risk matrix with the API with authentication"""
 
         EndpointTestsQueries.Auth.import_object(test.admin_client, "Risk matrix")
-        expect = {"BI-UG-ADM": True, "BI-UG-GAD": False, "BI-UG-GVA": False, "BI-UG-DMA": False, 
-            "BI-UG-ANA": False, "BI-UG-VAL": False, "BI-UG-AUD": False}
-        should_work = expect[test['user_group']]
-        if should_work: # this if should be removed, but it is not working as expected, todo
+        expect = {
+            "BI-UG-ADM": True,
+            "BI-UG-GAD": False,
+            "BI-UG-GVA": False,
+            "BI-UG-DMA": False,
+            "BI-UG-ANA": False,
+            "BI-UG-VAL": False,
+            "BI-UG-AUD": False,
+        }
+        should_work = expect[test["user_group"]]
+        if (
+            should_work
+        ):  # this if should be removed, but it is not working as expected, todo
             EndpointTestsQueries.Auth.delete_object(
-                test.client, "Risk matrices", RiskMatrix, user_group=test.user_group,  fails=not(should_work)
+                test.client,
+                "Risk matrices",
+                RiskMatrix,
+                user_group=test.user_group,
+                fails=not (should_work),
             )
-        if not should_work: # remove object
+        if not should_work:  # remove object
             EndpointTestsQueries.Auth.delete_object(
                 test.admin_client, "Risk matrices", RiskMatrix
             )
-
diff --git a/backend/app_tests/api/test_api_policies.py b/backend/app_tests/api/test_api_policies.py
@@ -85,7 +85,12 @@ def test_delete_policies(self):
 
 
 @pytest.mark.django_db
-@pytest.mark.parametrize("test", GROUPS_PERMISSIONS.keys(), ids=[GROUPS_PERMISSIONS[key]["name"] for key in GROUPS_PERMISSIONS.keys()], indirect=True)
+@pytest.mark.parametrize(
+    "test",
+    GROUPS_PERMISSIONS.keys(),
+    ids=[GROUPS_PERMISSIONS[key]["name"] for key in GROUPS_PERMISSIONS.keys()],
+    indirect=True,
+)
 class TestPolicysAuthenticated:
     """Perform tests on policies API endpoint with authentication"""
 
@@ -186,27 +191,23 @@ def test_delete_policies(self, test):
             },
             user_group=test.user_group,
         )
-    
+
     def test_get_category_choices(self, test):
         """test to get policies category choices from the API with authentication"""
 
         EndpointTestsQueries.Auth.get_object_options(
-            test.client, 
-            "Policies", 
-            "category", 
-            Policy.CATEGORY, 
-            user_group=test.user_group
+            test.client,
+            "Policies",
+            "category",
+            Policy.CATEGORY,
+            user_group=test.user_group,
         )
 
     def test_get_effort_choices(self, test):
         """test to get policies effort choices from the API with authentication"""
 
         EndpointTestsQueries.Auth.get_object_options(
-            test.client, 
-            "Policies", 
-            "effort", 
-            Policy.EFFORT, 
-            user_group=test.user_group
+            test.client, "Policies", "effort", Policy.EFFORT, user_group=test.user_group
         )
 
     def test_get_status_choices(self, test):