151bd13666
* Generate common/djangoapps import shims for LMS * Generate common/djangoapps import shims for Studio * Stop appending project root to sys.path * Stop appending common/djangoapps to sys.path * Import from common.djangoapps.course_action_state instead of course_action_state * Import from common.djangoapps.course_modes instead of course_modes * Import from common.djangoapps.database_fixups instead of database_fixups * Import from common.djangoapps.edxmako instead of edxmako * Import from common.djangoapps.entitlements instead of entitlements * Import from common.djangoapps.pipline_mako instead of pipeline_mako * Import from common.djangoapps.static_replace instead of static_replace * Import from common.djangoapps.student instead of student * Import from common.djangoapps.terrain instead of terrain * Import from common.djangoapps.third_party_auth instead of third_party_auth * Import from common.djangoapps.track instead of track * Import from common.djangoapps.util instead of util * Import from common.djangoapps.xblock_django instead of xblock_django * Add empty common/djangoapps/__init__.py to fix pytest collection * Fix pylint formatting violations * Exclude import_shims/ directory tree from linting
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
"""Tests for disabling rate limiting. """
|
|
|
|
|
|
import unittest
|
|
|
|
import mock
|
|
from django.conf import settings
|
|
from django.core.cache import cache
|
|
from django.test import TestCase
|
|
from rest_framework.exceptions import Throttled
|
|
from rest_framework.throttling import BaseThrottle
|
|
from rest_framework.views import APIView
|
|
|
|
from common.djangoapps.util.disable_rate_limit import can_disable_rate_limit
|
|
from common.djangoapps.util.models import RateLimitConfiguration
|
|
|
|
|
|
class FakeThrottle(BaseThrottle):
|
|
def allow_request(self, request, view):
|
|
return False
|
|
|
|
|
|
@can_disable_rate_limit
|
|
class FakeApiView(APIView):
|
|
authentication_classes = []
|
|
permission_classes = []
|
|
throttle_classes = [FakeThrottle]
|
|
|
|
|
|
@unittest.skipUnless(settings.ROOT_URLCONF == 'lms.urls', 'Test only valid in lms')
|
|
class DisableRateLimitTest(TestCase):
|
|
"""Check that we can disable rate limiting for perf testing. """
|
|
|
|
def setUp(self):
|
|
super(DisableRateLimitTest, self).setUp()
|
|
cache.clear()
|
|
self.view = FakeApiView()
|
|
|
|
def test_enable_rate_limit(self):
|
|
# Enable rate limiting using model-based config
|
|
RateLimitConfiguration.objects.create(enabled=True)
|
|
|
|
# By default, should enforce rate limiting
|
|
# Since our fake throttle always rejects requests,
|
|
# we should expect the request to be rejected.
|
|
request = mock.Mock()
|
|
with self.assertRaises(Throttled):
|
|
self.view.check_throttles(request)
|
|
|
|
def test_disable_rate_limit(self):
|
|
# Disable rate limiting using model-based config
|
|
RateLimitConfiguration.objects.create(enabled=False)
|
|
|
|
# With rate-limiting disabled, the request
|
|
# should get through. The `check_throttles()` call
|
|
# should return without raising an exception.
|
|
request = mock.Mock()
|
|
self.view.check_throttles(request)
|