mirror of
https://github.com/simple-login/app.git
synced 2024-11-15 13:14:36 +08:00
24 lines
650 B
Python
24 lines
650 B
Python
|
from flask import url_for
|
||
|
|
from urllib.parse import parse_qs
|
||
|
|
from urllib3.util import parse_url
|
||
|
|
|
||
|
|
from app.config import URL, PROTON_CLIENT_ID
|
||
|
|
|
||
|
|
|
||
|
|
def test_login_with_proton(flask_client):
|
||
|
|
r = flask_client.get(
|
||
|
|
url_for("auth.proton_login"),
|
||
|
|
follow_redirects=False,
|
||
|
|
)
|
||
|
|
location = r.headers.get("Location")
|
||
|
|
assert location is not None
|
||
|
|
|
||
|
|
parsed = parse_url(location)
|
||
|
|
query = parse_qs(parsed.query)
|
||
|
|
|
||
|
|
expected_redirect_url = f"{URL}/auth/proton/callback"
|
||
|
|
|
||
|
|
assert "code" == query["response_type"][0]
|
||
|
|
assert PROTON_CLIENT_ID == query["client_id"][0]
|
||
|
|
assert expected_redirect_url == query["redirect_uri"][0]
|