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)