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
|
from unittest import mock
import pytest
import responses
from flask_dance.consumer.requests import OAuth1Session, OAuth2Session
FAKE_OAUTH1_TOKEN = {"oauth_token": "abcdefg", "oauth_token_secret": "hijklmnop"}
FAKE_OAUTH2_TOKEN = {
"access_token": "deadbeef",
"scope": ["custom"],
"token_type": "bearer",
}
def test_oauth1session_authorized():
bp = mock.Mock(token=FAKE_OAUTH1_TOKEN)
sess = OAuth1Session(client_key="ckey", client_secret="csec", blueprint=bp)
sess.load_token = mock.Mock(wraps=sess.load_token)
assert sess.authorized == True
assert sess.load_token.called
def test_oauth1session_not_authorized():
bp = mock.Mock(token=None)
sess = OAuth1Session(client_key="ckey", client_secret="csec", blueprint=bp)
sess.load_token = mock.Mock(wraps=sess.load_token)
assert sess.authorized == False
assert sess.load_token.called
@responses.activate
def test_oauth1session_request():
responses.add(responses.GET, "https://example.com/test")
bp = mock.Mock(token=None)
sess = OAuth1Session(client_key="ckey", client_secret="csec", blueprint=bp)
sess.load_token = mock.Mock(wraps=sess.load_token)
sess.get("https://example.com/test")
assert sess.load_token.called
@responses.activate
def test_oauth1session_should_load_token():
responses.add(responses.GET, "https://example.com/test")
bp = mock.Mock(token=None)
sess = OAuth1Session(client_key="ckey", client_secret="csec", blueprint=bp)
sess.load_token = mock.Mock(wraps=sess.load_token)
sess.get("https://example.com/test", should_load_token=False)
assert not sess.load_token.called
def test_oauth2session_authorized():
bp = mock.Mock(token=FAKE_OAUTH2_TOKEN)
sess = OAuth2Session(client_id="cid", blueprint=bp)
assert sess.authorized == True
def test_oauth2session_not_authorized():
bp = mock.Mock(token=None)
sess = OAuth2Session(client_id="cid", blueprint=bp)
assert sess.authorized == False
def test_oauth2session_token():
bp = mock.Mock(token=FAKE_OAUTH2_TOKEN)
sess = OAuth2Session(client_id="cid", blueprint=bp)
assert sess.token == FAKE_OAUTH2_TOKEN
def test_oauth2session_unset_token():
bp = mock.Mock(token=None)
sess = OAuth2Session(client_id="cid", blueprint=bp)
assert sess.token == None
def test_oauth2session_access_token():
bp = mock.Mock(token=FAKE_OAUTH2_TOKEN)
sess = OAuth2Session(client_id="cid", blueprint=bp)
assert sess.access_token == "deadbeef"
def test_oauth2session_unset_access_token():
bp = mock.Mock(token=None)
sess = OAuth2Session(client_id="cid", blueprint=bp)
assert sess.access_token == None
|
