Skip to content

Commit 415e065

Browse files
dlmybaisk
authored andcommitted
add https_redirect_middle test
1 parent 35e23d9 commit 415e065

1 file changed

Lines changed: 13 additions & 1 deletion

File tree

tests/test_middlewares.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212

1313
from leancloud import user as user_module
1414
from leancloud.engine.cookie_session import CookieSessionMiddleware
15-
15+
from leancloud.engine import https_redirect_middleware
1616

1717
HOST, PORT = 'localhost', 80
1818
URL = 'http://{}:{}/'.format(HOST, PORT)
@@ -80,3 +80,15 @@ def test_cookie_session_middleware():
8080
break
8181

8282
remove_wsgi_intercept()
83+
84+
def test_https_redirect_middleware():
85+
https_redirect_middleware.is_prod = True
86+
app = https_redirect_middleware.HttpsRedirectMiddleware(application)
87+
add_wsgi_intercept(HOST, PORT, lambda: app)
88+
89+
response = requests.get(url=URL, allow_redirects=False)
90+
91+
assert response.is_redirect == True
92+
assert response.next.url[:5] == "https"
93+
94+
remove_wsgi_intercept()

0 commit comments

Comments
 (0)