@@ -375,46 +375,45 @@ def test_update_risk_acceptance_add_cross_engagement_fails(self):
375375 def test_risk_acceptance_created_filter (self ):
376376 # 1. Create a baseline Risk Acceptance using the existing test setup
377377 risk_acceptance = self .create_risk_acceptance ()
378-
378+
379379 # 2. Manually backdate the created date to test ranges
380- past_date = datetime .datetime .now (datetime .timezone . utc ) - datetime .timedelta (days = 10 )
380+ past_date = datetime .datetime .now (datetime .UTC ) - datetime .timedelta (days = 10 )
381381 risk_acceptance .created = past_date
382382 risk_acceptance .save ()
383383
384384 # 3. Test `created_before` (Less than / Before)
385385 # Should return the risk acceptance because it was created 10 days ago
386- future_date = datetime .datetime .now (datetime .timezone . utc ).strftime (' %Y-%m-%dT%H:%M:%S.%fZ' )
387- response = self .client .get (reverse (' risk_acceptance-list' ) + f' ?created_before={ future_date } ' )
386+ future_date = datetime .datetime .now (datetime .UTC ).strftime (" %Y-%m-%dT%H:%M:%S.%fZ" )
387+ response = self .client .get (reverse (" risk_acceptance-list" ) + f" ?created_before={ future_date } " )
388388 self .assertEqual (response .status_code , 200 )
389389 result_ids = {item ["id" ] for item in response .json ()["results" ]}
390390 self .assertIn (risk_acceptance .id , result_ids )
391391
392392 # 4. Test `created_after` (Greater than / After)
393393 # Should NOT return the risk acceptance because it is not newer than today
394- response = self .client .get (reverse (' risk_acceptance-list' ) + f' ?created_after={ future_date } ' )
394+ response = self .client .get (reverse (" risk_acceptance-list" ) + f" ?created_after={ future_date } " )
395395 self .assertEqual (response .status_code , 200 )
396396 result_ids = {item ["id" ] for item in response .json ()["results" ]}
397397 self .assertNotIn (risk_acceptance .id , result_ids )
398398
399-
400399 def test_risk_acceptance_updated_filter (self ):
401400 risk_acceptance = self .create_risk_acceptance ()
402-
401+
403402 # Manually backdate the updated date
404- past_date = datetime .datetime .now (datetime .timezone . utc ) - datetime .timedelta (days = 10 )
403+ past_date = datetime .datetime .now (datetime .UTC ) - datetime .timedelta (days = 10 )
405404 # We use .update() to bypass the auto_now=True behavior on the updated field
406405 type (risk_acceptance ).objects .filter (pk = risk_acceptance .id ).update (updated = past_date )
407406
408- future_date = datetime .datetime .now (datetime .timezone . utc ).strftime (' %Y-%m-%dT%H:%M:%S.%fZ' )
409-
407+ future_date = datetime .datetime .now (datetime .UTC ).strftime (" %Y-%m-%dT%H:%M:%S.%fZ" )
408+
410409 # Test updated_before
411- response = self .client .get (reverse (' risk_acceptance-list' ) + f' ?updated_before={ future_date } ' )
410+ response = self .client .get (reverse (" risk_acceptance-list" ) + f" ?updated_before={ future_date } " )
412411 self .assertEqual (response .status_code , 200 )
413412 result_ids = {item ["id" ] for item in response .json ()["results" ]}
414413 self .assertIn (risk_acceptance .id , result_ids )
415414
416415 # Test updated_after
417- response = self .client .get (reverse (' risk_acceptance-list' ) + f' ?updated_after={ future_date } ' )
416+ response = self .client .get (reverse (" risk_acceptance-list" ) + f" ?updated_after={ future_date } " )
418417 self .assertEqual (response .status_code , 200 )
419418 result_ids = {item ["id" ] for item in response .json ()["results" ]}
420419 self .assertNotIn (risk_acceptance .id , result_ids )
0 commit comments