rhodecode/apps/repository/tests/test_repo_settings.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 mock
import pytest

from rhodecode.lib.utils2 import str2bool
from rhodecode.lib.vcs.exceptions import RepositoryRequirementError
from rhodecode.model.db import Repository, UserRepoToPerm, Permission, User
from rhodecode.model.meta import Session
from rhodecode.tests import (
    url, HG_REPO, TEST_USER_ADMIN_LOGIN, TEST_USER_REGULAR_LOGIN,
    assert_session_flash)
from rhodecode.tests.fixture import Fixture

fixture = Fixture()


def route_path(name, params=None, **kwargs):
    import urllib

    base_url = {
        'edit_repo': '/{repo_name}/settings',
        'edit_repo_caches': '/{repo_name}/settings/caches',
        'edit_repo_perms': '/{repo_name}/settings/permissions',
    }[name].format(**kwargs)

    if params:
        base_url = '{}?{}'.format(base_url, urllib.urlencode(params))
    return base_url


def _get_permission_for_user(user, repo):
    perm = UserRepoToPerm.query()\
        .filter(UserRepoToPerm.repository ==
                Repository.get_by_repo_name(repo))\
        .filter(UserRepoToPerm.user == User.get_by_username(user))\
        .all()
    return perm


@pytest.mark.usefixtures('autologin_user', 'app')
class TestAdminRepoSettings(object):
    @pytest.mark.parametrize('urlname', [
        'edit_repo',
        'edit_repo_caches',
        'edit_repo_perms',
    ])
    def test_show_page(self, urlname, app, backend):
        app.get(route_path(urlname, repo_name=backend.repo_name), status=200)

    def test_edit_accessible_when_missing_requirements(
            self, backend_hg, autologin_user):
        scm_patcher = mock.patch.object(
            Repository, 'scm_instance', side_effect=RepositoryRequirementError)
        with scm_patcher:
            self.app.get(route_path('edit_repo', repo_name=backend_hg.repo_name))

    @pytest.mark.parametrize('urlname', [
        'edit_repo_advanced',
        'repo_vcs_settings',
        'edit_repo_fields',
        'repo_settings_issuetracker',
        'edit_repo_remote',
        'edit_repo_statistics',
    ])
    def test_show_page_pylons(self, urlname, app):
        app.get(url(urlname, repo_name=HG_REPO))

    @pytest.mark.parametrize('update_settings', [
        {'repo_description': 'alter-desc'},
        {'repo_owner': TEST_USER_REGULAR_LOGIN},
        {'repo_private': 'true'},
        {'repo_enable_locking': 'true'},
        {'repo_enable_downloads': 'true'},
    ])
    def test_update_repo_settings(self, update_settings, csrf_token, backend, user_util):
        repo = user_util.create_repo(repo_type=backend.alias)
        repo_name = repo.repo_name

        params = fixture._get_repo_create_params(
                csrf_token=csrf_token,
                repo_name=repo_name,
                repo_type=backend.alias,
                repo_owner=TEST_USER_ADMIN_LOGIN,
                repo_description='DESC',

                repo_private='false',
                repo_enable_locking='false',
                repo_enable_downloads='false')
        params.update(update_settings)
        self.app.post(
            route_path('edit_repo', repo_name=repo_name),
            params=params, status=302)

        repo = Repository.get_by_repo_name(repo_name)
        assert repo.user.username == \
               update_settings.get('repo_owner', repo.user.username)

        assert repo.description == \
               update_settings.get('repo_description', repo.description)

        assert repo.private == \
               str2bool(update_settings.get(
                   'repo_private', repo.private))

        assert repo.enable_locking == \
               str2bool(update_settings.get(
                   'repo_enable_locking', repo.enable_locking))

        assert repo.enable_downloads == \
               str2bool(update_settings.get(
                   'repo_enable_downloads', repo.enable_downloads))

    def test_update_repo_name_via_settings(self, csrf_token, user_util, backend):
        repo = user_util.create_repo(repo_type=backend.alias)
        repo_name = repo.repo_name

        repo_group = user_util.create_repo_group()
        repo_group_name = repo_group.group_name
        new_name = repo_group_name + '_' + repo_name

        params = fixture._get_repo_create_params(
                csrf_token=csrf_token,
                repo_name=new_name,
                repo_type=backend.alias,
                repo_owner=TEST_USER_ADMIN_LOGIN,
                repo_description='DESC',
                repo_private='false',
                repo_enable_locking='false',
                repo_enable_downloads='false')
        self.app.post(
            route_path('edit_repo', repo_name=repo_name),
            params=params, status=302)
        repo = Repository.get_by_repo_name(new_name)
        assert repo.repo_name == new_name

    def test_update_repo_group_via_settings(self, csrf_token, user_util, backend):
        repo = user_util.create_repo(repo_type=backend.alias)
        repo_name = repo.repo_name

        repo_group = user_util.create_repo_group()
        repo_group_name = repo_group.group_name
        repo_group_id = repo_group.group_id

        new_name = repo_group_name + '/' + repo_name
        params = fixture._get_repo_create_params(
                csrf_token=csrf_token,
                repo_name=repo_name,
                repo_type=backend.alias,
                repo_owner=TEST_USER_ADMIN_LOGIN,
                repo_description='DESC',
                repo_group=repo_group_id,
                repo_private='false',
                repo_enable_locking='false',
                repo_enable_downloads='false')
        self.app.post(
            route_path('edit_repo', repo_name=repo_name),
            params=params, status=302)
        repo = Repository.get_by_repo_name(new_name)
        assert repo.repo_name == new_name

    def test_set_private_flag_sets_default_user_permissions_to_none(
            self, autologin_user, backend, csrf_token):

        # initially repository perm should be read
        perm = _get_permission_for_user(user='default', repo=backend.repo_name)
        assert len(perm) == 1
        assert perm[0].permission.permission_name == 'repository.read'
        assert not backend.repo.private

        response = self.app.post(
            route_path('edit_repo', repo_name=backend.repo_name),
            params=fixture._get_repo_create_params(
                repo_private='true',
                repo_name=backend.repo_name,
                repo_type=backend.alias,
                repo_owner=TEST_USER_ADMIN_LOGIN,
                csrf_token=csrf_token), status=302)

        assert_session_flash(
            response,
            msg='Repository %s updated successfully' % (backend.repo_name))

        repo = Repository.get_by_repo_name(backend.repo_name)
        assert repo.private is True

        # now the repo default permission should be None
        perm = _get_permission_for_user(user='default', repo=backend.repo_name)
        assert len(perm) == 1
        assert perm[0].permission.permission_name == 'repository.none'

        response = self.app.post(
            route_path('edit_repo', repo_name=backend.repo_name),
            params=fixture._get_repo_create_params(
                repo_private='false',
                repo_name=backend.repo_name,
                repo_type=backend.alias,
                repo_owner=TEST_USER_ADMIN_LOGIN,
                csrf_token=csrf_token), status=302)

        assert_session_flash(
            response,
            msg='Repository %s updated successfully' % (backend.repo_name))
        assert backend.repo.private is False

        # we turn off private now the repo default permission should stay None
        perm = _get_permission_for_user(user='default', repo=backend.repo_name)
        assert len(perm) == 1
        assert perm[0].permission.permission_name == 'repository.none'

        # update this permission back
        perm[0].permission = Permission.get_by_key('repository.read')
        Session().add(perm[0])
        Session().commit()