From a9aa27d99da3724749a10a7f94c96b6d91eae6ef Mon Sep 17 00:00:00 2001 From: satoshi-sh Date: Sat, 13 Apr 2024 15:37:41 -0500 Subject: [PATCH] Fix test_views_rate_limit.py --- tests/www/views/test_views_rate_limit.py | 21 +++++++++++++++------ 1 file changed, 15 insertions(+), 6 deletions(-) diff --git a/tests/www/views/test_views_rate_limit.py b/tests/www/views/test_views_rate_limit.py index 3f5b411abbc45..284be424be24d 100644 --- 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")