Skip to content

Commit 9b3205d

Browse files
committed
test fixes
1 parent 1e20a06 commit 9b3205d

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

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

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from io import StringIO
2-
from unittest.mock import MagicMock, patch
2+
from unittest.mock import MagicMock, PropertyMock, patch
33

44
import pytest
55
from django.core.management import call_command
@@ -76,7 +76,7 @@ def test_handle_successful_update(self):
7676
setattr(mock_requirements, field, requirement_weight)
7777
mock_metric.project.level = "test_level"
7878
mock_metric.project.name = TEST_PROJECT_NAME
79-
mock_metric.project.is_level_compliant = MagicMock(return_value=True)
79+
type(mock_metric.project).is_level_compliant = PropertyMock(return_value=True)
8080
mock_metric.is_funding_requirements_compliant = True
8181
mock_metric.is_leader_requirements_compliant = True
8282
self.mock_metrics.return_value.select_related.return_value = [mock_metric]
@@ -149,7 +149,7 @@ def test_handle_with_compliance_penalty(self):
149149
mock_metric.project.level = LAB_LEVEL
150150
mock_metric.project.project_level_official = FLAGSHIP_LEVEL
151151
mock_metric.project.name = NON_COMPLIANT_PROJECT_NAME
152-
mock_metric.project.is_level_compliant = MagicMock(return_value=False)
152+
type(mock_metric.project).is_level_compliant = PropertyMock(return_value=False)
153153

154154
# Set compliance requirements
155155
mock_metric.is_funding_requirements_compliant = True
@@ -231,7 +231,7 @@ def test_handle_without_compliance_penalty(self):
231231
mock_metric.project.level = FLAGSHIP_LEVEL
232232
mock_metric.project.project_level_official = FLAGSHIP_LEVEL
233233
mock_metric.project.name = COMPLIANT_PROJECT_NAME
234-
mock_metric.project.is_level_compliant = MagicMock(return_value=True)
234+
type(mock_metric.project).is_level_compliant = PropertyMock(return_value=True)
235235

236236
mock_metric.is_funding_requirements_compliant = True
237237
mock_metric.is_leader_requirements_compliant = True
@@ -301,7 +301,7 @@ def test_handle_zero_penalty_weight(self):
301301
mock_metric.project.level = LAB_LEVEL
302302
mock_metric.project.project_level_official = FLAGSHIP_LEVEL
303303
mock_metric.project.name = TEST_PROJECT_NAME
304-
mock_metric.project.is_level_compliant = MagicMock(return_value=False)
304+
type(mock_metric.project).is_level_compliant = PropertyMock(return_value=False)
305305

306306
mock_metric.is_funding_requirements_compliant = True
307307
mock_metric.is_leader_requirements_compliant = True
@@ -370,7 +370,7 @@ def test_handle_maximum_penalty_weight(self):
370370
mock_metric.project.level = LAB_LEVEL
371371
mock_metric.project.project_level_official = FLAGSHIP_LEVEL
372372
mock_metric.project.name = TEST_PROJECT_NAME
373-
mock_metric.project.is_level_compliant = MagicMock(return_value=False)
373+
type(mock_metric.project).is_level_compliant = PropertyMock(return_value=False)
374374

375375
mock_metric.is_funding_requirements_compliant = True
376376
mock_metric.is_leader_requirements_compliant = True
@@ -440,7 +440,7 @@ def test_handle_penalty_weight_clamping(self):
440440
mock_metric.project.level = LAB_LEVEL
441441
mock_metric.project.project_level_official = FLAGSHIP_LEVEL
442442
mock_metric.project.name = TEST_PROJECT_NAME
443-
mock_metric.project.is_level_compliant = MagicMock(return_value=False)
443+
type(mock_metric.project).is_level_compliant = PropertyMock(return_value=False)
444444
mock_metric.is_funding_requirements_compliant = True
445445
mock_metric.is_leader_requirements_compliant = True
446446

0 commit comments

Comments
 (0)