|
|
# Copyright (C) 2010-2024 RhodeCode GmbH
|
|
|
#
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
|
# it under the terms of the GNU Affero General Public License, version 3
|
|
|
# (only), as published by the Free Software Foundation.
|
|
|
#
|
|
|
# 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 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/>.
|
|
|
#
|
|
|
# This program is dual-licensed. If you wish to learn more about the
|
|
|
# RhodeCode Enterprise Edition, including its added features, Support services,
|
|
|
# and proprietary license terms, please see https://rhodecode.com/licenses/
|
|
|
|
|
|
"""
|
|
|
Test suite for making push/pull operations, on specially modified INI files
|
|
|
|
|
|
"""
|
|
|
|
|
|
import pytest
|
|
|
|
|
|
from rhodecode.model.auth_token import AuthTokenModel
|
|
|
from rhodecode.model.db import Repository
|
|
|
from rhodecode.model.meta import Session
|
|
|
from rhodecode.tests import (GIT_REPO, HG_REPO)
|
|
|
|
|
|
from rhodecode.tests.vcs_operations import (Command, _check_proper_clone)
|
|
|
|
|
|
|
|
|
@pytest.mark.usefixtures(
|
|
|
"init_pyramid_app",
|
|
|
"repo_group_repos",
|
|
|
"disable_anonymous_user",
|
|
|
"disable_locking",
|
|
|
)
|
|
|
class TestVCSOperationsByAuthTokens:
|
|
|
def test_clone_by_auth_token(
|
|
|
self, rcstack, tmpdir, user_util, enable_auth_plugins):
|
|
|
|
|
|
enable_auth_plugins.enable([
|
|
|
'egg:rhodecode-enterprise-ce#token',
|
|
|
'egg:rhodecode-enterprise-ce#rhodecode'
|
|
|
])
|
|
|
|
|
|
user = user_util.create_user()
|
|
|
token = user.auth_tokens[1]
|
|
|
|
|
|
clone_url = rcstack.repo_clone_url(
|
|
|
HG_REPO, user=user.username, passwd=token)
|
|
|
|
|
|
stdout, stderr = Command(tmpdir.strpath).execute(
|
|
|
'hg clone', clone_url, tmpdir.strpath)
|
|
|
|
|
|
_check_proper_clone(stdout, stderr, 'hg')
|
|
|
|
|
|
def test_clone_by_auth_token_expired(
|
|
|
self, rcstack, tmpdir, user_util, enable_auth_plugins):
|
|
|
enable_auth_plugins.enable([
|
|
|
'egg:rhodecode-enterprise-ce#token',
|
|
|
'egg:rhodecode-enterprise-ce#rhodecode'
|
|
|
])
|
|
|
|
|
|
user = user_util.create_user()
|
|
|
auth_token = AuthTokenModel().create(
|
|
|
user.user_id, 'test-token', -10, AuthTokenModel.cls.ROLE_VCS)
|
|
|
token = auth_token.api_key
|
|
|
|
|
|
clone_url = rcstack.repo_clone_url(
|
|
|
HG_REPO, user=user.username, passwd=token)
|
|
|
|
|
|
stdout, stderr = Command(tmpdir.strpath).execute(
|
|
|
'hg clone', clone_url, tmpdir.strpath)
|
|
|
assert 'abort: authorization failed' in stderr
|
|
|
|
|
|
msg = 'reason: bad or inactive token.'
|
|
|
rcstack.assert_message_in_server_logs(msg)
|
|
|
|
|
|
def test_clone_by_auth_token_bad_role(
|
|
|
self, rcstack, tmpdir, user_util, enable_auth_plugins):
|
|
|
enable_auth_plugins.enable([
|
|
|
'egg:rhodecode-enterprise-ce#token',
|
|
|
'egg:rhodecode-enterprise-ce#rhodecode'
|
|
|
])
|
|
|
|
|
|
user = user_util.create_user()
|
|
|
auth_token = AuthTokenModel().create(
|
|
|
user.user_id, 'test-token', -1, AuthTokenModel.cls.ROLE_API)
|
|
|
token = auth_token.api_key
|
|
|
|
|
|
clone_url = rcstack.repo_clone_url(
|
|
|
HG_REPO, user=user.username, passwd=token)
|
|
|
|
|
|
stdout, stderr = Command(tmpdir.strpath).execute(
|
|
|
'hg clone', clone_url, tmpdir.strpath)
|
|
|
assert 'abort: authorization failed' in stderr
|
|
|
|
|
|
def test_clone_by_auth_token_user_disabled(
|
|
|
self, rcstack, tmpdir, user_util, enable_auth_plugins):
|
|
|
enable_auth_plugins.enable([
|
|
|
'egg:rhodecode-enterprise-ce#token',
|
|
|
'egg:rhodecode-enterprise-ce#rhodecode'
|
|
|
])
|
|
|
|
|
|
user = user_util.create_user()
|
|
|
user.active = False
|
|
|
Session().add(user)
|
|
|
Session().commit()
|
|
|
token = user.auth_tokens[1]
|
|
|
|
|
|
clone_url = rcstack.repo_clone_url(
|
|
|
HG_REPO, user=user.username, passwd=token)
|
|
|
|
|
|
stdout, stderr = Command(tmpdir.strpath).execute(
|
|
|
'hg clone', clone_url, tmpdir.strpath)
|
|
|
assert 'abort: authorization failed' in stderr
|
|
|
|
|
|
msg = 'reason: account not active.'
|
|
|
rcstack.assert_message_in_server_logs(msg)
|
|
|
|
|
|
def test_clone_by_auth_token_with_scope(
|
|
|
self, rcstack, tmpdir, user_util, enable_auth_plugins):
|
|
|
enable_auth_plugins.enable([
|
|
|
'egg:rhodecode-enterprise-ce#token',
|
|
|
'egg:rhodecode-enterprise-ce#rhodecode'
|
|
|
])
|
|
|
|
|
|
user = user_util.create_user()
|
|
|
auth_token = AuthTokenModel().create(
|
|
|
user.user_id, 'test-token', -1, AuthTokenModel.cls.ROLE_VCS)
|
|
|
token = auth_token.api_key
|
|
|
|
|
|
# manually set scope
|
|
|
auth_token.repo = Repository.get_by_repo_name(HG_REPO)
|
|
|
Session().add(auth_token)
|
|
|
Session().commit()
|
|
|
|
|
|
clone_url = rcstack.repo_clone_url(
|
|
|
HG_REPO, user=user.username, passwd=token)
|
|
|
|
|
|
stdout, stderr = Command(tmpdir.strpath).execute(
|
|
|
'hg clone', clone_url, tmpdir.strpath)
|
|
|
_check_proper_clone(stdout, stderr, 'hg')
|
|
|
|
|
|
def test_clone_by_auth_token_with_wrong_scope(
|
|
|
self, rcstack, tmpdir, user_util, enable_auth_plugins):
|
|
|
enable_auth_plugins.enable([
|
|
|
'egg:rhodecode-enterprise-ce#token',
|
|
|
'egg:rhodecode-enterprise-ce#rhodecode'
|
|
|
])
|
|
|
|
|
|
user = user_util.create_user()
|
|
|
auth_token = AuthTokenModel().create(
|
|
|
user.user_id, 'test-token', -1, AuthTokenModel.cls.ROLE_VCS)
|
|
|
token = auth_token.api_key
|
|
|
|
|
|
# manually set scope
|
|
|
auth_token.repo = Repository.get_by_repo_name(GIT_REPO)
|
|
|
Session().add(auth_token)
|
|
|
Session().commit()
|
|
|
|
|
|
clone_url = rcstack.repo_clone_url(
|
|
|
HG_REPO, user=user.username, passwd=token)
|
|
|
|
|
|
stdout, stderr = Command(tmpdir.strpath).execute(
|
|
|
'hg clone', clone_url, tmpdir.strpath)
|
|
|
|
|
|
assert 'abort: authorization failed' in stderr
|
|
|
|
|
|
msg = 'reason: bad or inactive token.'
|
|
|
rcstack.assert_message_in_server_logs(msg)
|
|
|
|
