feat(jira): resolve comments

prowler-cloud · Nov 8, 2024 · 08e9d49 · 08e9d49
1 parent c973636
commit 08e9d49
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/lib/outputs/jira/jira_test.py b/tests/lib/outputs/jira/jira_test.py
@@ -1032,8 +1032,10 @@ def test_send_findings_response_error(
                 findings=[finding], project_key="TEST", issue_type="Bug"
             )
 
-    def test_get_color_from_status(self):
+    @pytest.mark.parametrize(
+        "status, expected_color",
+        [("FAIL", "#FF0000"), ("PASS", "#008000"), ("MUTED", "#FFA500")],
+    )
+    def test_get_color_from_status(self, status, expected_color):
         """Test that get_color_from_status returns the correct color for a status."""
-        assert self.jira_integration.get_color_from_status("FAIL") == "#FF0000"
-        assert self.jira_integration.get_color_from_status("PASS") == "#008000"
-        assert self.jira_integration.get_color_from_status("MUTED") == "#FFA500"
+        assert self.jira_integration.get_color_from_status(status) == expected_color