mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
0187e99377
* Support 202403 prepare * refactor peeringdb_server/management/commands/pdb_delete_outdated_pending_affil_request.py for improved code structure fix tests/test_settings.py::TestAutoApproveAffiliation::test_setting by using rdap mocking * db schema docs and api docs regen * rir should be RIR --------- Co-authored-by: 20C <code@20c.com>
110 lines
3.5 KiB
Python
110 lines
3.5 KiB
Python
import re
|
|
import sys
|
|
from datetime import timedelta
|
|
from io import StringIO
|
|
from sys import stdout
|
|
from unittest.mock import ANY, MagicMock, patch
|
|
|
|
from django.core.management import call_command
|
|
from django.test import TestCase
|
|
from django.utils import timezone
|
|
from reversion import create_revision
|
|
|
|
from peeringdb_server.models import Organization, User, UserOrgAffiliationRequest
|
|
|
|
|
|
class TestDeleteOldRequests(TestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
with create_revision():
|
|
cls.org = Organization.objects.create(name="Test Org")
|
|
|
|
# Add users to admin usergroup
|
|
for index in range(0, 3):
|
|
user = User.objects.create_user(
|
|
username=f"admin_user_{index}",
|
|
email=f"admin_user_{index}@localhost",
|
|
first_name=f"admin_user_{index}",
|
|
last_name=f"admin_user_{index}",
|
|
)
|
|
cls.org.admin_usergroup.user_set.add(user)
|
|
user.save()
|
|
cls.org.save()
|
|
|
|
# Add affiliation requests (all outdated)
|
|
for index in range(0, 3):
|
|
user = User.objects.create_user(
|
|
username=f"user_{index}",
|
|
email=f"user_{index}@localhost",
|
|
first_name=f"user_{index}",
|
|
last_name=f"user_{index}",
|
|
)
|
|
created_date = timezone.now() - timedelta(days=2) # Set all outdated
|
|
cls.org.affiliation_requests.create(
|
|
user=user,
|
|
status="pending",
|
|
created=created_date,
|
|
)
|
|
|
|
def test_delete_old_requests_dry_run(self):
|
|
|
|
# Capture output
|
|
output = StringIO()
|
|
|
|
# Call the command with dry-run flag commit=False
|
|
call_command(
|
|
"pdb_delete_outdated_pending_affil_request",
|
|
days_old=0,
|
|
commit=False,
|
|
stdout=output,
|
|
)
|
|
|
|
# Verify dry-run message
|
|
expected_message = f"Dry Run: Would have deleted {UserOrgAffiliationRequest.objects.filter(status='pending').count()} old pending requests."
|
|
actual_output = output.getvalue().strip()
|
|
|
|
assert expected_message in actual_output
|
|
|
|
def test_delete_old_requests_with_limit(self):
|
|
|
|
# Capture output
|
|
output = StringIO()
|
|
|
|
# Call the command with limit and commit (ensure commit=True)
|
|
call_command(
|
|
"pdb_delete_outdated_pending_affil_request",
|
|
days_old=0,
|
|
commit=True,
|
|
limit=2,
|
|
stdout=output,
|
|
)
|
|
|
|
actual_output = output.getvalue().strip()
|
|
|
|
# Verify deletion and success message (adjust for limit)
|
|
assert UserOrgAffiliationRequest.objects.count() == 1
|
|
|
|
assert "Successfully deleted 2 old pending requests." in actual_output
|
|
|
|
def test_delete_old_requests_no_limit(self):
|
|
|
|
# Capture output
|
|
output = StringIO()
|
|
|
|
count = UserOrgAffiliationRequest.objects.all().count()
|
|
|
|
# Call the command with no limit and commit (ensure commit=True)
|
|
call_command(
|
|
"pdb_delete_outdated_pending_affil_request",
|
|
days_old=0,
|
|
commit=True,
|
|
stdout=output,
|
|
)
|
|
|
|
# Verify deletion and success message (count from objects)
|
|
assert UserOrgAffiliationRequest.objects.count() == 0
|
|
assert (
|
|
f"Successfully deleted {count} old pending requests."
|
|
in output.getvalue().strip()
|
|
)
|