60 lines
3.1 KiB
Python
60 lines
3.1 KiB
Python
"""delete_users: کاربر و همهی دادهاش در همهی اپها، و هیچ چیزِ کاربرِ دیگری."""
|
|
from io import StringIO
|
|
|
|
from django.core.management import CommandError, call_command
|
|
from django.test import TestCase, override_settings
|
|
|
|
from accounts.models import OTPCode, Profile, User
|
|
from ai_tutor.models import Conversation, Message
|
|
from notifications.models import Notification
|
|
from payment.models import Order
|
|
from progress.models import UserStats
|
|
|
|
|
|
class DeleteUsersTests(TestCase):
|
|
def setUp(self):
|
|
self.doomed = User.objects.create(phone='09121110001', is_guest=False)
|
|
self.kept = User.objects.create(phone='09121110002', is_guest=False)
|
|
for user in (self.doomed, self.kept):
|
|
Profile.objects.create(user=user)
|
|
UserStats.objects.create(user_id=user.id, xp=10)
|
|
Message.objects.create(conversation=Conversation.objects.create(user_id=user.id), role='user', text='؟')
|
|
Notification.objects.create(user_id=user.id, type='xp_earned', title='ت')
|
|
OTPCode.objects.create(phone=user.phone, code='123456', expires_at='2030-01-01T00:00Z')
|
|
|
|
def run_command(self, *args):
|
|
out = StringIO()
|
|
call_command('delete_users', *args, stdout=out)
|
|
return out.getvalue()
|
|
|
|
def test_dry_run_changes_nothing(self):
|
|
out = self.run_command('--phones', self.doomed.phone)
|
|
self.assertIn('progress.UserStats.user_id: 1', out)
|
|
self.assertTrue(User.objects.filter(pk=self.doomed.pk).exists())
|
|
|
|
def test_everything_of_that_user_goes_and_nothing_else(self):
|
|
self.run_command('--phones', self.doomed.phone, '--yes')
|
|
self.assertFalse(User.objects.filter(pk=self.doomed.pk).exists())
|
|
for model in (UserStats, Conversation, Notification):
|
|
self.assertFalse(model.objects.filter(user_id=self.doomed.pk).exists(), model)
|
|
self.assertTrue(model.objects.filter(user_id=self.kept.pk).exists(), model)
|
|
self.assertFalse(OTPCode.objects.filter(phone=self.doomed.phone).exists())
|
|
self.assertTrue(OTPCode.objects.filter(phone=self.kept.phone).exists())
|
|
self.assertEqual(Message.objects.count(), 1)
|
|
|
|
def test_paid_orders_need_explicit_permission(self):
|
|
Order.objects.create(user_id=self.doomed.id, amount=1000, status='paid', idempotency_key='k1')
|
|
with self.assertRaises(CommandError):
|
|
self.run_command('--ids', str(self.doomed.pk), '--yes')
|
|
self.assertTrue(User.objects.filter(pk=self.doomed.pk).exists())
|
|
self.run_command('--ids', str(self.doomed.pk), '--yes', '--include-paid')
|
|
self.assertFalse(Order.objects.filter(user_id=self.doomed.pk).exists())
|
|
|
|
@override_settings(TEST_LOGIN_PHONES={'09121110002': '1234'})
|
|
def test_real_phones_keeps_demo_numbers_and_guests(self):
|
|
guest = User.objects.create(device_id='guest-x', is_guest=True)
|
|
self.run_command('--real-phones', '--yes')
|
|
self.assertFalse(User.objects.filter(pk=self.doomed.pk).exists())
|
|
self.assertTrue(User.objects.filter(pk=self.kept.pk).exists())
|
|
self.assertTrue(User.objects.filter(pk=guest.pk).exists())
|