regscale-cli 6.26.0.0__py3-none-any.whl → 6.27.0.1__py3-none-any.whl

tests/regscale/integrations/commercial/wizv2/test_wizv2_utils.py

@@ -14,11 +14,12 @@ logger = logging.getLogger("regscale")
 class TestWizUtils(unittest.TestCase):
     """Test cases for Wiz utility functions."""
 
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    @patch("regscale.integrations.commercial.wizv2.utils.MAX_RETRIES", 3)
-    def test_get_report_url_and_status_completed(self, mock_rerun_report, mock_download_report):
+    @patch("regscale.integrations.commercial.wizv2.utils.main.time.sleep")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.download_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.rerun_expired_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
+    @patch("regscale.integrations.commercial.wizv2.utils.main.MAX_RETRIES", 3)
+    def test_get_report_url_and_status_completed(self, mock_rerun_report, mock_download_report, mock_sleep):
         """Test get_report_url_and_status with COMPLETED status."""
         # Mock response for completed report
         mock_response = MagicMock()
@@ -35,13 +36,17 @@ class TestWizUtils(unittest.TestCase):
         self.assertEqual(result, "https://example.com/report.csv")
         mock_download_report.assert_called_once_with({"reportId": "test-report-id"})
         mock_rerun_report.assert_not_called()
-
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.get_report_url_and_status")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    @patch("regscale.integrations.commercial.wizv2.utils.MAX_RETRIES", 3)
-    def test_get_report_url_and_status_expired(self, mock_recursive_call, mock_rerun_report, mock_download_report):
+        mock_sleep.assert_not_called()  # Should not sleep on first success
+
+    @patch("regscale.integrations.commercial.wizv2.utils.main.time.sleep")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.download_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.rerun_expired_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.get_report_url_and_status")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
+    @patch("regscale.integrations.commercial.wizv2.utils.main.MAX_RETRIES", 3)
+    def test_get_report_url_and_status_expired(
+        self, mock_recursive_call, mock_rerun_report, mock_download_report, mock_sleep
+    ):
         """Test get_report_url_and_status with EXPIRED status."""
         # Mock response for expired report
         mock_response = MagicMock()
@@ -65,14 +70,16 @@ class TestWizUtils(unittest.TestCase):
         mock_download_report.assert_called_once_with({"reportId": "test-report-id"})
         mock_rerun_report.assert_called_once_with({"reportId": "test-report-id"})
         mock_recursive_call.assert_called_once_with("test-report-id")
-
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    @patch("regscale.integrations.commercial.wizv2.utils.MAX_RETRIES", 5)
-    def test_get_report_url_and_status_rate_limit(self, mock_rerun_report, mock_download_report):
+        mock_sleep.assert_not_called()  # Should not sleep on first call before recursion
+
+    @patch("regscale.integrations.commercial.wizv2.utils.main.time.sleep")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.download_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.rerun_expired_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
+    @patch("regscale.integrations.commercial.wizv2.utils.main.MAX_RETRIES", 5)
+    def test_get_report_url_and_status_rate_limit(self, mock_rerun_report, mock_download_report, mock_sleep):
         """Test get_report_url_and_status with rate limit error."""
-        from regscale.integrations.commercial.wizv2.constants import RATE_LIMIT_MSG
+        from regscale.integrations.commercial.wizv2.core.constants import RATE_LIMIT_MSG
 
         # Mock response with rate limit error
         mock_response = MagicMock()
@@ -94,21 +101,22 @@ class TestWizUtils(unittest.TestCase):
         # Configure mock to return different responses on subsequent calls
         mock_download_report.side_effect = [mock_response, mock_response2]
 
-        with patch("time.sleep") as mock_sleep:
-            # Call the function
-            result = get_report_url_and_status("test-report-id")
-
-            # Verify the result
-            self.assertEqual(result, "https://example.com/report.csv")
-            self.assertEqual(mock_download_report.call_count, 2)
-            mock_rerun_report.assert_not_called()
-            self.assertEqual(mock_sleep.call_count, 5)
-
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    @patch("regscale.integrations.commercial.wizv2.utils.MAX_RETRIES", 3)
-    def test_get_report_url_and_status_failed_response(self, mock_rerun_report, mock_download_report):
+        # Call the function
+        result = get_report_url_and_status("test-report-id")
+
+        # Verify the result
+        self.assertEqual(result, "https://example.com/report.csv")
+        self.assertEqual(mock_download_report.call_count, 2)
+        mock_rerun_report.assert_not_called()
+        # Sleep is called twice: once for rate limit, once for retry interval
+        self.assertEqual(mock_sleep.call_count, 2)
+
+    @patch("regscale.integrations.commercial.wizv2.utils.main.time.sleep")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.download_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.rerun_expired_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
+    @patch("regscale.integrations.commercial.wizv2.utils.main.MAX_RETRIES", 3)
+    def test_get_report_url_and_status_failed_response(self, mock_rerun_report, mock_download_report, mock_sleep):
         """Test get_report_url_and_status with failed response."""
         # Mock failed response
         mock_response = MagicMock()
@@ -123,11 +131,12 @@ class TestWizUtils(unittest.TestCase):
         mock_download_report.assert_called_once_with({"reportId": "test-report-id"})
         mock_rerun_report.assert_not_called()
 
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    @patch("regscale.integrations.commercial.wizv2.utils.MAX_RETRIES", 3)
-    def test_get_report_url_and_status_none_response(self, mock_rerun_report, mock_download_report):
+    @patch("regscale.integrations.commercial.wizv2.utils.main.time.sleep")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.download_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.rerun_expired_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
+    @patch("regscale.integrations.commercial.wizv2.utils.main.MAX_RETRIES", 3)
+    def test_get_report_url_and_status_none_response(self, mock_rerun_report, mock_download_report, mock_sleep):
         """Test get_report_url_and_status with None response."""
         # Mock None response
         mock_download_report.return_value = None
@@ -140,11 +149,12 @@ class TestWizUtils(unittest.TestCase):
         mock_download_report.assert_called_once_with({"reportId": "test-report-id"})
         mock_rerun_report.assert_not_called()
 
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    @patch("regscale.integrations.commercial.wizv2.utils.MAX_RETRIES", 3)  # Reduce retries for faster testing
-    def test_get_report_url_and_status_other_error(self, mock_rerun_report, mock_download_report):
+    @patch("regscale.integrations.commercial.wizv2.utils.main.time.sleep")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.download_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.rerun_expired_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
+    @patch("regscale.integrations.commercial.wizv2.utils.main.MAX_RETRIES", 3)  # Reduce retries for faster testing
+    def test_get_report_url_and_status_other_error(self, mock_rerun_report, mock_download_report, mock_sleep):
         """Test get_report_url_and_status with other error in response."""
         # Mock response with other error
         mock_response = MagicMock()
@@ -161,11 +171,12 @@ class TestWizUtils(unittest.TestCase):
         self.assertEqual(mock_download_report.call_count, 3)
         mock_rerun_report.assert_not_called()
 
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    @patch("regscale.integrations.commercial.wizv2.utils.MAX_RETRIES", 3)  # Reduce retries for faster testing
-    def test_get_report_url_and_status_unknown_status(self, mock_rerun_report, mock_download_report):
+    @patch("regscale.integrations.commercial.wizv2.utils.main.time.sleep")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.download_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.rerun_expired_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
+    @patch("regscale.integrations.commercial.wizv2.utils.main.MAX_RETRIES", 3)  # Reduce retries for faster testing
+    def test_get_report_url_and_status_unknown_status(self, mock_rerun_report, mock_download_report, mock_sleep):
         """Test get_report_url_and_status with unknown status."""
         # Mock response with unknown status
         mock_response = MagicMock()
@@ -182,11 +193,12 @@ class TestWizUtils(unittest.TestCase):
         self.assertEqual(mock_download_report.call_count, 3)
         mock_rerun_report.assert_not_called()
 
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    @patch("regscale.integrations.commercial.wizv2.utils.MAX_RETRIES", 3)  # Reduce retries for faster testing
-    def test_get_report_url_and_status_missing_status(self, mock_rerun_report, mock_download_report):
+    @patch("regscale.integrations.commercial.wizv2.utils.main.time.sleep")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.download_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.rerun_expired_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
+    @patch("regscale.integrations.commercial.wizv2.utils.main.MAX_RETRIES", 3)  # Reduce retries for faster testing
+    def test_get_report_url_and_status_missing_status(self, mock_rerun_report, mock_download_report, mock_sleep):
         """Test get_report_url_and_status with missing status in response."""
         # Mock response with missing status
         mock_response = MagicMock()
@@ -203,12 +215,13 @@ class TestWizUtils(unittest.TestCase):
         self.assertEqual(mock_download_report.call_count, 3)
         mock_rerun_report.assert_not_called()
 
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    @patch("regscale.integrations.commercial.wizv2.utils.MAX_RETRIES", 3)
+    @patch("regscale.integrations.commercial.wizv2.utils.main.time.sleep")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.download_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.rerun_expired_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
+    @patch("regscale.integrations.commercial.wizv2.utils.main.MAX_RETRIES", 3)
     def test_get_report_url_and_status_multiple_attempts_before_completion(
-        self, mock_rerun_report, mock_download_report
+        self, mock_rerun_report, mock_download_report, mock_sleep
     ):
         """Test get_report_url_and_status with multiple attempts before completion."""
         # Mock responses: first two with unknown status, third with completed
@@ -229,132 +242,14 @@ class TestWizUtils(unittest.TestCase):
         # Configure mock to return different responses on subsequent calls
         mock_download_report.side_effect = [mock_response1, mock_response2, mock_response3]
 
-        with patch("time.sleep") as mock_sleep:
-            # Call the function
-            result = get_report_url_and_status("test-report-id")
-
-            # Verify the result
-            self.assertEqual(result, "https://example.com/report.csv")
-            self.assertEqual(mock_download_report.call_count, 3)
-            # Should sleep twice (after first and second attempts)
-            self.assertEqual(mock_sleep.call_count, 2)
-            mock_rerun_report.assert_not_called()
-
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.MAX_RETRIES", 3)
-    def test_get_report_url_and_status_rate_limit(self, mock_rerun_report, mock_download_report):
-        """Test get_report_url_and_status with rate limit error."""
-        from regscale.integrations.commercial.wizv2.constants import RATE_LIMIT_MSG
-        import time
-
-        # Mock response with rate limit error
-        mock_response = MagicMock()
-        mock_response.ok = True
-        mock_response.json.return_value = {
-            "errors": [{"message": "Rate limit exceeded", "extensions": {"retryAfter": 5}}]
-        }
-        mock_download_report.return_value = mock_response
-
-        # Mock second response for successful completion
-        mock_response2 = MagicMock()
-        mock_response2.ok = True
-        mock_response2.json.return_value = {
-            "data": {"report": {"lastRun": {"status": "COMPLETED", "url": "https://example.com/report.csv"}}}
-        }
-
-        # Configure mock to return different responses on subsequent calls
-        mock_download_report.side_effect = [mock_response, mock_response2]
-
-        with patch("time.sleep") as mock_sleep:
-            # Call the function
-            result = get_report_url_and_status("test-report-id")
-
-            # Verify the result
-            self.assertEqual(result, "https://example.com/report.csv")
-            self.assertEqual(mock_download_report.call_count, 2)
-            # Check that time.sleep was called with the rate limit retry value (5)
-            mock_sleep.assert_any_call(5)
-            mock_rerun_report.assert_not_called()
-
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    def test_get_report_url_and_status_other_error(self, mock_rerun_report, mock_download_report):
-        """Test get_report_url_and_status with other error in response."""
-        # Mock response with other error
-        mock_response = MagicMock()
-        mock_response.ok = True
-        mock_response.json.return_value = {"errors": [{"message": "Some other error occurred"}]}
-        mock_download_report.return_value = mock_response
-
-        # Call the function and expect exception after max retries
-        with self.assertRaises(Exception) as context:
-            get_report_url_and_status("test-report-id")
-
-        self.assertIn("Download failed, exceeding the maximum number of retries", str(context.exception))
-        # Should be called MAX_RETRIES times
-        self.assertEqual(mock_download_report.call_count, 100)
-        mock_rerun_report.assert_not_called()
-
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    def test_get_report_url_and_status_unknown_status(self, mock_rerun_report, mock_download_report):
-        """Test get_report_url_and_status with unknown status."""
-        # Mock response with unknown status
-        mock_response = MagicMock()
-        mock_response.ok = True
-        mock_response.json.return_value = {"data": {"report": {"lastRun": {"status": "UNKNOWN_STATUS"}}}}
-        mock_download_report.return_value = mock_response
-
-        # Call the function and expect exception after max retries
-        with self.assertRaises(Exception) as context:
-            get_report_url_and_status("test-report-id")
-
-        self.assertIn("Download failed, exceeding the maximum number of retries", str(context.exception))
-        # Should be called MAX_RETRIES times
-        self.assertEqual(mock_download_report.call_count, 100)
-        mock_rerun_report.assert_not_called()
-
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    def test_get_report_url_and_status_missing_status(self, mock_rerun_report, mock_download_report):
-        """Test get_report_url_and_status with missing status in response."""
-        # Mock response with missing status
-        mock_response = MagicMock()
-        mock_response.ok = True
-        mock_response.json.return_value = {"data": {"report": {"lastRun": {}}}}
-        mock_download_report.return_value = mock_response
-
-        # Call the function and expect exception after max retries
-        with self.assertRaises(Exception) as context:
-            get_report_url_and_status("test-report-id")
-
-        self.assertIn("Download failed, exceeding the maximum number of retries", str(context.exception))
-        # Should be called MAX_RETRIES times
-        self.assertEqual(mock_download_report.call_count, 100)
-        mock_rerun_report.assert_not_called()
-
-    @patch("regscale.integrations.commercial.wizv2.utils.download_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.rerun_expired_report")
-    @patch("regscale.integrations.commercial.wizv2.utils.CHECK_INTERVAL_FOR_DOWNLOAD_REPORT", 0.001)
-    def test_get_report_url_and_status_missing_data(self, mock_rerun_report, mock_download_report):
-        """Test get_report_url_and_status with missing data in response."""
-        # Mock response with missing data
-        mock_response = MagicMock()
-        mock_response.ok = True
-        mock_response.json.return_value = {}
-        mock_download_report.return_value = mock_response
-
-        # Call the function and expect exception after max retries
-        with self.assertRaises(Exception) as context:
-            get_report_url_and_status("test-report-id")
+        # Call the function
+        result = get_report_url_and_status("test-report-id")
 
-        self.assertIn("Download failed, exceeding the maximum number of retries", str(context.exception))
-        # Should be called MAX_RETRIES times
-        self.assertEqual(mock_download_report.call_count, 100)
+        # Verify the result
+        self.assertEqual(result, "https://example.com/report.csv")
+        self.assertEqual(mock_download_report.call_count, 3)
+        # Should sleep twice (after first and second attempts)
+        self.assertEqual(mock_sleep.call_count, 2)
         mock_rerun_report.assert_not_called()
 
 
@@ -371,8 +266,8 @@ class TestGetOrCreateReportId(unittest.TestCase):
         ]
         self.target_framework = "NIST_SP_800-53_Revision_5"
 
-    @patch("regscale.integrations.commercial.wizv2.utils.Application")
-    @patch("regscale.integrations.commercial.wizv2.utils.is_report_expired")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.Application")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.is_report_expired")
     def test_get_or_create_report_id_existing_valid_report(self, mock_is_expired, mock_application):
         """Test returning existing report ID when report is valid (not expired)."""
         # Mock Application
@@ -400,9 +295,9 @@ class TestGetOrCreateReportId(unittest.TestCase):
         mock_app.config.get.assert_called_once_with("wizReportAge", 15)
         mock_is_expired.assert_called_once_with("2023-07-15T14:37:55.450532Z", 15)
 
-    @patch("regscale.integrations.commercial.wizv2.utils.Application")
-    @patch("regscale.integrations.commercial.wizv2.utils.is_report_expired")
-    @patch("regscale.integrations.commercial.wizv2.utils.create_compliance_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.Application")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.is_report_expired")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.create_compliance_report")
     def test_get_or_create_report_id_existing_expired_report(
         self, mock_create_report, mock_is_expired, mock_application
     ):
@@ -440,8 +335,8 @@ class TestGetOrCreateReportId(unittest.TestCase):
             framework_id="framework-1",
         )
 
-    @patch("regscale.integrations.commercial.wizv2.utils.Application")
-    @patch("regscale.integrations.commercial.wizv2.utils.create_compliance_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.Application")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.create_compliance_report")
     def test_get_or_create_report_id_no_existing_report(self, mock_create_report, mock_application):
         """Test creating new report when no existing report is found."""
         # Mock Application
@@ -467,8 +362,8 @@ class TestGetOrCreateReportId(unittest.TestCase):
             framework_id="framework-1",
         )
 
-    @patch("regscale.integrations.commercial.wizv2.utils.Application")
-    @patch("regscale.integrations.commercial.wizv2.utils.is_report_expired")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.Application")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.is_report_expired")
     def test_get_or_create_report_id_missing_run_at(self, mock_is_expired, mock_application):
         """Test behavior when existing report has no runAt timestamp."""
         # Mock Application
@@ -497,7 +392,7 @@ class TestGetOrCreateReportId(unittest.TestCase):
         # is_report_expired should not be called when runAt is missing
         mock_is_expired.assert_not_called()
 
-    @patch("regscale.integrations.commercial.wizv2.utils.Application")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.Application")
     def test_get_or_create_report_id_framework_not_found(self, mock_application):
         """Test ValueError when target framework is not in frameworks list."""
         # Mock Application
@@ -518,8 +413,8 @@ class TestGetOrCreateReportId(unittest.TestCase):
         # The actual error message from list.index() is different
         self.assertIn("is not in list", str(context.exception))
 
-    @patch("regscale.integrations.commercial.wizv2.utils.Application")
-    @patch("regscale.integrations.commercial.wizv2.utils.is_report_expired")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.Application")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.is_report_expired")
     def test_get_or_create_report_id_custom_report_age(self, mock_is_expired, mock_application):
         """Test using custom wizReportAge configuration."""
         # Mock Application with custom age
@@ -547,9 +442,9 @@ class TestGetOrCreateReportId(unittest.TestCase):
         mock_app.config.get.assert_called_once_with("wizReportAge", 15)
         mock_is_expired.assert_called_once_with("2023-07-15T14:37:55.450532Z", 30)
 
-    @patch("regscale.integrations.commercial.wizv2.utils.Application")
-    @patch("regscale.integrations.commercial.wizv2.utils.is_report_expired")
-    @patch("regscale.integrations.commercial.wizv2.utils.create_compliance_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.Application")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.is_report_expired")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.create_compliance_report")
     def test_get_or_create_report_id_multiple_reports_first_match(
         self, mock_create_report, mock_is_expired, mock_application
     ):
@@ -585,9 +480,9 @@ class TestGetOrCreateReportId(unittest.TestCase):
         mock_is_expired.assert_called_once_with("2023-07-15T14:37:55.450532Z", 15)
         mock_create_report.assert_not_called()
 
-    @patch("regscale.integrations.commercial.wizv2.utils.Application")
-    @patch("regscale.integrations.commercial.wizv2.utils.is_report_expired")
-    @patch("regscale.integrations.commercial.wizv2.utils.create_compliance_report")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.Application")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.is_report_expired")
+    @patch("regscale.integrations.commercial.wizv2.utils.main.create_compliance_report")
     def test_get_or_create_report_id_different_report_names(
         self, mock_create_report, mock_is_expired, mock_application
     ):

tests/regscale/integrations/commercial/wizv2/utils/__init__.py

@@ -0,0 +1 @@
+"""Tests for Wiz V2 Utils"""