@@ -284,6 +284,8 @@ def test_observation_before_fields_are_reset(
284284 self , mock_obs_manager , mock_rule_engine_cls , mock_copy , mock_normalize
285285 ):
286286 obs_before = MagicMock (spec = Observation )
287+ obs_before .rule_priority = 3
288+ obs_before .rule_rego_priority = 3
287289 mock_copy .return_value = obs_before
288290
289291 self ._setup_queryset_mock (mock_obs_manager , [self .mock_observation ])
@@ -295,8 +297,9 @@ def test_observation_before_fields_are_reset(
295297 self .assertEqual (obs_before .rule_status , "" )
296298 self .assertEqual (obs_before .rule_rego_status , "" )
297299 self .assertEqual (obs_before .rule_severity , "" )
298- self .assertIsNone (obs_before .rule_priority )
299- self .assertIsNone (obs_before .rule_rego_priority )
300+ # priorities are intentionally kept, see Rule_Engine
301+ self .assertEqual (obs_before .rule_priority , 3 )
302+ self .assertEqual (obs_before .rule_rego_priority , 3 )
300303 self .assertEqual (obs_before .rule_vex_justification , "" )
301304 self .assertEqual (obs_before .rule_rego_vex_justification , "" )
302305 self .assertIsNone (obs_before .general_rule )
0 commit comments