pagure/tests/test_pagure_exclude_group_index.py

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

"""
 (c) 2017 - Copyright Red Hat Inc

 Authors:
   Pierre-Yves Chibon <pingou@pingoured.fr>

"""

from __future__ import unicode_literals, absolute_import

import unittest
import shutil
import sys
import os

import mock

sys.path.insert(0, os.path.join(os.path.dirname(
    os.path.abspath(__file__)), '..'))

import pagure.lib.query
import pagure.lib.model
import tests

class PagureExcludeGroupIndex(tests.Modeltests):
    """ Tests the EXCLUDE_GROUP_INDEX configuration key in pagure """

    def setUp(self):
        """ Set up the environnment, ran before every tests. """
        super(PagureExcludeGroupIndex, self).setUp()

        tests.create_projects(self.session)
        tests.create_projects_git(os.path.join(self.path, 'repos'), bare=True)

        # Create a ``provenpackger`` group:
        msg = pagure.lib.query.add_group(
            self.session,
            group_name='provenpackager',
            display_name='Proven Packagers',
            description='Packagers having access to all the repo',
            group_type='user',
            user='pingou',
            is_admin=False,
            blacklist=[],
        )
        self.session.commit()
        self.assertEqual(
            msg, 'User `pingou` added to the group `provenpackager`.')

        # Add the `provenpackager` group to the test2 project
        project = pagure.lib.query._get_project(self.session, 'test2')
        msg = pagure.lib.query.add_group_to_project(
            session=self.session,
            project=project,
            new_group='provenpackager',
            user='pingou',
        )
        self.session.commit()
        self.assertEqual(msg, 'Group added')

    def test_defaults_pingou(self):
        """ Test which repo pingou has by default. """

        repos = pagure.lib.query.search_projects(
            self.session,
            username='pingou',
            fork=False,
        )

        self.assertEqual(len(repos), 3)
        for idx, name in enumerate(['test', 'test2', 'test3']):
            self.assertEqual(repos[idx].name, name)

    def test_defaults_foo(self):
        """ Test which repo foo has by default. """

        repos = pagure.lib.query.search_projects(
            self.session,
            username='foo',
            fork=False,
        )

        self.assertEqual(len(repos), 0)


    def test_add_foo_test(self):
        """ Test adding foo to the test project. """

        group = pagure.lib.query.search_groups(
            self.session, group_name='provenpackager')
        self.assertEqual(group.group_name, 'provenpackager')

        # List all foo's project before (ie: there should be none)
        repos = pagure.lib.query.search_projects(
            self.session,
            username='foo',
            fork=False,
        )

        self.assertEqual(len(repos), 0)

        # Adding `foo` to the `provenpackager` group
        msg = pagure.lib.query.add_user_to_group(
            self.session,
            username='foo',
            group=group,
            user='pingou',
            is_admin=False,
        )
        self.assertEqual(
            msg, 'User `foo` added to the group `provenpackager`.')

        # Test that foo has now one project, via the provenpackager group
        repos = pagure.lib.query.search_projects(
            self.session,
            username='foo',
            fork=False,
        )

        self.assertEqual(len(repos), 1)
        self.assertEqual(repos[0].name, 'test2')

    def test_excluding_provenpackager(self):
        """ Test retrieving user's repo with a group excluded. """

        # Add `foo` to `provenpackager`
        group = pagure.lib.query.search_groups(
            self.session, group_name='provenpackager')
        self.assertEqual(group.group_name, 'provenpackager')

        msg = pagure.lib.query.add_user_to_group(
            self.session,
            username='foo',
            group=group,
            user='pingou',
            is_admin=False,
        )
        self.assertEqual(
            msg, 'User `foo` added to the group `provenpackager`.')

        # Get foo's project outside of proven packager
        repos = pagure.lib.query.search_projects(
            self.session,
            username='foo',
            exclude_groups=['provenpackager'],
            fork=False,
        )

        self.assertEqual(len(repos), 0)

        # Get pingou's project outside of proven packager (nothing changes)
        repos = pagure.lib.query.search_projects(
            self.session,
            username='pingou',
            exclude_groups=['provenpackager'],
            fork=False,
        )
        repos2 = pagure.lib.query.search_projects(
            self.session,
            username='pingou',
            fork=False,
        )

        self.assertEqual(repos, repos2)
        self.assertEqual(len(repos), 3)
        for idx, name in enumerate(['test', 'test2', 'test3']):
            self.assertEqual(repos[idx].name, name)


if __name__ == '__main__':
    unittest.main(verbosity=2)