Skip to content

Commit 2748680

Browse files
committed
sonar fixes
1 parent 021ec86 commit 2748680

File tree

2 files changed

+14
-8
lines changed

2 files changed

+14
-8
lines changed

backend/tests/apps/owasp/management/commands/owasp_update_project_health_metrics_test.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -226,15 +226,21 @@ def test_update_official_levels_level_mapping(self):
226226
]
227227

228228
for official_level, expected_mapped in test_cases:
229-
with patch(PROJECT_FILTER_PATCH) as mock_filter:
229+
with patch(PROJECT_FILTER_PATCH) as mock_filter, \
230+
patch(PROJECT_BULK_SAVE_PATCH) as mock_bulk_save:
230231

231232
mock_filter.return_value = [project]
232233
project.project_level_official = "other" # Reset
233234

234235
official_levels = {TEST_PROJECT_NAME: official_level}
235-
self.command.update_official_levels(official_levels)
236+
updated_count = self.command.update_official_levels(official_levels)
236237

237238
assert project.project_level_official == expected_mapped
239+
if expected_mapped != "other": # Only count as update if level changed
240+
assert updated_count == 1
241+
mock_bulk_save.assert_called_once()
242+
else:
243+
assert updated_count == 0
238244

239245
@patch('requests.get')
240246
def test_handle_with_official_levels_integration(self, mock_get):

backend/tests/apps/owasp/management/commands/project_level_compliance_integration_test.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -225,9 +225,9 @@ def test_api_failure_handling(self, mock_get):
225225

226226
project = self.create_mock_project("Test Project", "lab", "lab")
227227

228-
with patch("apps.owasp.models.project.Project.objects.filter") as mock_projects, \
229-
patch("apps.owasp.models.project_health_metrics.ProjectHealthMetrics.bulk_save"), \
230-
patch("sys.stdout", new=self.stdout):
228+
with patch(PROJECT_FILTER_PATCH) as mock_projects, \
229+
patch(METRICS_BULK_SAVE_PATCH), \
230+
patch(STDOUT_PATCH, new=self.stdout):
231231

232232
mock_projects.return_value = [project]
233233

@@ -245,9 +245,9 @@ def test_skip_official_levels_flag(self):
245245
"""Test that --skip-official-levels flag works correctly."""
246246
project = self.create_mock_project("Test Project", "lab", "flagship")
247247

248-
with patch("apps.owasp.models.project.Project.objects.filter") as mock_projects, \
249-
patch("apps.owasp.models.project_health_metrics.ProjectHealthMetrics.bulk_save"), \
250-
patch("sys.stdout", new=self.stdout):
248+
with patch(PROJECT_FILTER_PATCH) as mock_projects, \
249+
patch(METRICS_BULK_SAVE_PATCH), \
250+
patch(STDOUT_PATCH, new=self.stdout):
251251

252252
mock_projects.return_value = [project]
253253

0 commit comments

Comments
 (0)