From 8b30218d151c544c3d497341eb45a970dcc4a975 Mon Sep 17 00:00:00 2001
From: Artem Muterko <artieua@gmail.com>
Date: Thu, 12 Jan 2017 17:39:05 +0200
Subject: [PATCH] Add tests for AnonRateThrottle

---
 tests/test_throttling.py | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/tests/test_throttling.py b/tests/test_throttling.py
index f5b434458..680b038d9 100644
--- a/tests/test_throttling.py
+++ b/tests/test_throttling.py
@@ -7,13 +7,16 @@ import pytest
 from django.contrib.auth.models import User
 from django.core.cache import cache
 from django.core.exceptions import ImproperlyConfigured
+from django.http import HttpRequest
 from django.test import TestCase
 
+from rest_framework.request import Request
 from rest_framework.response import Response
 from rest_framework.settings import api_settings
-from rest_framework.test import APIRequestFactory
+from rest_framework.test import APIRequestFactory, force_authenticate
 from rest_framework.throttling import (
-    BaseThrottle, ScopedRateThrottle, SimpleRateThrottle, UserRateThrottle
+    AnonRateThrottle, BaseThrottle, ScopedRateThrottle, SimpleRateThrottle,
+    UserRateThrottle
 )
 from rest_framework.views import APIView
 
@@ -414,3 +417,22 @@ class SimpleRateThrottleTests(TestCase):
         throttle.now = throttle.timer()
         throttle.history = [throttle.timer() for _ in range(3)]
         assert throttle.wait() is None
+
+
+class AnonRateThrottleTests(TestCase):
+
+    def setUp(self):
+        self.throttle = AnonRateThrottle()
+
+    def test_authenticated_user_not_affected(self):
+        request = Request(HttpRequest())
+        user = User.objects.create(username='test')
+
+        force_authenticate(request, user, token='test_token')
+        request.user = user
+        assert self.throttle.get_cache_key(request, view={}) is None
+
+    def test_get_cache_key_returns_correct_value(self):
+        request = Request(HttpRequest())
+        cache_key = self.throttle.get_cache_key(request, view={})
+        assert cache_key == 'throttle_anon_None'