Fix test_views_rate_limit.py

potiuk · Apr 13, 2024 · a9aa27d · a9aa27d
1 parent 195d545
commit a9aa27d
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/tests/www/views/test_views_rate_limit.py b/tests/www/views/test_views_rate_limit.py
@@ -53,17 +53,26 @@ def factory():
 
 def test_rate_limit_one(app_with_rate_limit_one):
     client_with_login(
-        app_with_rate_limit_one, expected_response_code=302, username="test_admin", password="test_admin"
+        app_with_rate_limit_one,
+        expected_path=b"/login/?next=/home",
+        username="test_admin",
+        password="test_admin",
     )
     client_with_login(
-        app_with_rate_limit_one, expected_response_code=429, username="test_admin", password="test_admin"
+        app_with_rate_limit_one,
+        expected_path=b"/login/",
+        username="test_admin",
+        password="test_admin",
     )
     client_with_login(
-        app_with_rate_limit_one, expected_response_code=429, username="test_admin", password="test_admin"
+        app_with_rate_limit_one,
+        expected_path=b"/login/",
+        username="test_admin",
+        password="test_admin",
     )
 
 
 def test_rate_limit_disabled(app):
-    client_with_login(app, expected_response_code=302, username="test_admin", password="test_admin")
-    client_with_login(app, expected_response_code=302, username="test_admin", password="test_admin")
-    client_with_login(app, expected_response_code=302, username="test_admin", password="test_admin")
+    client_with_login(app, username="test_admin", password="test_admin")
+    client_with_login(app, username="test_admin", password="test_admin")
+    client_with_login(app, username="test_admin", password="test_admin")