From 5186e029b3fc89db58517c6f059739ee256dddb0 Mon Sep 17 00:00:00 2001
From: Pepe Fagoaga <pepe@prowler.com>
Date: Thu, 30 Jan 2025 20:48:51 +0545
Subject: [PATCH] fix(set_report_color): Add more details to error (#6751)

---
 prowler/lib/outputs/outputs.py    | 4 +++-
 tests/lib/outputs/outputs_test.py | 6 +++---
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/prowler/lib/outputs/outputs.py b/prowler/lib/outputs/outputs.py
index 2a61e17c36..b4699157d7 100644
--- a/prowler/lib/outputs/outputs.py
+++ b/prowler/lib/outputs/outputs.py
@@ -83,7 +83,9 @@ def set_report_color(status: str, muted: bool = False) -> str:
     elif status == "MANUAL":
         color = Fore.YELLOW
     else:
-        raise Exception("Invalid Report Status. Must be PASS, FAIL or MANUAL.")
+        raise Exception(
+            f"Invalid Report Status: {status}. Must be PASS, FAIL or MANUAL."
+        )
     return color
 
 
diff --git a/tests/lib/outputs/outputs_test.py b/tests/lib/outputs/outputs_test.py
index 624366af5e..dbf7d5c8bd 100644
--- a/tests/lib/outputs/outputs_test.py
+++ b/tests/lib/outputs/outputs_test.py
@@ -21,7 +21,6 @@ from prowler.lib.outputs.utils import (
 
 
 class TestOutputs:
-
     def test_set_report_color(self):
         test_status = ["PASS", "FAIL", "MANUAL"]
         test_colors = [Fore.GREEN, Fore.RED, Fore.YELLOW]
@@ -35,8 +34,9 @@ class TestOutputs:
         with pytest.raises(Exception) as exc:
             set_report_color(test_status)
 
-        assert "Invalid Report Status. Must be PASS, FAIL or MANUAL" in str(exc.value)
-        assert exc.type == Exception
+        assert "Invalid Report Status: INVALID. Must be PASS, FAIL or MANUAL" in str(
+            exc.value
+        )
 
     def test_unroll_list_no_separator(self):
         list = ["test", "test1", "test2"]