Show More
test_utils.py
93 lines
| 3.3 KiB
| text/x-python
|
PythonLexer
r1 | # -*- coding: utf-8 -*- | |||
r1271 | # Copyright (C) 2010-2017 RhodeCode GmbH | |||
r1 | # | |||
# 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/ | ||||
import pytest | ||||
from rhodecode.controllers import utils | ||||
from rhodecode.lib.vcs.exceptions import RepositoryError | ||||
import mock | ||||
@pytest.mark.parametrize('alias, expected', [ | ||||
('hg', [None, 'name']), | ||||
('git', [None, 'name']), | ||||
('svn', ['name', 'raw_id']), | ||||
]) | ||||
def test_parse_path_ref_understands_format_ref_id_result(alias, expected): | ||||
repo = mock.Mock(alias=alias) | ||||
# Formatting of reference ids as it is used by controllers | ||||
format_ref_id = utils.get_format_ref_id(repo) | ||||
formatted_ref_id = format_ref_id(name='name', raw_id='raw_id') | ||||
# Parsing such a reference back as it is used by controllers | ||||
result = utils.parse_path_ref(formatted_ref_id) | ||||
assert list(result) == expected | ||||
@pytest.mark.parametrize('ref, default_path, expected', [ | ||||
('a', None, (None, 'a')), | ||||
('a', 'path', ('path', 'a')), | ||||
('p@a', 'path', ('p', 'a')), | ||||
('p@a', None, ('p', 'a')), | ||||
]) | ||||
def test_parse_path_ref(ref, default_path, expected): | ||||
result = utils.parse_path_ref(ref, default_path) | ||||
assert list(result) == list(expected) | ||||
@pytest.mark.parametrize('alias, expected', [ | ||||
('hg', 'name'), | ||||
('git', 'name'), | ||||
('svn', 'name@raw_id'), | ||||
]) | ||||
def test_format_ref_id(alias, expected): | ||||
repo = mock.Mock(alias=alias) | ||||
format_ref_id = utils.get_format_ref_id(repo) | ||||
result = format_ref_id(name='name', raw_id='raw_id') | ||||
assert result == expected | ||||
class TestGetCommit(object): | ||||
@pytest.mark.parametrize('ref_type', [None, 'book', 'tag', 'branch']) | ||||
def test_get_commit_from_ref_name_found(self, ref_type): | ||||
ref_name = 'a_None_id' | ||||
repo = mock.Mock() | ||||
scm_instance = repo.scm_instance() | ||||
scm_instance.branches = {ref_name: 'a_branch_id'} | ||||
scm_instance.tags = {ref_name: 'a_tag_id'} | ||||
scm_instance.bookmarks = {ref_name: 'a_book_id'} | ||||
scm_instance.get_commit.return_value = 'test' | ||||
commit = utils.get_commit_from_ref_name(repo, ref_name, ref_type) | ||||
scm_instance.get_commit.assert_called_once_with('a_%s_id' % ref_type) | ||||
assert commit == 'test' | ||||
@pytest.mark.parametrize('ref_type', ['book', 'tag', 'branch']) | ||||
def test_get_commit_from_ref_name_not_found(self, ref_type): | ||||
ref_name = 'invalid_ref' | ||||
repo = mock.Mock() | ||||
scm_instance = repo.scm_instance() | ||||
repo.scm_instance().branches = {} | ||||
repo.scm_instance().tags = {} | ||||
repo.scm_instance().bookmarks = {} | ||||
with pytest.raises(RepositoryError): | ||||
utils.get_commit_from_ref_name(repo, ref_name, ref_type) | ||||