Add test

Author: ronanboiteau

django_ratelimit/tests.py

@@ -7,13 +7,17 @@
 from django.utils.decorators import method_decorator
 from django.views.generic import View
 
+from rest_framework.decorators import api_view
+from rest_framework.test import APIRequestFactory
+
 from django_ratelimit.decorators import ratelimit
 from django_ratelimit.exceptions import Ratelimited
 from django_ratelimit.core import (get_usage, is_ratelimited,
                                    _split_rate, _get_ip)
 
 
 rf = RequestFactory()
+rest_rf = APIRequestFactory()
 
 
 class MockUser:
@@ -152,6 +156,17 @@ def view(request):
         assert not view(rf.post('/', {'foo': 'b'}))
         assert view(rf.post('/', {'foo': 'b'}))
 
+    def test_key_data(self):
+        @api_view(['POST'])
+        @ratelimit(key='data:foo', rate='1/m', block=False)
+        def view(request):
+            return request.limited
+
+        assert not view(rest_rf.post('/', {'foo': 'a'}))
+        assert view(rest_rf.post('/', {'foo': 'a'}))
+        assert not view(rest_rf.post('/', {'foo': 'b'}))
+        assert view(rest_rf.post('/', {'foo': 'b'}))
+
     def test_key_header(self):
         def _req():
             req = rf.post('/')

test_settings.py

@@ -4,6 +4,7 @@
 
 INSTALLED_APPS = (
     'django_ratelimit',
+    'rest_framework',
 )
 
 RATELIMIT_USE_CACHE = 'default'

tox.ini

@@ -15,6 +15,7 @@ deps =
     django41: Django>=4.1,<4.2
     django42: Django>=4.2,<4.3
     djangomain: https://github.com/django/django/archive/main.tar.gz
+    djangorestframework~=3.14.0
     pymemcache>=4.0,<5.0
     django-redis>=5.2,<6.0
     flake8