1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
import pytest
from authlib.oauth2.rfc6749.grants import ImplicitGrant
authorize_url = "/oauth/authorize?response_type=token&client_id=client-id"
@pytest.fixture(autouse=True)
def server(server):
server.register_grant(ImplicitGrant)
return server
@pytest.fixture(autouse=True)
def client(client, db):
client.set_client_metadata(
{
"redirect_uris": ["https://client.test/authorized"],
"scope": "profile",
"response_types": ["token"],
"grant_types": ["implicit"],
"token_endpoint_auth_method": "none",
}
)
db.session.add(client)
db.session.commit()
return client
def test_get_authorize(test_client):
rv = test_client.get(authorize_url)
assert rv.data == b"ok"
def test_confidential_client(test_client, db, client):
client.client_secret = "client-secret"
client.set_client_metadata(
{
"redirect_uris": ["https://client.test/authorized"],
"scope": "profile",
"response_types": ["token"],
"grant_types": ["implicit"],
"token_endpoint_auth_method": "client_secret_basic",
}
)
db.session.add(client)
db.session.commit()
rv = test_client.get(authorize_url)
assert b"invalid_client" in rv.data
def test_unsupported_client(test_client, db, client):
client.set_client_metadata(
{
"redirect_uris": ["https://client.test/authorized"],
"scope": "profile",
"response_types": ["code"],
"grant_types": ["implicit"],
"token_endpoint_auth_method": "none",
}
)
db.session.add(client)
db.session.commit()
rv = test_client.get(authorize_url)
assert "unauthorized_client" in rv.location
def test_invalid_authorize(test_client, server):
rv = test_client.post(authorize_url)
assert "#error=access_denied" in rv.location
server.scopes_supported = ["profile"]
rv = test_client.post(authorize_url + "&scope=invalid")
assert "#error=invalid_scope" in rv.location
def test_authorize_token(test_client):
rv = test_client.post(authorize_url, data={"user_id": "1"})
assert "access_token=" in rv.location
url = authorize_url + "&state=bar&scope=profile"
rv = test_client.post(url, data={"user_id": "1"})
assert "access_token=" in rv.location
assert "state=bar" in rv.location
assert "scope=profile" in rv.location
def test_token_generator(test_client, app, server):
m = "tests.flask.test_oauth2.oauth2_server:token_generator"
app.config.update({"OAUTH2_ACCESS_TOKEN_GENERATOR": m})
server.load_config(app.config)
rv = test_client.post(authorize_url, data={"user_id": "1"})
assert "access_token=c-implicit.1." in rv.location
|
