Skip to content

Commit 0c6eee3

Browse files
committed
Fix test
1 parent 9d9082b commit 0c6eee3

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

unittests/test_finding_model.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from datetime import datetime, timedelta
22

33
from crum import impersonate
4+
from django.test import override_settings
45
from django.urls import reverse
56
from rest_framework.authtoken.models import Token
67
from rest_framework.test import APIClient
@@ -538,6 +539,7 @@ def test_sla_expiration_date_after_sla_not_enforced(self):
538539
self.assertEqual(finding.sla_days_remaining(), None)
539540
self.assertEqual(finding.sla_deadline(), None)
540541

542+
@override_settings(SLA_CONFIG_ON_NON_PRODUCT_LEVELS=True)
541543
def test_sla_expiration_date_after_eng_override_added(self):
542544
"""
543545
Tests if the SLA expiration date and SLA days remaining are calculated correctly
@@ -577,6 +579,7 @@ def test_sla_expiration_date_after_eng_override_added(self):
577579
self.assertEqual(finding.sla_expiration_date, datetime.now().date() + timedelta(days=expected_sla_days))
578580
self.assertEqual(finding.sla_days_remaining(), expected_sla_days)
579581

582+
@override_settings(SLA_CONFIG_ON_NON_PRODUCT_LEVELS=True)
580583
def test_sla_expiration_date_after_test_override_added(self):
581584
"""
582585
Tests if the SLA expiration date and SLA days remaining are calculated correctly

0 commit comments

Comments
 (0)