rhodecode/tests/lib/middleware/test_simplegit.py

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

# Copyright (C) 2010-2016  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 pytest
import urlparse

from rhodecode.tests.lib.middleware import mock_scm_app
import rhodecode.lib.middleware.simplegit as simplegit


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/info/refs?service=git-upload-pack', 'pull'),
        ('/foo/bar/info/refs?service=git-receive-pack', 'push'),
        ('/foo/bar/git-upload-pack', 'pull'),
        ('/foo/bar/git-receive-pack', 'push'),
        # Edge case: missing data for info/refs
        ('/foo/info/refs?service=', 'pull'),
        ('/foo/info/refs', 'pull'),
        # Edge case: git command comes with service argument
        ('/foo/git-upload-pack?service=git-receive-pack', 'pull'),
        ('/foo/git-receive-pack?service=git-upload-pack', 'push'),
        # Edge case: repo name conflicts with git commands
        ('/git-receive-pack/git-upload-pack', 'pull'),
        ('/git-receive-pack/git-receive-pack', 'push'),
        ('/git-upload-pack/git-upload-pack', 'pull'),
        ('/git-upload-pack/git-receive-pack', 'push'),
        ('/foo/git-receive-pack', 'push'),
        # Edge case: not a smart protocol url
        ('/foo/bar', 'pull'),
    ])
def test_get_action(url, expected_action, pylonsapp):
    app = simplegit.SimpleGit(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/info/refs?service=git-upload-pack', 'foo'),
        ('/foo/bar/info/refs?service=git-receive-pack', 'foo/bar'),
        ('/foo/git-upload-pack', 'foo'),
        ('/foo/git-receive-pack', 'foo'),
        ('/foo/bar/git-upload-pack', 'foo/bar'),
        ('/foo/bar/git-receive-pack', 'foo/bar'),
    ])
def test_get_repository_name(url, expected_repo_name, pylonsapp):
    app = simplegit.SimpleGit(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):
    app = simplegit.SimpleGit(application=None,
                              config={'auth_ret_code': '', 'base_path': ''},
                              registry=None)
    extras = {'foo': 'FOO', 'bar': 'BAR'}

    # We copy the extras as the method below will change the contents.
    config = app._create_config(dict(extras), repo_name='test-repo')
    expected_config = dict(extras)
    expected_config.update({
        'git_update_server_info': False,
    })

    assert config == expected_config


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