class TestGranularUserRateThrottle(TestCase):
def setUp(self):
self.throttle = GranularUserRateThrottle()
def test_backwards_compatible_format(self):
# test the original DRF rate string format x/timeperiod works
assert self.throttle.parse_rate('1/minute') == (1, 60)
assert self.throttle.parse_rate('24/s') == (24, 1)
assert self.throttle.parse_rate('456/hour') == (456, 3600)
def test_granular_format(self):
assert self.throttle.parse_rate('1/5minute') == (1, 60 * 5)
assert self.throttle.parse_rate('24/1s') == (24, 1)
assert self.throttle.parse_rate('456/7hour') == (456, 7 * 3600)
@mock.patch('rest_framework.throttling.UserRateThrottle.allow_request')
def test_allow_request_if_api_throttling_setting_is_false(
self, allow_request_mock):
request = RequestFactory().get('/test')
view = object()
# Pretend the parent class would always throttle requests if called.
allow_request_mock.return_value = False
# With the setting set to True (the default), throttle as normal.
assert settings.API_THROTTLING is True
assert self.throttle.allow_request(request, view) is False
assert allow_request_mock.call_count == 1
# With the setting set to False, ignore throttling.
with override_settings(API_THROTTLING=False):
assert settings.API_THROTTLING is False
assert self.throttle.allow_request(request, view) is True
# The parent class hasn't been called an additional time.
assert allow_request_mock.call_count == 1
# And again set to True to be sure.
assert settings.API_THROTTLING is True
assert self.throttle.allow_request(request, view) is False
assert allow_request_mock.call_count == 2
class TestGranularUserRateThrottle(TestCase):
def setUp(self):
self.throttle = GranularUserRateThrottle()
def test_backwards_compatible_format(self):
# test the original DRF rate string format x/timeperiod works
assert self.throttle.parse_rate('1/minute') == (1, 60)
assert self.throttle.parse_rate('24/s') == (24, 1)
assert self.throttle.parse_rate('456/hour') == (456, 3600)
def test_granular_format(self):
assert self.throttle.parse_rate('1/5minute') == (1, 60 * 5)
assert self.throttle.parse_rate('24/1s') == (24, 1)
assert self.throttle.parse_rate('456/7hour') == (456, 7 * 3600)
def setUp(self):
self.throttle = GranularUserRateThrottle()
def setUp(self):
self.throttle = GranularUserRateThrottle()
class TestGranularUserRateThrottle(TestCase):
def setUp(self):
self.throttle = GranularUserRateThrottle()
def test_backwards_compatible_format(self):
# test the original DRF rate string format x/timeperiod works
assert self.throttle.parse_rate('1/minute') == (1, 60)
assert self.throttle.parse_rate('24/s') == (24, 1)
assert self.throttle.parse_rate('456/hour') == (456, 3600)
def test_granular_format(self):
assert self.throttle.parse_rate('1/5minute') == (1, 60 * 5)
assert self.throttle.parse_rate('24/1s') == (24, 1)
assert self.throttle.parse_rate('456/7hour') == (456, 7 * 3600)
@mock.patch('rest_framework.throttling.UserRateThrottle.allow_request')
def test_allow_request_if_api_throttling_setting_is_false(
self, allow_request_mock):
request = RequestFactory().get('/test')
view = object()
# Pretend the parent class would always throttle requests if called.
allow_request_mock.return_value = False
# With the setting set to True (the default), throttle as normal.
assert settings.API_THROTTLING is True
assert self.throttle.allow_request(request, view) is False
assert allow_request_mock.call_count == 1
# With the setting set to False, ignore throttling.
with override_settings(API_THROTTLING=False):
assert settings.API_THROTTLING is False
assert self.throttle.allow_request(request, view) is True
# The parent class hasn't been called an additional time.
assert allow_request_mock.call_count == 1
# And again set to True to be sure.
assert settings.API_THROTTLING is True
assert self.throttle.allow_request(request, view) is False
assert allow_request_mock.call_count == 2
def test_freeze_time_works_with_throttling(self):
old_time = self.throttle.timer()
with freeze_time('2019-04-08 15:16:23.42'):
self.throttle.timer() == 1554736583.42
new_time = self.throttle.timer()
assert new_time != 1554736583.42
assert old_time != 1554736583.42
assert old_time != new_time
@mock.patch('rest_framework.throttling.UserRateThrottle.allow_request')
def test_activity_log(self, allow_request_mock):
request = RequestFactory().get('/test')
view = object()
allow_request_mock.return_value = False
assert self.throttle.allow_request(request, view) is False
# Shouldn't be any ActivityLog since there was no user associated with
# that request.
assert not ActivityLog.objects.exists()
allow_request_mock.return_value = True
request.user = user_factory()
assert self.throttle.allow_request(request, view) is True
# Shouldn't be any ActivityLog since the request was not throttled.
assert not ActivityLog.objects.exists()
allow_request_mock.return_value = False
request.user = user_factory()
assert self.throttle.allow_request(request, view) is False
assert ActivityLog.objects.exists()
activity = ActivityLog.objects.get()
assert activity.action == amo.LOG.THROTTLED.id
assert activity.arguments == [self.throttle.scope]
assert activity.user == request.user
activity_str = str(activity)
assert '/user/%d/' % request.user.pk in activity_str
assert self.throttle.scope in activity_str