# -*- 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)