# -*- coding: utf-8 -*- """ (c) 2018 - Copyright Red Hat Inc Authors: Patrick Uiterwijk """ 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)