Show More
test_simplehg.py
125 lines
| 5.0 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 urlparse | ||||
import mock | ||||
import pytest | ||||
import simplejson as json | ||||
from rhodecode.lib.vcs.backends.base import Config | ||||
from rhodecode.tests.lib.middleware import mock_scm_app | ||||
import rhodecode.lib.middleware.simplehg as simplehg | ||||
def get_environ(url): | ||||
"""Construct a minimum WSGI environ based on the URL.""" | ||||
parsed_url = urlparse.urlparse(url) | ||||
environ = { | ||||
'PATH_INFO': parsed_url.path, | ||||
'QUERY_STRING': parsed_url.query, | ||||
} | ||||
return environ | ||||
@pytest.mark.parametrize( | ||||
'url, expected_action', | ||||
[ | ||||
('/foo/bar?cmd=unbundle&key=tip', 'push'), | ||||
('/foo/bar?cmd=pushkey&key=tip', 'push'), | ||||
('/foo/bar?cmd=listkeys&key=tip', 'pull'), | ||||
('/foo/bar?cmd=changegroup&key=tip', 'pull'), | ||||
# Edge case: unknown argument: assume pull | ||||
('/foo/bar?cmd=unknown&key=tip', 'pull'), | ||||
('/foo/bar?cmd=&key=tip', 'pull'), | ||||
# Edge case: not cmd argument | ||||
('/foo/bar?key=tip', 'pull'), | ||||
]) | ||||
r2140 | def test_get_action(url, expected_action, request_stub): | |||
r2358 | app = simplehg.SimpleHg(config={'auth_ret_code': '', 'base_path': ''}, | |||
r2140 | registry=request_stub.registry) | |||
r1 | assert expected_action == app._get_action(get_environ(url)) | |||
@pytest.mark.parametrize( | ||||
'url, expected_repo_name', | ||||
[ | ||||
('/foo?cmd=unbundle&key=tip', 'foo'), | ||||
('/foo/bar?cmd=pushkey&key=tip', 'foo/bar'), | ||||
('/foo/bar/baz?cmd=listkeys&key=tip', 'foo/bar/baz'), | ||||
# Repos with trailing slashes. | ||||
('/foo/?cmd=unbundle&key=tip', 'foo'), | ||||
('/foo/bar/?cmd=pushkey&key=tip', 'foo/bar'), | ||||
('/foo/bar/baz/?cmd=listkeys&key=tip', 'foo/bar/baz'), | ||||
]) | ||||
r2140 | def test_get_repository_name(url, expected_repo_name, request_stub): | |||
r2358 | app = simplehg.SimpleHg(config={'auth_ret_code': '', 'base_path': ''}, | |||
r2140 | registry=request_stub.registry) | |||
r1 | assert expected_repo_name == app._get_repository_name(get_environ(url)) | |||
r2351 | def test_get_config(user_util, baseapp, request_stub): | |||
r1570 | repo = user_util.create_repo(repo_type='git') | |||
r2358 | app = simplehg.SimpleHg(config={'auth_ret_code': '', 'base_path': ''}, | |||
r2140 | registry=request_stub.registry) | |||
r2112 | extras = [('foo', 'FOO', 'bar', 'BAR')] | |||
r1 | ||||
r1570 | hg_config = app._create_config(extras, repo_name=repo.repo_name) | |||
config = simplehg.utils.make_db_config(repo=repo.repo_name) | ||||
config.set('rhodecode', 'RC_SCM_DATA', json.dumps(extras)) | ||||
hg_config_org = config | ||||
r1 | ||||
r1570 | expected_config = [ | |||
('vcs_svn_tag', 'ff89f8c714d135d865f44b90e5413b88de19a55f', '/tags/*'), | ||||
('web', 'push_ssl', 'False'), | ||||
('web', 'allow_push', '*'), | ||||
('web', 'allow_archive', 'gz zip bz2'), | ||||
('web', 'baseurl', '/'), | ||||
('vcs_git_lfs', 'store_location', hg_config_org.get('vcs_git_lfs', 'store_location')), | ||||
('vcs_svn_branch', '9aac1a38c3b8a0cdc4ae0f960a5f83332bc4fa5e', '/branches/*'), | ||||
('vcs_svn_branch', 'c7e6a611c87da06529fd0dd733308481d67c71a8', '/trunk'), | ||||
('largefiles', 'usercache', hg_config_org.get('largefiles', 'usercache')), | ||||
('hooks', 'preoutgoing.pre_pull', 'python:vcsserver.hooks.pre_pull'), | ||||
('hooks', 'prechangegroup.pre_push', 'python:vcsserver.hooks.pre_push'), | ||||
('hooks', 'outgoing.pull_logger', 'python:vcsserver.hooks.log_pull_action'), | ||||
('hooks', 'pretxnchangegroup.pre_push', 'python:vcsserver.hooks.pre_push'), | ||||
('hooks', 'changegroup.push_logger', 'python:vcsserver.hooks.log_push_action'), | ||||
('hooks', 'changegroup.repo_size', 'python:vcsserver.hooks.repo_size'), | ||||
('phases', 'publish', 'True'), | ||||
('extensions', 'largefiles', ''), | ||||
('paths', '/', hg_config_org.get('paths', '/')), | ||||
r2112 | ('rhodecode', 'RC_SCM_DATA', '[["foo", "FOO", "bar", "BAR"]]') | |||
r1570 | ] | |||
for entry in expected_config: | ||||
assert entry in hg_config | ||||
r1 | ||||
r2140 | def test_create_wsgi_app_uses_scm_app_from_simplevcs(request_stub): | |||
r1 | config = { | |||
'auth_ret_code': '', | ||||
'base_path': '', | ||||
'vcs.scm_app_implementation': | ||||
'rhodecode.tests.lib.middleware.mock_scm_app', | ||||
} | ||||
r2358 | app = simplehg.SimpleHg(config=config, registry=request_stub.registry) | |||
r1 | wsgi_app = app._create_wsgi_app('/tmp/test', 'test_repo', {}) | |||
assert wsgi_app is mock_scm_app.mock_hg_wsgi | ||||