mirror of
https://github.com/barkeser2002/offline-db.git
synced 2026-09-25 06:00:11 +03:00
1679d766c9
Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com> Co-authored-by: Senpai-YoloBot <41898282+github-actions[bot]@users.noreply.github.com>
20 lines
1.9 KiB
Python
20 lines
1.9 KiB
Python
import glob
|
|
import os
|
|
|
|
test_files = glob.glob('users/tests/test_*.py')
|
|
for file in test_files:
|
|
if file in ('users/tests/test_profile_view.py', 'users/tests/test_serializers.py', 'users/tests/test_badge_strategy_base.py', 'users/tests/test_services_performance.py', 'users/tests/test_watch_log_api.py', 'users/tests/test_websocket_notifications.py', 'users/tests/test_notifications.py', 'users/tests/test_api_rate_limits.py', 'users/tests/test_api_throttling.py', 'users/tests/test_profile_update.py', 'users/tests/test_admin_performance.py', 'users/tests/test_models.py', 'users/tests/test_idor_auth.py', 'users/tests/test_validation.py', 'users/tests/test_badges_api.py'):
|
|
continue
|
|
|
|
with open(file, 'r') as f:
|
|
content = f.read()
|
|
|
|
new_content = content.replace(" assert UserBadge.objects.filter(user=user, badge=super_fan_badge).exists()", " from django.core.cache import cache\n cache.delete(f'user_{user.id}_badges_checked')\n check_badges(user)\n assert UserBadge.objects.filter(user=user, badge=super_fan_badge).exists()")
|
|
|
|
new_content = new_content.replace(" self.assertTrue(\n UserBadge.objects.filter(user=self.user, badge__slug='daily-viewer').exists(),", " from django.core.cache import cache\n cache.delete(f'user_{self.user.id}_badges_checked')\n check_badges(self.user)\n self.assertTrue(\n UserBadge.objects.filter(user=self.user, badge__slug='daily-viewer').exists(),")
|
|
|
|
new_content = new_content.replace(" self.assertTrue(\n UserBadge.objects.filter(user=self.user, badge__slug='streak-master').exists(),", " from django.core.cache import cache\n cache.delete(f'user_{self.user.id}_badges_checked')\n check_badges(self.user)\n self.assertTrue(\n UserBadge.objects.filter(user=self.user, badge__slug='streak-master').exists(),")
|
|
|
|
with open(file, 'w') as f:
|
|
f.write(new_content)
|