File: test_implicit_grant.py

package info (click to toggle)
python-authlib 1.6.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 3,016 kB
  • sloc: python: 26,998; makefile: 53; sh: 14
file content (77 lines) | stat: -rw-r--r-- 2,753 bytes parent folder | download | duplicates (2)
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
import pytest

from authlib.common.urls import url_decode
from authlib.common.urls import urlparse
from authlib.oauth2.rfc6749 import errors
from authlib.oauth2.rfc6749 import grants

from .models import Client
from .models import User
from .oauth2_server import TestCase


class ImplicitTest(TestCase):
    def create_server(self):
        server = super().create_server()
        server.register_grant(grants.ImplicitGrant)
        return server

    def prepare_data(self, response_type="token", scope=""):
        user = User(username="foo")
        user.save()
        client = Client(
            user_id=user.pk,
            client_id="client",
            response_type=response_type,
            scope=scope,
            token_endpoint_auth_method="none",
            default_redirect_uri="https://a.b",
        )
        client.save()

    def test_get_consent_grant_client(self):
        server = self.create_server()
        url = "/authorize?response_type=token"
        request = self.factory.get(url)
        with pytest.raises(errors.InvalidClientError):
            server.get_consent_grant(request)

        url = "/authorize?response_type=token&client_id=client"
        request = self.factory.get(url)
        with pytest.raises(errors.InvalidClientError):
            server.get_consent_grant(request)

        self.prepare_data(response_type="")
        with pytest.raises(errors.UnauthorizedClientError):
            server.get_consent_grant(request)

    def test_get_consent_grant_scope(self):
        server = self.create_server()
        server.scopes_supported = ["profile"]

        self.prepare_data()
        base_url = "/authorize?response_type=token&client_id=client"
        url = base_url + "&scope=invalid"
        request = self.factory.get(url)
        with pytest.raises(errors.InvalidScopeError):
            server.get_consent_grant(request)

    def test_create_authorization_response(self):
        server = self.create_server()
        self.prepare_data()
        data = {"response_type": "token", "client_id": "client"}
        request = self.factory.post("/authorize", data=data)
        grant = server.get_consent_grant(request)

        resp = server.create_authorization_response(request, grant=grant)
        assert resp.status_code == 302
        params = dict(url_decode(urlparse.urlparse(resp["Location"]).fragment))
        assert params["error"] == "access_denied"

        grant_user = User.objects.get(username="foo")
        resp = server.create_authorization_response(
            request, grant=grant, grant_user=grant_user
        )
        assert resp.status_code == 302
        params = dict(url_decode(urlparse.urlparse(resp["Location"]).fragment))
        assert "access_token" in params