blob: 9161f5a5f6721b5bf408b15245fa3e5220785131 (
plain)
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
|
"""Ensure that the server correctly uses the OIDC flavor of
the Refresh token grant type when appropriate.
When the OpenID scope is provided, the refresh token response
should include a fresh ID token.
"""
import json
from unittest import mock
from oauthlib.openid import RequestValidator
from oauthlib.openid.connect.core.endpoints.pre_configured import Server
from tests.unittest import TestCase
class TestRefreshToken(TestCase):
def setUp(self):
self.validator = mock.MagicMock(spec=RequestValidator)
self.validator.get_id_token.return_value='id_token'
self.server = Server(self.validator)
def test_refresh_token_with_openid(self):
request_body = 'scope=openid+test_scope&grant_type=refresh_token&refresh_token=abc'
headers, body, status = self.server.create_token_response('', body=request_body)
self.assertIn('id_token', json.loads(body))
def test_refresh_token_no_openid(self):
request_body = 'scope=test_scope&grant_type=refresh_token&refresh_token=abc'
headers, body, status = self.server.create_token_response('', body=request_body)
self.assertNotIn('id_token', json.loads(body))
|
