rhodecode/tests/lib/middleware/test_simplehg.py

# -*- coding: utf-8 -*-

# Copyright (C) 2010-2017 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/

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'),
    ])
def test_get_action(url, expected_action):
    app = simplehg.SimpleHg(application=None,
                            config={'auth_ret_code': '', 'base_path': ''},
                            registry=None)
    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'),
    ])
def test_get_repository_name(url, expected_repo_name):
    app = simplehg.SimpleHg(application=None,
                            config={'auth_ret_code': '', 'base_path': ''},
                            registry=None)
    assert expected_repo_name == app._get_repository_name(get_environ(url))


def test_get_config(pylonsapp, user_util):
    repo = user_util.create_repo(repo_type='git')
    app = simplehg.SimpleHg(application=None,
                            config={'auth_ret_code': '', 'base_path': ''},
                            registry=None)
    extras = {'foo': 'FOO', 'bar': 'BAR'}

    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

    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', '/')),
        ('rhodecode', 'RC_SCM_DATA', '{"foo": "FOO", "bar": "BAR"}')
    ]
    for entry in expected_config:
        assert entry in hg_config


def test_create_wsgi_app_uses_scm_app_from_simplevcs():
    config = {
        'auth_ret_code': '',
        'base_path': '',
        'vcs.scm_app_implementation':
            'rhodecode.tests.lib.middleware.mock_scm_app',
    }
    app = simplehg.SimpleHg(application=None, config=config, registry=None)
    wsgi_app = app._create_wsgi_app('/tmp/test', 'test_repo', {})
    assert wsgi_app is mock_scm_app.mock_hg_wsgi