adding test for different http response code

jazzband · Aug 19, 2021 · 2368a7a · 2368a7a
1 parent 6858aea
commit 2368a7a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_helpers.py b/tests/test_helpers.py
@@ -677,6 +677,11 @@ def test_get_lockout_response(self):
         response = get_lockout_response(self.request, self.credentials)
         self.assertEqual(403, response.status_code)
 
+    @override_settings(AXES_HTTP_RESPONSE_CODE=429)
+    def test_get_lockout_response_with_custom_http_response_code(self):
+        response = get_lockout_response(self.request, self.credentials)
+        self.assertEqual(429, response.status_code)
+
     @override_settings(AXES_LOCKOUT_CALLABLE=mock_get_lockout_response)
     def test_get_lockout_response_override_callable(self):
         response = get_lockout_response(self.request, self.credentials)