173 lines
7.9 KiB
Python
173 lines
7.9 KiB
Python
# Copyright (C) 2014 Andrey Antukh <niwi@niwi.be>
|
|
# Copyright (C) 2014 Jesús Espino <jespinog@gmail.com>
|
|
# Copyright (C) 2014 David Barragán <bameda@dbarragan.com>
|
|
# Copyright (C) 2014 Anler Hernández <hello@anler.me>
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import pytest
|
|
|
|
from unittest.mock import patch, Mock
|
|
from taiga.base.connectors import github
|
|
from taiga.base.connectors import exceptions as exc
|
|
|
|
|
|
def test_url_builder():
|
|
assert (github._build_url("login", "authorize") ==
|
|
"https://api.github.com/login/oauth/authorize")
|
|
assert (github._build_url("login","access-token") ==
|
|
"https://api.github.com/login/oauth/access_token")
|
|
assert (github._build_url("user", "profile") ==
|
|
"https://api.github.com/user")
|
|
assert (github._build_url("user", "emails") ==
|
|
"https://api.github.com/user/emails")
|
|
|
|
|
|
def test_login_without_settings_params():
|
|
with pytest.raises(exc.GitHubApiError) as e, \
|
|
patch("taiga.base.connectors.github.requests") as m_requests:
|
|
m_requests.post.return_value = m_response = Mock()
|
|
m_response.status_code = 200
|
|
m_response.json.return_value = {"access_token": "xxxxxxxx"}
|
|
|
|
auth_info = github.login("*access-code*", "**client-id**", "*ient-secret*", github.HEADERS)
|
|
assert e.value.status_code == 400
|
|
assert "error_message" in e.value.detail
|
|
|
|
|
|
def test_login_success():
|
|
with patch("taiga.base.connectors.github.requests") as m_requests, \
|
|
patch("taiga.base.connectors.github.CLIENT_ID") as CLIENT_ID, \
|
|
patch("taiga.base.connectors.github.CLIENT_SECRET") as CLIENT_SECRET:
|
|
CLIENT_ID = "*CLIENT_ID*"
|
|
CLIENT_SECRET = "*CLIENT_SECRET*"
|
|
m_requests.post.return_value = m_response = Mock()
|
|
m_response.status_code = 200
|
|
m_response.json.return_value = {"access_token": "xxxxxxxx"}
|
|
|
|
auth_info = github.login("*access-code*", "**client-id**", "*client-secret*", github.HEADERS)
|
|
|
|
assert auth_info.access_token == "xxxxxxxx"
|
|
m_requests.post.assert_called_once_with("https://github.com/login/oauth/access_token",
|
|
headers=github.HEADERS,
|
|
params={'code': '*access-code*',
|
|
'scope': 'user:emails',
|
|
'client_id': '**client-id**',
|
|
'client_secret': '*client-secret*'})
|
|
|
|
|
|
def test_login_whit_errors():
|
|
with pytest.raises(exc.GitHubApiError) as e, \
|
|
patch("taiga.base.connectors.github.requests") as m_requests, \
|
|
patch("taiga.base.connectors.github.CLIENT_ID") as CLIENT_ID, \
|
|
patch("taiga.base.connectors.github.CLIENT_SECRET") as CLIENT_SECRET:
|
|
CLIENT_ID = "*CLIENT_ID*"
|
|
CLIENT_SECRET = "*CLIENT_SECRET*"
|
|
m_requests.post.return_value = m_response = Mock()
|
|
m_response.status_code = 200
|
|
m_response.json.return_value = {"error": "Invalid credentials"}
|
|
|
|
auth_info = github.login("*access-code*", "**client-id**", "*ient-secret*", github.HEADERS)
|
|
assert e.value.status_code == 400
|
|
assert e.value.detail["status_code"] == 200
|
|
assert e.value.detail["error"] == "Invalid credentials"
|
|
|
|
|
|
def test_get_user_profile_success():
|
|
with patch("taiga.base.connectors.github.requests") as m_requests:
|
|
m_requests.get.return_value = m_response = Mock()
|
|
m_response.status_code = 200
|
|
m_response.json.return_value = {"id": 1955,
|
|
"login": "mmcfly",
|
|
"name": "martin seamus mcfly",
|
|
"bio": "time traveler"}
|
|
|
|
user_profile = github.get_user_profile(github.HEADERS)
|
|
|
|
assert user_profile.id == 1955
|
|
assert user_profile.username == "mmcfly"
|
|
assert user_profile.full_name == "martin seamus mcfly"
|
|
assert user_profile.bio == "time traveler"
|
|
m_requests.get.assert_called_once_with("https://api.github.com/user",
|
|
headers=github.HEADERS)
|
|
|
|
|
|
def test_get_user_profile_whit_errors():
|
|
with pytest.raises(exc.GitHubApiError) as e, \
|
|
patch("taiga.base.connectors.github.requests") as m_requests:
|
|
m_requests.get.return_value = m_response = Mock()
|
|
m_response.status_code = 401
|
|
m_response.json.return_value = {"error": "Invalid credentials"}
|
|
|
|
auth_info = github.get_user_profile(github.HEADERS)
|
|
assert e.value.status_code == 400
|
|
assert e.value.detail["status_code"] == 401
|
|
assert e.value.detail["error"] == "Invalid credentials"
|
|
|
|
|
|
def test_get_user_emails_success():
|
|
with patch("taiga.base.connectors.github.requests") as m_requests:
|
|
m_requests.get.return_value = m_response = Mock()
|
|
m_response.status_code = 200
|
|
m_response.json.return_value = [{"email": "darth-vader@bttf.com", "primary": False},
|
|
{"email": "mmcfly@bttf.com", "primary": True}]
|
|
|
|
emails = github.get_user_emails(github.HEADERS)
|
|
|
|
assert len(emails) == 2
|
|
assert emails[0].email == "darth-vader@bttf.com"
|
|
assert not emails[0].is_primary
|
|
assert emails[1].email == "mmcfly@bttf.com"
|
|
assert emails[1].is_primary
|
|
m_requests.get.assert_called_once_with("https://api.github.com/user/emails",
|
|
headers=github.HEADERS)
|
|
|
|
|
|
def test_get_user_emails_whit_errors():
|
|
with pytest.raises(exc.GitHubApiError) as e, \
|
|
patch("taiga.base.connectors.github.requests") as m_requests:
|
|
m_requests.get.return_value = m_response = Mock()
|
|
m_response.status_code = 401
|
|
m_response.json.return_value = {"error": "Invalid credentials"}
|
|
|
|
emails = github.get_user_emails(github.HEADERS)
|
|
assert e.value.status_code == 400
|
|
assert e.value.detail["status_code"] == 401
|
|
assert e.value.detail["error"] == "Invalid credentials"
|
|
|
|
|
|
def test_me():
|
|
with patch("taiga.base.connectors.github.login") as m_login, \
|
|
patch("taiga.base.connectors.github.get_user_profile") as m_get_user_profile, \
|
|
patch("taiga.base.connectors.github.get_user_emails") as m_get_user_emails:
|
|
m_login.return_value = github.AuthInfo(access_token="xxxxxxxx")
|
|
m_get_user_profile.return_value = github.User(id=1955,
|
|
username="mmcfly",
|
|
full_name="martin seamus mcfly",
|
|
bio="time traveler")
|
|
m_get_user_emails.return_value = [github.Email(email="darth-vader@bttf.com", is_primary=False),
|
|
github.Email(email="mmcfly@bttf.com", is_primary=True)]
|
|
|
|
email, user = github.me("**access-code**")
|
|
|
|
assert email == "mmcfly@bttf.com"
|
|
assert user.id == 1955
|
|
assert user.username == "mmcfly"
|
|
assert user.full_name == "martin seamus mcfly"
|
|
assert user.bio == "time traveler"
|
|
|
|
headers = github.HEADERS.copy()
|
|
headers["Authorization"] = "token xxxxxxxx"
|
|
m_get_user_profile.assert_called_once_with(headers=headers)
|
|
m_get_user_emails.assert_called_once_with(headers=headers)
|