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

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

@@ -93,7 +93,7 @@ class TestWizClientIdHandling:
     @pytest.fixture
     def mock_wiz_auth(self):
         """Mock wiz_authenticate."""
-        with patch("regscale.integrations.commercial.wizv2.wiz_auth.wiz_authenticate") as mock:
+        with patch("regscale.integrations.commercial.wizv2.core.auth.wiz_authenticate") as mock:
             yield mock
 
     @pytest.fixture

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

@@ -481,6 +481,7 @@ class TestWizComplianceReportItem(unittest.TestCase):
 
 
 @pytest.mark.no_parallel
+@patch("regscale.integrations.compliance_integration.ComplianceIntegration.__init__", return_value=None)
 class TestWizComplianceReportProcessor(unittest.TestCase):
     """Test suite for WizComplianceReportProcessor class."""
 
@@ -527,9 +528,21 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
             },
         ]
 
+    def _initialize_processor_attributes(self, processor):
+        """Initialize parent class attributes that would normally be set by __init__."""
+        from collections import defaultdict
+
+        processor.all_compliance_items = []
+        processor.failed_compliance_items = []
+        processor.passing_controls = {}
+        processor.failing_controls = {}
+        processor.asset_compliance_map = defaultdict(list)
+        processor.plan_id = self.plan_id
+        processor.title = "Wiz Compliance"
+
     @patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager")
     @patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate")
-    def test_init_successful_authentication(self, mock_auth, mock_report_manager):
+    def test_init_successful_authentication(self, mock_auth, mock_report_manager, mock_parent_init):
         """Test successful initialization with authentication."""
         mock_auth.return_value = "test-token"
         mock_report_manager_instance = MagicMock()
@@ -541,6 +554,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
             client_id=self.client_id,
             client_secret=self.client_secret,
         )
+        self._initialize_processor_attributes(processor)
 
         mock_auth.assert_called_once_with(self.client_id, self.client_secret)
         mock_report_manager.assert_called_once()
@@ -551,7 +565,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
     @patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager")
     @patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate")
     @patch("regscale.integrations.commercial.wizv2.compliance_report.error_and_exit")
-    def test_init_failed_authentication(self, mock_error_exit, mock_auth, mock_report_manager):
+    def test_init_failed_authentication(self, mock_error_exit, mock_auth, mock_report_manager, mock_parent_init):
         """Test initialization with failed authentication."""
         mock_auth.return_value = None
         mock_error_exit.side_effect = SystemExit(1)
@@ -566,7 +580,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
 
         mock_error_exit.assert_called_once_with("Failed to authenticate with Wiz")
 
-    def test_parse_csv_report_regular_file(self):
+    def test_parse_csv_report_regular_file(self, mock_parent_init):
         """Test parse_csv_report with regular CSV file."""
         # Create temporary CSV file
         with tempfile.NamedTemporaryFile(mode="w", suffix=".csv", delete=False) as temp_file:
@@ -599,7 +613,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
         finally:
             os.unlink(temp_file_path)
 
-    def test_parse_csv_report_gzipped_file(self):
+    def test_parse_csv_report_gzipped_file(self, mock_parent_init):
         """Test parse_csv_report with gzipped CSV file."""
         # Create temporary gzipped CSV file
         with tempfile.NamedTemporaryFile(suffix=".csv.gz", delete=False) as temp_file:
@@ -631,7 +645,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
         finally:
             os.unlink(temp_file_path)
 
-    def test_parse_csv_report_file_not_found(self):
+    def test_parse_csv_report_file_not_found(self, mock_parent_init):
         """Test parse_csv_report with non-existent file."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             with patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager"):
@@ -647,7 +661,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
 
     @patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager")
     @patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate")
-    def test_fetch_compliance_data_with_existing_report(self, mock_auth, mock_report_manager):
+    def test_fetch_compliance_data_with_existing_report(self, mock_auth, mock_report_manager, mock_parent_init):
         """Test fetch_compliance_data with existing report file."""
         mock_auth.return_value = "test-token"
 
@@ -678,7 +692,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
 
     @patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager")
     @patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate")
-    def test_fetch_compliance_data_report_creation(self, mock_auth, mock_report_manager):
+    def test_fetch_compliance_data_report_creation(self, mock_auth, mock_report_manager, mock_parent_init):
         """Test fetch_compliance_data creates new report when none exists."""
         mock_auth.return_value = "test-token"
 
@@ -706,7 +720,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
         finally:
             os.unlink(temp_file_path)
 
-    def test_fetch_compliance_data_file_read_error(self):
+    def test_fetch_compliance_data_file_read_error(self, mock_parent_init):
         """Test fetch_compliance_data handles file read errors."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             with patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager"):
@@ -723,7 +737,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
                 raw_data = processor.fetch_compliance_data()
                 self.assertEqual(len(raw_data), 0)
 
-    def test_create_compliance_item(self):
+    def test_create_compliance_item(self, mock_parent_init):
         """Test create_compliance_item creates WizComplianceReportItem."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             with patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager"):
@@ -740,7 +754,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
 
     @patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager")
     @patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate")
-    def test_process_compliance_data_bypass_filtering(self, mock_auth, mock_report_manager):
+    def test_process_compliance_data_bypass_filtering(self, mock_auth, mock_report_manager, mock_parent_init):
         """Test process_compliance_data with bypass_control_filtering=True."""
         mock_auth.return_value = "test-token"
 
@@ -758,7 +772,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
 
     @patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager")
     @patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate")
-    def test_process_compliance_data_normal_filtering(self, mock_auth, mock_report_manager):
+    def test_process_compliance_data_normal_filtering(self, mock_auth, mock_report_manager, mock_parent_init):
         """Test process_compliance_data with normal filtering."""
         mock_auth.return_value = "test-token"
 
@@ -775,7 +789,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
             processor.process_compliance_data()
             mock_parent.assert_called_once()
 
-    def test_process_compliance_data_without_filtering_categorization(self):
+    def test_process_compliance_data_without_filtering_categorization(self, mock_parent_init):
         """Test _process_compliance_data_without_filtering categorizes controls correctly."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             with patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager"):
@@ -785,6 +799,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
                     client_id=self.client_id,
                     client_secret=self.client_secret,
                 )
+        self._initialize_processor_attributes(processor)
 
         # Mock fetch_compliance_data to return our test data
         with patch.object(processor, "fetch_compliance_data", return_value=self.sample_csv_data):
@@ -798,7 +813,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
                 # Should call fail-first categorization
                 mock_categorize.assert_called_once()
 
-    def test_process_compliance_sync(self):
+    def test_process_compliance_sync(self, mock_parent_init):
         """Test process_compliance_sync calls sync_compliance."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             with patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager"):
@@ -813,7 +828,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
             processor.process_compliance_sync()
             mock_sync.assert_called_once()
 
-    def test_get_or_create_report_existing_recent_report(self):
+    def test_get_or_create_report_existing_recent_report(self, mock_parent_init):
         """Test _get_or_create_report uses existing recent report."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             with patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager"):
@@ -828,7 +843,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
             result = processor._get_or_create_report()
             self.assertEqual(result, "existing_report.csv")
 
-    def test_get_or_create_report_creates_new_report(self):
+    def test_get_or_create_report_creates_new_report(self, mock_parent_init):
         """Test _get_or_create_report creates new report when none exists."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             with patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager"):
@@ -844,7 +859,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
                 result = processor._get_or_create_report()
                 self.assertEqual(result, "new_report.csv")
 
-    def test_find_recent_report_no_artifacts_dir(self):
+    def test_find_recent_report_no_artifacts_dir(self, mock_parent_init):
         """Test _find_recent_report returns None when artifacts directory doesn't exist."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             with patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager"):
@@ -859,7 +874,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
             result = processor._find_recent_report()
             self.assertIsNone(result)
 
-    def test_find_recent_report_finds_recent_file(self):
+    def test_find_recent_report_finds_recent_file(self, mock_parent_init):
         """Test _find_recent_report finds and returns recent file."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             with patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager"):
@@ -880,7 +895,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
                     expected_path = f"artifacts/wiz/{expected_filename}"
                     self.assertEqual(result, expected_path)
 
-    def test_find_recent_report_ignores_old_files(self):
+    def test_find_recent_report_ignores_old_files(self, mock_parent_init):
         """Test _find_recent_report ignores files older than max_age_hours."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             with patch("regscale.integrations.commercial.wizv2.compliance_report.WizReportManager"):
@@ -903,7 +918,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
 
     @patch("regscale.integrations.commercial.wizv2.compliance_report.ReportFileCleanup")
     @patch("os.makedirs")
-    def test_create_and_download_report_success(self, mock_makedirs, mock_cleanup):
+    def test_create_and_download_report_success(self, mock_makedirs, mock_cleanup, mock_parent_init):
         """Test _create_and_download_report successful report creation."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             mock_report_manager = MagicMock()
@@ -931,7 +946,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
         mock_makedirs.assert_called_once()
         mock_cleanup.cleanup_old_files.assert_called_once()
 
-    def test_create_and_download_report_creation_failure(self):
+    def test_create_and_download_report_creation_failure(self, mock_parent_init):
         """Test _create_and_download_report handles report creation failure."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             mock_report_manager = MagicMock()
@@ -952,7 +967,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
         result = processor._create_and_download_report()
         self.assertIsNone(result)
 
-    def test_create_and_download_report_download_failure(self):
+    def test_create_and_download_report_download_failure(self, mock_parent_init):
         """Test _create_and_download_report handles download failure."""
         with patch("regscale.integrations.commercial.wizv2.compliance_report.wiz_authenticate", return_value="token"):
             mock_report_manager = MagicMock()
@@ -978,6 +993,7 @@ class TestWizComplianceReportProcessor(unittest.TestCase):
 
 
 @pytest.mark.no_parallel
+@patch("regscale.integrations.compliance_integration.ComplianceIntegration.__init__", return_value=None)
 class TestWizComplianceStatusMatching(unittest.TestCase):
     """Test suite for status matching and case-insensitive logic."""
 
@@ -1000,7 +1016,17 @@ class TestWizComplianceStatusMatching(unittest.TestCase):
             "Remediation Steps": "Fix this",
         }
 
-    def test_pass_status_matching_case_insensitive(self):
+    def _initialize_processor_attributes(self, processor):
+        """Initialize parent class attributes that would normally be set by __init__."""
+        from collections import defaultdict
+
+        processor.all_compliance_items = []
+        processor.failed_compliance_items = []
+        processor.passing_controls = {}
+        processor.failing_controls = {}
+        processor.asset_compliance_map = defaultdict(list)
+
+    def test_pass_status_matching_case_insensitive(self, mock_parent_init):
         """Test that pass statuses are matched case-insensitively."""
         # Current implementation supports: Pass, PASS, pass but not Passed, PASSED, passed
         pass_values = ["Pass", "PASS", "pass"]
@@ -1025,7 +1051,7 @@ class TestWizComplianceStatusMatching(unittest.TestCase):
                 self.assertEqual(item.get_status(), "Other Than Satisfied")
                 self.assertEqual(item.get_implementation_status(), "In Remediation")
 
-    def test_fail_status_matching_case_insensitive(self):
+    def test_fail_status_matching_case_insensitive(self, mock_parent_init):
         """Test that fail statuses are matched case-insensitively."""
         fail_values = ["Fail", "FAIL", "fail", "Failed", "FAILED", "failed"]
 
@@ -1037,7 +1063,7 @@ class TestWizComplianceStatusMatching(unittest.TestCase):
                 self.assertEqual(item.get_status(), "Other Than Satisfied")
                 self.assertEqual(item.get_implementation_status(), "In Remediation")
 
-    def test_status_categorization_in_processor(self):
+    def test_status_categorization_in_processor(self, mock_parent_init):
         """Test that WizComplianceReportProcessor correctly categorizes statuses."""
         # Test data with mixed case statuses
         test_data = [
@@ -1058,6 +1084,7 @@ class TestWizComplianceStatusMatching(unittest.TestCase):
                     client_secret="secret",
                     bypass_control_filtering=False,  # Use threshold-based logic
                 )
+        self._initialize_processor_attributes(processor)
 
         # Mock fetch_compliance_data to return test data
         with patch.object(processor, "fetch_compliance_data", return_value=test_data):
@@ -1074,10 +1101,21 @@ class TestWizComplianceStatusMatching(unittest.TestCase):
 
 
 @pytest.mark.no_parallel
+@patch("regscale.integrations.compliance_integration.ComplianceIntegration.__init__", return_value=None)
 class TestWizComplianceControlCategorization(unittest.TestCase):
     """Test suite for control categorization and aggregation logic."""
 
-    def test_control_categorization_all_pass(self):
+    def _initialize_processor_attributes(self, processor):
+        """Initialize parent class attributes that would normally be set by __init__."""
+        from collections import defaultdict
+
+        processor.all_compliance_items = []
+        processor.failed_compliance_items = []
+        processor.passing_controls = {}
+        processor.failing_controls = {}
+        processor.asset_compliance_map = defaultdict(list)
+
+    def test_control_categorization_all_pass(self, mock_parent_init):
         """Test control categorization when all items pass."""
         test_data = [
             {
@@ -1109,6 +1147,7 @@ class TestWizComplianceControlCategorization(unittest.TestCase):
                     client_secret="secret",
                     bypass_control_filtering=False,  # Use threshold-based logic
                 )
+        self._initialize_processor_attributes(processor)
 
         # Fill in required fields for test data
         full_test_data = []
@@ -1139,7 +1178,7 @@ class TestWizComplianceControlCategorization(unittest.TestCase):
         self.assertEqual(len(processor.failing_controls), 0)
         self.assertIn("ac-2(1)", processor.passing_controls)
 
-    def test_control_categorization_all_fail(self):
+    def test_control_categorization_all_fail(self, mock_parent_init):
         """Test control categorization when all items fail."""
         test_data = [
             {
@@ -1171,6 +1210,7 @@ class TestWizComplianceControlCategorization(unittest.TestCase):
                     client_secret="secret",
                     bypass_control_filtering=False,  # Use threshold-based logic
                 )
+        self._initialize_processor_attributes(processor)
 
         # Fill in required fields for test data
         full_test_data = []
@@ -1201,7 +1241,7 @@ class TestWizComplianceControlCategorization(unittest.TestCase):
         self.assertEqual(len(processor.failing_controls), 1)
         self.assertIn("ac-3", processor.failing_controls)
 
-    def test_control_categorization_mixed_results_high_failure_rate(self):
+    def test_control_categorization_mixed_results_high_failure_rate(self, mock_parent_init):
         """Test control categorization with mixed results above failure threshold."""
         # 5 items: 2 pass, 3 fail = 60% failure rate (above default 20% threshold)
         test_data = [
@@ -1221,6 +1261,7 @@ class TestWizComplianceControlCategorization(unittest.TestCase):
                     client_secret="secret",
                     bypass_control_filtering=False,  # Use threshold-based logic
                 )
+        self._initialize_processor_attributes(processor)
 
         # Fill in required fields - all same control
         full_test_data = []
@@ -1251,7 +1292,7 @@ class TestWizComplianceControlCategorization(unittest.TestCase):
         self.assertEqual(len(processor.failing_controls), 1)
         self.assertIn("ac-4", processor.failing_controls)
 
-    def test_control_categorization_mixed_results_fail_first(self):
+    def test_control_categorization_mixed_results_fail_first(self, mock_parent_init):
         """Test control categorization with mixed results using fail-first logic."""
         # 10 items: 9 pass, 1 fail - with fail-first logic, any failure makes the control fail
         test_data = []
@@ -1268,6 +1309,7 @@ class TestWizComplianceControlCategorization(unittest.TestCase):
                     client_secret="secret",
                     bypass_control_filtering=False,
                 )
+        self._initialize_processor_attributes(processor)
 
         # Fill in required fields - all same control
         full_test_data = []
@@ -1298,7 +1340,7 @@ class TestWizComplianceControlCategorization(unittest.TestCase):
         self.assertEqual(len(processor.failing_controls), 1)
         self.assertIn("si-2", processor.failing_controls)
 
-    def test_control_categorization_fail_first_logic(self):
+    def test_control_categorization_fail_first_logic(self, mock_parent_init):
         """Test that Wiz compliance uses fail-first logic regardless of bypass_control_filtering setting."""
         # 10 items: 7 pass, 3 fail - with fail-first logic, this should be a failing control
         test_data = []
@@ -1316,6 +1358,7 @@ class TestWizComplianceControlCategorization(unittest.TestCase):
                     client_secret="secret",
                     bypass_control_filtering=False,
                 )
+        self._initialize_processor_attributes(processor)
 
         # Fill in required fields - all same control
         full_test_data = []