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
|
import os
from unittest.mock import patch
from oauthlib import signals
from oauthlib.oauth2 import DeviceClient
from tests.unittest import TestCase
class DeviceClientTest(TestCase):
client_id = "someclientid"
kwargs = {
"some": "providers",
"require": "extra arguments"
}
client_secret = "asecret"
device_code = "somedevicecode"
scope = ["profile", "email"]
body = "not=empty"
body_up = "not=empty&grant_type=urn:ietf:params:oauth:grant-type:device_code"
body_code = body_up + "&device_code=somedevicecode"
body_kwargs = body_code + "&some=providers&require=extra+arguments"
uri = "https://example.com/path?query=world"
uri_id = uri + "&client_id=" + client_id
uri_grant = uri_id + "&grant_type=urn:ietf:params:oauth:grant-type:device_code"
uri_secret = uri_grant + "&client_secret=asecret"
uri_scope = uri_secret + "&scope=profile+email"
def test_request_body(self):
client = DeviceClient(self.client_id)
# Basic, no extra arguments
body = client.prepare_request_body(self.device_code, body=self.body)
self.assertFormBodyEqual(body, self.body_code)
rclient = DeviceClient(self.client_id)
body = rclient.prepare_request_body(self.device_code, body=self.body)
self.assertFormBodyEqual(body, self.body_code)
# With extra parameters
body = client.prepare_request_body(
self.device_code, body=self.body, **self.kwargs)
self.assertFormBodyEqual(body, self.body_kwargs)
def test_request_uri(self):
client = DeviceClient(self.client_id)
uri = client.prepare_request_uri(self.uri)
self.assertURLEqual(uri, self.uri_grant)
client = DeviceClient(self.client_id, client_secret=self.client_secret)
uri = client.prepare_request_uri(self.uri)
self.assertURLEqual(uri, self.uri_secret)
uri = client.prepare_request_uri(self.uri, scope=self.scope)
self.assertURLEqual(uri, self.uri_scope)
|
