pagure/tests/test_pagure_utils.py

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

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

 Authors:
   Patrick Uiterwijk <puiterwijk@redhat.com>

"""

from __future__ import unicode_literals, absolute_import

import unittest
import shutil
import sys
import os

import mock
import munch
import pygit2
import werkzeug

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

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


class PagureUtilsTests(tests.SimplePagureTest):
    """ Tests for pagure.utils """

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

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

        project = pagure.lib.query._get_project(self.session, 'test')
        # Add a deploy key to the project
        new_key_obj = pagure.lib.model.SSHKey(
            project_id=project.id,
            pushaccess=False,
            public_ssh_key='\n foo bar',
            ssh_short_key='\n foo bar',
            ssh_search_key='\n foo bar',
            creator_user_id=1  # pingou
        )

        self.session.add(new_key_obj)
        self.session.commit()

    def test_lookup_deploykey_non_deploykey(self):
        """ Test lookup_deploykey with a non-deploykey username. """
        project = pagure.lib.query._get_project(self.session, 'test')
        res = pagure.utils.lookup_deploykey(project, 'pingou')
        self.assertEquals(res, None)

    def test_lookup_deploykey_different_project(self):
        """ Test lookup_deploykey with a username for another project. """
        project = pagure.lib.query._get_project(self.session, 'test2')
        res = pagure.utils.lookup_deploykey(project, 'deploykey_test_1')
        self.assertEquals(res, None)

    def test_lookup_deploykey_non_existent_key(self):
        """ Test lookup_deploykey with a non-existing deploykey. """
        project = pagure.lib.query._get_project(self.session, 'test')
        res = pagure.utils.lookup_deploykey(project, 'deploykey_test_2')
        self.assertEquals(res, None)

    def test_lookup_deploykey(self):
        """ Test lookup_deploykey with a correct username. """
        project = pagure.lib.query._get_project(self.session, 'test')
        res = pagure.utils.lookup_deploykey(project, 'deploykey_test_1')
        self.assertNotEquals(res, None)
        self.assertFalse(res.pushaccess)