Skip to content

Commit 60c7c69

Browse files
committed
style: fix ruff issues in risk acceptance tests
1 parent 5ce06e8 commit 60c7c69

1 file changed

Lines changed: 11 additions & 12 deletions

File tree

unittests/test_risk_acceptance_api.py

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)