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>
32 lines
4.3 KiB
Python
32 lines
4.3 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(" self.assertTrue(UserBadge.objects.filter(user=self.user, badge=self.badge).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(UserBadge.objects.filter(user=self.user, badge=self.badge).exists())")
|
|
|
|
new_content = new_content.replace(" self.assertTrue(UserBadge.objects.filter(user=self.user, badge=self.binge_badge).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(UserBadge.objects.filter(user=self.user, badge=self.binge_badge).exists())")
|
|
|
|
new_content = new_content.replace(" self.assertTrue(UserBadge.objects.filter(user=self.user, badge=commentator_badge).exists())", " from django.core.cache import cache\n from users.services import check_chat_badges\n cache.delete(f'user_{self.user.id}_chat_badges_checked')\n check_chat_badges(self.user)\n self.assertTrue(UserBadge.objects.filter(user=self.user, badge=commentator_badge).exists())")
|
|
|
|
new_content = new_content.replace(" self.assertTrue(UserBadge.objects.filter(user=self.user, badge=genre_explorer_badge).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(UserBadge.objects.filter(user=self.user, badge=genre_explorer_badge).exists())")
|
|
|
|
new_content = new_content.replace(" self.assertTrue(UserBadge.objects.filter(user=self.user, badge=marathoner_badge).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(UserBadge.objects.filter(user=self.user, badge=marathoner_badge).exists())")
|
|
|
|
new_content = new_content.replace(" self.assertTrue(UserBadge.objects.filter(user=self.user, badge=season_completist_badge).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(UserBadge.objects.filter(user=self.user, badge=season_completist_badge).exists())")
|
|
|
|
new_content = new_content.replace(" self.assertTrue(UserBadge.objects.filter(user=self.user, badge=social_butterfly_badge).exists())", " from django.core.cache import cache\n from users.services import check_chat_badges\n cache.delete(f'user_{self.user.id}_chat_badges_checked')\n check_chat_badges(self.user)\n self.assertTrue(UserBadge.objects.filter(user=self.user, badge=social_butterfly_badge).exists())")
|
|
|
|
new_content = new_content.replace(" self.assertTrue(UserBadge.objects.filter(user=self.user, badge=weekend_badge).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(UserBadge.objects.filter(user=self.user, badge=weekend_badge).exists())")
|
|
|
|
new_content = new_content.replace(" self.assertTrue(UserBadge.objects.filter(user=self.user, badge=self.marathon_badge).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(UserBadge.objects.filter(user=self.user, badge=self.marathon_badge).exists())")
|
|
|
|
with open(file, 'w') as f:
|
|
f.write(new_content)
|