blob: a16c9044cb13b7e768db1c7f0fa3b764cca9ecad [file] [log] [blame]
# Copyright 2017 Google Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import json
import os
import mock
import pytest
from google.oauth2 import oauthlib
DATA_DIR = os.path.join(os.path.dirname(__file__), '..', 'data')
CLIENT_SECRETS_FILE = os.path.join(DATA_DIR, 'client_secrets.json')
with open(CLIENT_SECRETS_FILE, 'r') as fh:
CLIENT_SECRETS_INFO = json.load(fh)
def test_session_from_client_config_web():
session, config = oauthlib.session_from_client_config(
CLIENT_SECRETS_INFO, scopes=mock.sentinel.scopes)
assert config == CLIENT_SECRETS_INFO
assert session.client_id == CLIENT_SECRETS_INFO['web']['client_id']
assert session.scope == mock.sentinel.scopes
def test_session_from_client_config_installed():
info = {'installed': CLIENT_SECRETS_INFO['web']}
session, config = oauthlib.session_from_client_config(
info, scopes=mock.sentinel.scopes)
assert config == info
assert session.client_id == info['installed']['client_id']
assert session.scope == mock.sentinel.scopes
def test_session_from_client_config_bad_format():
with pytest.raises(ValueError):
oauthlib.session_from_client_config({}, scopes=[])
def test_session_from_client_config_missing_keys():
with pytest.raises(ValueError):
oauthlib.session_from_client_config({'web': {}}, scopes=[])
def test_session_from_client_secrets_file():
session, config = oauthlib.session_from_client_secrets_file(
CLIENT_SECRETS_FILE, scopes=mock.sentinel.scopes)
assert config == CLIENT_SECRETS_INFO
assert session.client_id == CLIENT_SECRETS_INFO['web']['client_id']
assert session.scope == mock.sentinel.scopes
@pytest.fixture
def session():
session, _ = oauthlib.session_from_client_config(
CLIENT_SECRETS_INFO, scopes=mock.sentinel.scopes)
yield session
def test_credentials_from_session(session):
session.token = {
'access_token': mock.sentinel.access_token,
'refresh_token': mock.sentinel.refresh_token
}
credentials = oauthlib.credentials_from_session(
session, CLIENT_SECRETS_INFO['web'])
assert credentials.token == mock.sentinel.access_token
assert credentials._refresh_token == mock.sentinel.refresh_token
assert credentials._client_id == CLIENT_SECRETS_INFO['web']['client_id']
assert (credentials._client_secret ==
CLIENT_SECRETS_INFO['web']['client_secret'])
assert credentials._token_uri == CLIENT_SECRETS_INFO['web']['token_uri']
def test_bad_credentials(session):
with pytest.raises(ValueError):
oauthlib.credentials_from_session(session)