Expose id_token in OAuth 2.0 credentials (#150)
diff --git a/tests/oauth2/test_credentials.py b/tests/oauth2/test_credentials.py
index b117ad4..e15766a 100644
--- a/tests/oauth2/test_credentials.py
+++ b/tests/oauth2/test_credentials.py
@@ -58,6 +58,7 @@
def test_refresh_success(self, now_mock, refresh_grant_mock):
token = 'token'
expiry = _helpers.utcnow() + datetime.timedelta(seconds=500)
+ grant_response = {'id_token': mock.sentinel.id_token}
refresh_grant_mock.return_value = (
# Access token
token,
@@ -66,7 +67,7 @@
# Expiry,
expiry,
# Extra data
- {})
+ grant_response)
request_mock = mock.Mock()
# Refresh credentials
@@ -80,6 +81,7 @@
# Check that the credentials have the token and expiry
assert self.credentials.token == token
assert self.credentials.expiry == expiry
+ assert self.credentials.id_token == mock.sentinel.id_token
# Check that the credentials are valid (have a token and are not
# expired)