# -*- coding: utf-8 -*- """ (c) 2015-2017 - Copyright Red Hat Inc Authors: Pierre-Yves Chibon """ __requires__ = ['SQLAlchemy >= 0.8'] import pkg_resources import datetime import json import unittest import shutil import sys import time import os import pygit2 from mock import patch sys.path.insert(0, os.path.join(os.path.dirname( os.path.abspath(__file__)), '..')) import pagure import pagure.lib import tests from pagure.lib.repo import PagureRepo class PagureFlaskInternaltests(tests.Modeltests): """ Tests for flask Internal controller of pagure """ def setUp(self): """ Set up the environnment, ran before every tests. """ super(PagureFlaskInternaltests, self).setUp() pagure.config.config['IP_ALLOWED_INTERNAL'] = list(set( pagure.config.config['IP_ALLOWED_INTERNAL'] + [None])) pagure.config.config['GIT_FOLDER'] = os.path.join( self.path, 'repos') pagure.config.config['REQUESTS_FOLDER'] = None pagure.config.config['TICKETS_FOLDER'] = None pagure.config.config['DOCS_FOLDER'] = None @patch('pagure.lib.notify.send_email') def test_pull_request_add_comment(self, send_email): """ Test the pull_request_add_comment function. """ send_email.return_value = True tests.create_projects(self.session) repo = pagure.lib.get_authorized_project(self.session, 'test') req = pagure.lib.new_pull_request( session=self.session, repo_from=repo, branch_from='feature', repo_to=repo, branch_to='master', title='PR from the feature branch', user='pingou', requestfolder=None, ) self.session.commit() self.assertEqual(req.id, 1) self.assertEqual(req.title, 'PR from the feature branch') request = repo.requests[0] self.assertEqual(len(request.comments), 0) self.assertEqual(len(request.discussion), 0) data = { 'objid': 'foo', } # Wrong http request output = self.app.post('/pv/pull-request/comment/', data=data) self.assertEqual(output.status_code, 405) # Invalid request output = self.app.put('/pv/pull-request/comment/', data=data) self.assertEqual(output.status_code, 400) data = { 'objid': 'foo', 'useremail': 'foo@pingou.com', } # Invalid objid output = self.app.put('/pv/pull-request/comment/', data=data) self.assertEqual(output.status_code, 404) data = { 'objid': request.uid, 'useremail': 'foo@pingou.com', } # Valid objid, in-complete data for a comment output = self.app.put('/pv/pull-request/comment/', data=data) self.assertEqual(output.status_code, 400) data = { 'objid': request.uid, 'useremail': 'foo@pingou.com', 'comment': 'Looks good to me!', } # Add comment output = self.app.put('/pv/pull-request/comment/', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data) self.assertDictEqual(js_data, {'message': 'Comment added'}) self.session.commit() repo = pagure.lib.get_authorized_project(self.session, 'test') request = repo.requests[0] self.assertEqual(len(request.comments), 1) self.assertEqual(len(request.discussion), 1) # Check the @localonly before = pagure.config.config['IP_ALLOWED_INTERNAL'][:] pagure.config.config['IP_ALLOWED_INTERNAL'] = [] output = self.app.put('/pv/pull-request/comment/', data=data) self.assertEqual(output.status_code, 403) pagure.config.config['IP_ALLOWED_INTERNAL'] = before[:] @patch('pagure.lib.notify.send_email') def test_ticket_add_comment(self, send_email): """ Test the ticket_add_comment function. """ send_email.return_value = True tests.create_projects(self.session) # Create issues to play with repo = pagure.lib.get_authorized_project(self.session, 'test') msg = pagure.lib.new_issue( session=self.session, repo=repo, title='Test issue', content='We should work on this', user='pingou', ticketfolder=None ) self.session.commit() self.assertEqual(msg.title, 'Test issue') issue = repo.issues[0] self.assertEqual(len(issue.comments), 0) data = { 'objid': 'foo', } # Wrong http request output = self.app.post('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 405) # Invalid request output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 400) data = { 'objid': 'foo', 'useremail': 'foo@pingou.com', } # Invalid objid output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 404) data = { 'objid': issue.uid, 'useremail': 'foo@pingou.com', } # Valid objid, in-complete data for a comment output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 400) data = { 'objid': issue.uid, 'useremail': 'foo@pingou.com', 'comment': 'Looks good to me!', } # Add comment output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data) self.assertDictEqual(js_data, {'message': 'Comment added'}) self.session.commit() repo = pagure.lib.get_authorized_project(self.session, 'test') issue = repo.issues[0] self.assertEqual(len(issue.comments), 1) # Check the @localonly pagure.config.config['IP_ALLOWED_INTERNAL'].remove(None) before = pagure.config.config['IP_ALLOWED_INTERNAL'][:] pagure.config.config['IP_ALLOWED_INTERNAL'] = [] output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 403) pagure.config.config['IP_ALLOWED_INTERNAL'] = before[:] @patch('pagure.lib.notify.send_email') def test_private_ticket_add_comment(self, send_email): """ Test the ticket_add_comment function on a private ticket. """ send_email.return_value = True tests.create_projects(self.session) # Create issues to play with repo = pagure.lib.get_authorized_project(self.session, 'test') msg = pagure.lib.new_issue( session=self.session, repo=repo, title='Test issue', content='We should work on this, really', user='pingou', private=True, ticketfolder=None ) self.session.commit() self.assertEqual(msg.title, 'Test issue') issue = repo.issues[0] self.assertEqual(len(issue.comments), 0) data = { 'objid': 'foo', } # Wrong http request output = self.app.post('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 405) # Invalid request output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 400) data = { 'objid': 'foo', 'useremail': 'foo@pingou.com', } # Invalid objid output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 404) data = { 'objid': issue.uid, 'useremail': 'foo@bar.com', } # Valid objid, un-allowed user for this (private) ticket output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 403) data = { 'objid': issue.uid, 'useremail': 'foo@pingou.com', } # Valid objid, un-allowed user for this (private) ticket output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 400) data = { 'objid': issue.uid, 'useremail': 'foo@pingou.com', 'comment': 'Looks good to me!', } # Add comment output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data) self.assertDictEqual(js_data, {'message': 'Comment added'}) self.session.commit() repo = pagure.lib.get_authorized_project(self.session, 'test') issue = repo.issues[0] self.assertEqual(len(issue.comments), 1) # Check the @localonly before = pagure.config.config['IP_ALLOWED_INTERNAL'][:] pagure.config.config['IP_ALLOWED_INTERNAL'] = [] output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 403) pagure.config.config['IP_ALLOWED_INTERNAL'] = before[:] @patch('pagure.lib.notify.send_email') def test_private_ticket_add_comment_acl(self, send_email): """ Test the ticket_add_comment function on a private ticket. """ send_email.return_value = True tests.create_projects(self.session) # Create issues to play with repo = pagure.lib.get_authorized_project(self.session, 'test') msg = pagure.lib.new_issue( session=self.session, repo=repo, title='Test issue', content='We should work on this, really', user='pingou', private=True, ticketfolder=None ) self.session.commit() self.assertEqual(msg.title, 'Test issue') repo = pagure.lib.get_authorized_project(self.session, 'test') issue = repo.issues[0] self.assertEqual(len(issue.comments), 0) # Currently, he is just an average user, # He doesn't have any access in this repo data = { 'objid': issue.uid, 'useremail': 'foo@bar.com', 'comment': 'Looks good to me!', } # Valid objid, un-allowed user for this (private) ticket output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 403) repo = pagure.lib.get_authorized_project(self.session, 'test') # Let's promote him to be a ticketer # He shoudn't be able to comment even then though msg = pagure.lib.add_user_to_project( self.session, project=repo, new_user='foo', user='pingou', access='ticket' ) self.session.commit() self.assertEqual(msg, 'User added') repo = pagure.lib.get_authorized_project(self.session, 'test') self.assertEqual( sorted([u.username for u in repo.users]), ['foo']) self.assertEqual( sorted([u.username for u in repo.committers]), []) self.assertEqual( sorted([u.username for u in repo.admins]), []) output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 403) repo = pagure.lib.get_authorized_project(self.session, 'test') # Let's promote him to be a committer # He should be able to comment msg = pagure.lib.add_user_to_project( self.session, project=repo, new_user='foo', user='pingou', access='commit' ) self.session.commit() self.assertEqual(msg, 'User access updated') repo = pagure.lib.get_authorized_project(self.session, 'test') self.assertEqual( sorted([u.username for u in repo.users]), ['foo']) self.assertEqual( sorted([u.username for u in repo.committers]), ['foo']) self.assertEqual( sorted([u.username for u in repo.admins]), []) # Add comment output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data) self.assertDictEqual(js_data, {'message': 'Comment added'}) repo = pagure.lib.get_authorized_project(self.session, 'test') issue = repo.issues[0] self.assertEqual(len(issue.comments), 1) # Let's promote him to be a admin # He should be able to comment msg = pagure.lib.add_user_to_project( self.session, project=repo, new_user='foo', user='pingou', access='admin' ) self.session.commit() self.assertEqual(msg, 'User access updated') repo = pagure.lib.get_authorized_project(self.session, 'test') self.assertEqual( sorted([u.username for u in repo.users]), ['foo']) self.assertEqual( sorted([u.username for u in repo.committers]), ['foo']) self.assertEqual( sorted([u.username for u in repo.admins]), ['foo']) # Add comment output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data) self.assertDictEqual(js_data, {'message': 'Comment added'}) repo = pagure.lib.get_authorized_project(self.session, 'test') issue = repo.issues[0] self.assertEqual(len(issue.comments), 2) # Check the @localonly before = pagure.config.config['IP_ALLOWED_INTERNAL'][:] pagure.config.config['IP_ALLOWED_INTERNAL'] = [] output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 403) pagure.config.config['IP_ALLOWED_INTERNAL'] = before[:] @patch('pagure.lib.notify.send_email') def test_mergeable_request_pull_FF(self, send_email): """ Test the mergeable_request_pull endpoint with a fast-forward merge. """ send_email.return_value = True # Create a git repo to play with origgitrepo = os.path.join(self.path, 'repos', 'test.git') self.assertFalse(os.path.exists(origgitrepo)) os.makedirs(origgitrepo) orig_repo = pygit2.init_repository(origgitrepo, bare=True) os.makedirs(os.path.join(self.path, 'repos_tmp')) gitrepo = os.path.join(self.path, 'repos_tmp', 'test.git') repo = pygit2.clone_repository(origgitrepo, gitrepo) # Create a file in that git repo with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/master', # the name of the reference to update author, committer, 'Add sources file for testing', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [] ) first_commit = repo.revparse_single('HEAD') refname = 'refs/heads/master:refs/heads/master' ori_remote = repo.remotes[0] PagureRepo.push(ori_remote, refname) # Edit the sources file again with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar\nbaz\n boose') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/feature', # the name of the reference to update author, committer, 'Add baz and boose to the sources\n\n There are more objects to ' 'consider', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [first_commit.oid.hex] ) second_commit = repo.revparse_single('HEAD') refname = 'refs/heads/feature:refs/heads/feature' ori_remote = repo.remotes[0] PagureRepo.push(ori_remote, refname) # Create a PR for these changes tests.create_projects(self.session) project = pagure.lib.get_authorized_project(self.session, 'test') req = pagure.lib.new_pull_request( session=self.session, repo_from=project, branch_from='feature', repo_to=project, branch_to='master', title='PR from the feature branch', user='pingou', requestfolder=None, ) self.session.commit() self.assertEqual(req.id, 1) self.assertEqual(req.title, 'PR from the feature branch') # Check if the PR can be merged data = { 'objid': 'blah', } # Missing CSRF output = self.app.post('/pv/pull-request/merge', data=data) self.assertEqual(output.status_code, 400) user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): output = self.app.get('/test/adduser') csrf_token = output.data.split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] # Missing request identifier data = { 'csrf_token': csrf_token, } output = self.app.post('/pv/pull-request/merge', data=data) self.assertEqual(output.status_code, 404) # With all the desired information project = pagure.lib.get_authorized_project(self.session, 'test') data = { 'csrf_token': csrf_token, 'requestid': project.requests[0].uid, } output = self.app.post('/pv/pull-request/merge', data=data) self.assertEqual(output.status_code, 200) exp = { "code": "FFORWARD", "message": "The pull-request can be merged and fast-forwarded", "short_code": "Ok" } js_data = json.loads(output.data) self.assertDictEqual(js_data, exp) @patch('pagure.lib.notify.send_email') def test_mergeable_request_pull_no_change(self, send_email): """ Test the mergeable_request_pull endpoint when there are no changes to merge. """ send_email.return_value = True # Create a git repo to play with gitrepo = os.path.join(self.path, 'repos', 'test.git') self.assertFalse(os.path.exists(gitrepo)) os.makedirs(gitrepo) repo = pygit2.init_repository(gitrepo) # Create a file in that git repo with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/master', # the name of the reference to update author, committer, 'Add sources file for testing', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [] ) first_commit = repo.revparse_single('HEAD') # Edit the sources file again with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar\nbaz\n boose') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/master', # the name of the reference to update author, committer, 'Add baz and boose to the sources\n\n There are more objects to ' 'consider', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [first_commit.oid.hex] ) second_commit = repo.revparse_single('HEAD') # Create a PR for these changes tests.create_projects(self.session) project = pagure.lib.get_authorized_project(self.session, 'test') req = pagure.lib.new_pull_request( session=self.session, repo_from=project, branch_from='master', repo_to=project, branch_to='master', title='PR from the feature branch', user='pingou', requestfolder=None, ) self.session.commit() self.assertEqual(req.id, 1) self.assertEqual(req.title, 'PR from the feature branch') # Check if the PR can be merged data = { 'objid': 'blah', } # Missing CSRF output = self.app.post('/pv/pull-request/merge', data=data) self.assertEqual(output.status_code, 400) user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): output = self.app.get('/test/adduser') csrf_token = output.data.split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] # Missing request identifier data = { 'csrf_token': csrf_token, } output = self.app.post('/pv/pull-request/merge', data=data) self.assertEqual(output.status_code, 404) # With all the desired information project = pagure.lib.get_authorized_project(self.session, 'test') data = { 'csrf_token': csrf_token, 'requestid': project.requests[0].uid, } output = self.app.post('/pv/pull-request/merge', data=data) self.assertEqual(output.status_code, 200) exp = { "code": "NO_CHANGE", "message": "Nothing to change, git is up to date", "short_code": "No changes" } js_data = json.loads(output.data) self.assertDictEqual(js_data, exp) @patch('pagure.lib.notify.send_email') def test_mergeable_request_pull_merge(self, send_email): """ Test the mergeable_request_pull endpoint when the changes can be merged with a merge commit. """ send_email.return_value = True # Create a git repo to play with origgitrepo = os.path.join(self.path, 'repos', 'test.git') self.assertFalse(os.path.exists(origgitrepo)) os.makedirs(origgitrepo) orig_repo = pygit2.init_repository(origgitrepo, bare=True) os.makedirs(os.path.join(self.path, 'repos_tmp')) gitrepo = os.path.join(self.path, 'repos_tmp', 'test.git') repo = pygit2.clone_repository(origgitrepo, gitrepo) # Create a file in that git repo with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/master', # the name of the reference to update author, committer, 'Add sources file for testing', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [] ) first_commit = repo.revparse_single('HEAD') refname = 'refs/heads/master:refs/heads/master' ori_remote = repo.remotes[0] PagureRepo.push(ori_remote, refname) # Edit the sources file again with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar\nbaz\n boose') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/feature', # the name of the reference to update author, committer, 'Add baz and boose to the sources\n\n There are more objects to ' 'consider', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [first_commit.oid.hex] ) refname = 'refs/heads/feature:refs/heads/feature' ori_remote = repo.remotes[0] PagureRepo.push(ori_remote, refname) # Create another file in the master branch with open(os.path.join(gitrepo, '.gitignore'), 'w') as stream: stream.write('*~') repo.index.add('.gitignore') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/master', # the name of the reference to update author, committer, 'Add .gitignore file for testing', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [first_commit.oid.hex] ) refname = 'refs/heads/master:refs/heads/master' ori_remote = repo.remotes[0] PagureRepo.push(ori_remote, refname) # Create a PR for these changes tests.create_projects(self.session) project = pagure.lib.get_authorized_project(self.session, 'test') req = pagure.lib.new_pull_request( session=self.session, repo_from=project, branch_from='feature', repo_to=project, branch_to='master', title='PR from the feature branch', user='pingou', requestfolder=None, ) self.session.commit() self.assertEqual(req.id, 1) self.assertEqual(req.title, 'PR from the feature branch') # Check if the PR can be merged data = {} # Missing CSRF output = self.app.post('/pv/pull-request/merge', data=data) self.assertEqual(output.status_code, 400) user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): output = self.app.get('/test/adduser') csrf_token = output.data.split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] # Missing request identifier data = { 'csrf_token': csrf_token, } output = self.app.post('/pv/pull-request/merge', data=data) self.assertEqual(output.status_code, 404) # With all the desired information project = pagure.lib.get_authorized_project(self.session, 'test') data = { 'csrf_token': csrf_token, 'requestid': project.requests[0].uid, } output = self.app.post('/pv/pull-request/merge', data=data) self.assertEqual(output.status_code, 200) exp = { "code": "MERGE", "message": "The pull-request can be merged with a merge commit", "short_code": "With merge" } js_data = json.loads(output.data) self.assertDictEqual(js_data, exp) @patch('pagure.lib.notify.send_email') def test_mergeable_request_pull_conflicts(self, send_email): """ Test the mergeable_request_pull endpoint when the changes cannot be merged due to conflicts. """ send_email.return_value = True # Create a git repo to play with origgitrepo = os.path.join(self.path, 'repos', 'test.git') self.assertFalse(os.path.exists(origgitrepo)) os.makedirs(origgitrepo) orig_repo = pygit2.init_repository(origgitrepo, bare=True) os.makedirs(os.path.join(self.path, 'repos_tmp')) gitrepo = os.path.join(self.path, 'repos_tmp', 'test.git') repo = pygit2.clone_repository(origgitrepo, gitrepo) # Create a file in that git repo with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/master', # the name of the reference to update author, committer, 'Add sources file for testing', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [] ) first_commit = repo.revparse_single('HEAD') refname = 'refs/heads/master:refs/heads/master' ori_remote = repo.remotes[0] PagureRepo.push(ori_remote, refname) # Edit the sources file again with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar\nbaz\n boose') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/feature', # the name of the reference to update author, committer, 'Add baz and boose to the sources\n\n There are more objects to ' 'consider', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [first_commit.oid.hex] ) refname = 'refs/heads/feature:refs/heads/feature' ori_remote = repo.remotes[0] PagureRepo.push(ori_remote, refname) # Create another file in the master branch with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar\nbaz\n') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/master', # the name of the reference to update author, committer, 'Add .gitignore file for testing', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [first_commit.oid.hex] ) refname = 'refs/heads/master:refs/heads/master' ori_remote = repo.remotes[0] PagureRepo.push(ori_remote, refname) # Create a PR for these changes tests.create_projects(self.session) project = pagure.lib.get_authorized_project(self.session, 'test') req = pagure.lib.new_pull_request( session=self.session, repo_from=project, branch_from='feature', repo_to=project, branch_to='master', title='PR from the feature branch', user='pingou', requestfolder=None, ) self.session.commit() self.assertEqual(req.id, 1) self.assertEqual(req.title, 'PR from the feature branch') # Check if the PR can be merged data = {} # Missing CSRF output = self.app.post('/pv/pull-request/merge', data=data) self.assertEqual(output.status_code, 400) user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): output = self.app.get('/test/adduser') csrf_token = output.data.split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] # Missing request identifier data = { 'csrf_token': csrf_token, } output = self.app.post('/pv/pull-request/merge', data=data) self.assertEqual(output.status_code, 404) # With all the desired information project = pagure.lib.get_authorized_project(self.session, 'test') data = { 'csrf_token': csrf_token, 'requestid': project.requests[0].uid, } output = self.app.post('/pv/pull-request/merge', data=data) self.assertEqual(output.status_code, 200) exp = { "code": "CONFLICTS", "message": "The pull-request cannot be merged due to conflicts", "short_code": "Conflicts" } js_data = json.loads(output.data) self.assertDictEqual(js_data, exp) def test_get_branches_of_commit(self): ''' Test the get_branches_of_commit from the internal API. ''' tests.create_projects(self.session) tests.create_projects_git(os.path.join(self.path, 'repos')) user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): output = self.app.get('/test/adduser') self.assertEqual(output.status_code, 200) csrf_token = output.data.split( b'name="csrf_token" type="hidden" value="')[1].split(b'">')[0] # No CSRF token data = { 'repo': 'fakerepo', 'commit_id': 'foo', } output = self.app.post('/pv/branches/commit/', data=data) self.assertEqual(output.status_code, 400) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, {u'code': u'ERROR', u'message': u'Invalid input submitted'} ) # Invalid repo data = { 'repo': 'fakerepo', 'commit_id': 'foo', 'csrf_token': csrf_token, } output = self.app.post('/pv/branches/commit/', data=data) self.assertEqual(output.status_code, 404) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, { u'code': u'ERROR', u'message': u'No repo found with the information provided' } ) # Rigth repo, no commit data = { 'repo': 'test', 'csrf_token': csrf_token, } output = self.app.post('/pv/branches/commit/', data=data) self.assertEqual(output.status_code, 400) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, {u'code': u'ERROR', u'message': u'No commit id submitted'} ) # Request is fine, but git repo doesn't exist item = pagure.lib.model.Project( user_id=1, # pingou name='test20', description='test project #20', hook_token='aaabbbhhh', ) self.session.add(item) self.session.commit() data = { 'repo': 'test20', 'commit_id': 'foo', 'csrf_token': csrf_token, } output = self.app.post('/pv/branches/commit/', data=data) self.assertEqual(output.status_code, 404) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, { u'code': u'ERROR', u'message': u'No git repo found with the information provided' } ) # Create a git repo to play with gitrepo = os.path.join(self.path, 'repos', 'test.git') self.assertTrue(os.path.exists(gitrepo)) repo = pygit2.Repository(gitrepo) # Create a file in that git repo with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/master', # the name of the reference to update author, committer, 'Add sources file for testing', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [] ) first_commit = repo.revparse_single('HEAD') # Edit the sources file again with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar\nbaz\n boose') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/feature', # the name of the reference to update author, committer, 'Add baz and boose to the sources\n\n There are more objects to ' 'consider', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [first_commit.oid.hex] ) # Create another file in the master branch with open(os.path.join(gitrepo, '.gitignore'), 'w') as stream: stream.write('*~') repo.index.add('.gitignore') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') commit_hash = repo.create_commit( 'refs/heads/feature_branch', # the name of the reference to update author, committer, 'Add .gitignore file for testing', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [first_commit.oid.hex] ) # All good but the commit id data = { 'repo': 'test', 'commit_id': 'foo', 'csrf_token': csrf_token, } output = self.app.post('/pv/branches/commit/', data=data) self.assertEqual(output.status_code, 404) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, { u'code': u'ERROR', u'message': 'This commit could not be found in this repo' } ) # All good data = { 'repo': 'test', 'commit_id': commit_hash, 'csrf_token': csrf_token, } output = self.app.post('/pv/branches/commit/', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, { u'code': u'OK', u'branches': ['feature_branch'], } ) def test_get_branches_of_commit_with_unrelated_branches(self): ''' Test the get_branches_of_commit from the internal API. ''' tests.create_projects(self.session) tests.create_projects_git(os.path.join(self.path, 'repos')) user = tests.FakeUser(username='pingou') with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() # Create a git repo to play with gitrepo = os.path.join(self.path, 'repos', 'test.git') self.assertTrue(os.path.exists(gitrepo)) repo = pygit2.Repository(gitrepo) # Create a file in that git repo with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/master', # the name of the reference to update author, committer, 'Add sources file for testing', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [] ) first_commit = repo.revparse_single('HEAD') # Edit the sources file again with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar\nbaz\n boose') repo.index.add('sources') repo.index.write() # Commits the files added, but unrelated with the first commit tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') commit = repo.create_commit( 'refs/heads/feature', # the name of the reference to update author, committer, 'Add baz and boose to the sources\n\n There are more objects to ' 'consider', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [] ) commit_hash = commit.hex # All good data = { 'repo': 'test', 'commit_id': commit_hash, 'csrf_token': csrf_token, } output = self.app.post('/pv/branches/commit/', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, { u'code': u'OK', u'branches': ['feature'], } ) def test_get_branches_head(self): ''' Test the get_branches_head from the internal API. ''' tests.create_projects(self.session) tests.create_projects_git(os.path.join(self.path, 'repos')) user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() # No CSRF token data = { 'repo': 'fakerepo', } output = self.app.post('/pv/branches/heads/', data=data) self.assertEqual(output.status_code, 400) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, {u'code': u'ERROR', u'message': u'Invalid input submitted'} ) # Invalid repo data = { 'repo': 'fakerepo', 'commit_id': 'foo', 'csrf_token': csrf_token, } output = self.app.post('/pv/branches/heads/', data=data) self.assertEqual(output.status_code, 404) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, { u'code': u'ERROR', u'message': u'No repo found with the information provided' } ) # Rigth repo, no commit data = { 'repo': 'test', 'csrf_token': csrf_token, } output = self.app.post('/pv/branches/heads/', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, {u"branches": {}, u"code": u"OK", u"heads": {}} ) # Request is fine, but git repo doesn't exist item = pagure.lib.model.Project( user_id=1, # pingou name='test20', description='test project #20', hook_token='aaabbbhhh', ) self.session.add(item) self.session.commit() data = { 'repo': 'test20', 'csrf_token': csrf_token, } output = self.app.post('/pv/branches/heads/', data=data) self.assertEqual(output.status_code, 404) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, { u'code': u'ERROR', u'message': u'No git repo found with the information provided' } ) # Create a git repo to play with gitrepo = os.path.join(self.path, 'repos', 'test.git') self.assertTrue(os.path.exists(gitrepo)) repo = pygit2.Repository(gitrepo) # Create a file in that git repo with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/master', # the name of the reference to update author, committer, 'Add sources file for testing', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [] ) first_commit = repo.revparse_single('HEAD') # Edit the sources file again with open(os.path.join(gitrepo, 'sources'), 'w') as stream: stream.write('foo\n bar\nbaz\n boose') repo.index.add('sources') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') repo.create_commit( 'refs/heads/feature', # the name of the reference to update author, committer, 'Add baz and boose to the sources\n\n There are more objects to ' 'consider', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [first_commit.oid.hex] ) # Create another file in the master branch with open(os.path.join(gitrepo, '.gitignore'), 'w') as stream: stream.write('*~') repo.index.add('.gitignore') repo.index.write() # Commits the files added tree = repo.index.write_tree() author = pygit2.Signature( 'Alice Author', 'alice@authors.tld') committer = pygit2.Signature( 'Cecil Committer', 'cecil@committers.tld') commit_hash = repo.create_commit( 'refs/heads/feature_branch', # the name of the reference to update author, committer, 'Add .gitignore file for testing', # binary string representing the tree object ID tree, # list of binary strings representing parents of the new commit [first_commit.oid.hex] ) # All good data = { 'repo': 'test', 'csrf_token': csrf_token, } output = self.app.post('/pv/branches/heads/', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) # We can't test the content since the commit hash will change all # the time, so let's just check the structure self.assertEqual( sorted(js_data.keys()), ['branches', 'code', 'heads']) self.assertEqual(js_data['code'], 'OK') self.assertEqual(len(js_data['heads']), 3) self.assertEqual(len(js_data['branches']), 3) def test_get_stats_commits_no_token(self): ''' Test the get_stats_commits from the internal API. ''' # No CSRF token data = { 'repo': 'fakerepo', } output = self.app.post('/pv/stats/commits/authors', data=data) self.assertEqual(output.status_code, 400) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, {u'code': u'ERROR', u'message': u'Invalid input submitted'} ) def test_get_stats_commits_invalid_repo(self): ''' Test the get_stats_commits from the internal API. ''' user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() # Invalid repo data = { 'repo': 'fakerepo', 'csrf_token': csrf_token, } output = self.app.post('/pv/stats/commits/authors', data=data) self.assertEqual(output.status_code, 404) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, {u'code': u'ERROR', u'message': u'No repo found with the information provided'} ) def test_get_stats_commits_empty_git(self): ''' Test the get_stats_commits from the internal API. ''' tests.create_projects(self.session) tests.create_projects_git(os.path.join(self.path, 'repos')) user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() # No content in git data = { 'repo': 'test', 'csrf_token': csrf_token, } output = self.app.post('/pv/stats/commits/authors', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) self.assertEqual( sorted(js_data.keys()), ['code', 'message', 'task_id', 'url'] ) self.assertEqual(js_data['code'], 'OK') self.assertEqual(js_data['message'], 'Stats asked') self.assertTrue(js_data['url'].startswith('/pv/task/')) output = self.app.get(js_data['url']) js_data2 = json.loads(output.data.decode('utf-8')) self.assertTrue( js_data2 in [ {u'results': u"reference 'refs/heads/master' not found"}, {u'results': u"Reference 'refs/heads/master' not found"} ] ) def test_get_stats_commits_git_populated(self): ''' Test the get_stats_commits from the internal API. ''' 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', 'test.git')) user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() # Content in git data = { 'repo': 'test', 'csrf_token': csrf_token, } output = self.app.post('/pv/stats/commits/authors', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) self.assertEqual( sorted(js_data.keys()), ['code', 'message', 'task_id', 'url'] ) self.assertEqual(js_data['code'], 'OK') self.assertEqual(js_data['message'], 'Stats asked') self.assertTrue(js_data['url'].startswith('/pv/task/')) output = self.app.get(js_data['url']) while output.status_code == 418: time.sleep(0.5) output = self.app.get(js_data['url']) js_data2 = json.loads(output.data.decode('utf-8')) self.assertTrue(js_data2['results'][3] > 1509110062) js_data2['results'][3] = 1509110062 self.assertDictEqual( js_data2, {u'results': [ 2, [[2, [[u'Alice Author', u'alice@authors.tld']]]], 1, 1509110062 ] } ) def test_get_stats_commits_trend_no_token(self): ''' Test the get_stats_commits_trend from the internal API. ''' # No CSRF token data = { 'repo': 'fakerepo', } output = self.app.post('/pv/stats/commits/trend', data=data) self.assertEqual(output.status_code, 400) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, {u'code': u'ERROR', u'message': u'Invalid input submitted'} ) def test_get_stats_commits_trend_invalid_repo(self): """ Test the get_stats_commits_trend from the internal API. """ user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() # Invalid repo data = { 'repo': 'fakerepo', 'csrf_token': csrf_token, } output = self.app.post('/pv/stats/commits/trend', data=data) self.assertEqual(output.status_code, 404) js_data = json.loads(output.data.decode('utf-8')) self.assertDictEqual( js_data, {u'code': u'ERROR', u'message': u'No repo found with the information provided'} ) def test_get_stats_commits_trend_empty_git(self): ''' Test the get_stats_commits_trend from the internal API. ''' tests.create_projects(self.session) tests.create_projects_git(os.path.join(self.path, 'repos')) user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() # No content in git data = { 'repo': 'test', 'csrf_token': csrf_token, } output = self.app.post('/pv/stats/commits/trend', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) self.assertEqual( sorted(js_data.keys()), ['code', 'message', 'task_id', 'url'] ) self.assertEqual(js_data['code'], 'OK') self.assertEqual(js_data['message'], 'Stats asked') self.assertTrue(js_data['url'].startswith('/pv/task/')) output = self.app.get(js_data['url']) js_data2 = json.loads(output.data.decode('utf-8')) self.assertTrue( js_data2 in [ {u'results': u"reference 'refs/heads/master' not found"}, {u'results': u"Reference 'refs/heads/master' not found"} ] ) def test_get_stats_commits_trend_git_populated(self): ''' Test the get_stats_commits_trend from the internal API. ''' 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', 'test.git')) user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() # Content in git data = { 'repo': 'test', 'csrf_token': csrf_token, } output = self.app.post('/pv/stats/commits/trend', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) self.assertEqual( sorted(js_data.keys()), ['code', 'message', 'task_id', 'url'] ) self.assertEqual(js_data['code'], 'OK') self.assertEqual(js_data['message'], 'Stats asked') self.assertTrue(js_data['url'].startswith('/pv/task/')) output = self.app.get(js_data['url']) js_data2 = json.loads(output.data.decode('utf-8')) today = datetime.datetime.utcnow().date() self.assertDictEqual( js_data2, {u'results': [[str(today), 2]]} ) def test_get_project_family_no_project(self): ''' Test the get_project_family from the internal API. ''' output = self.app.post('/pv/test/family') self.assertEqual(output.status_code, 404) def test_get_project_family_no_csrf(self): ''' Test the get_project_family from the internal API. ''' 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', 'test.git')) output = self.app.post('/pv/test/family') self.assertEqual(output.status_code, 400) js_data = json.loads(output.data.decode('utf-8')) self.assertEqual( sorted(js_data.keys()), [u'code', u'message'] ) self.assertEqual(js_data['code'], u'ERROR') self.assertEqual(js_data['message'], u'Invalid input submitted') def test_get_project_family(self): ''' Test the get_project_family from the internal API. ''' 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', 'test.git')) user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() data = { 'csrf_token': csrf_token, } output = self.app.post('/pv/test/family', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) self.assertEqual( sorted(js_data.keys()), [u'code', u'family'] ) self.assertEqual(js_data['code'], 'OK') self.assertEqual(js_data['family'], [u'test']) def test_get_project_larger_family(self): ''' Test the get_project_family from the internal API. ''' tests.create_projects(self.session) tests.create_projects_git( os.path.join(self.path, 'repos'), bare=True) # Create a 3rd user item = pagure.lib.model.User( user='ralph', fullname='Ralph bar', password='ralph_foo', default_email='ralph@bar.com', ) self.session.add(item) item = pagure.lib.model.UserEmail( user_id=3, email='ralph@bar.com') self.session.add(item) self.session.commit() # Create a couple of forks of the test project item = pagure.lib.model.Project( user_id=2, # foo name='test', is_fork=True, parent_id=1, # test description='test project #1', hook_token='aaabbbcccddd', ) item.close_status = ['Invalid', 'Insufficient data', 'Fixed', 'Duplicate'] self.session.add(item) item = pagure.lib.model.Project( user_id=3, # Ralph name='test', is_fork=True, parent_id=1, # test description='test project #1', hook_token='aaabbbccceee', ) item.close_status = ['Invalid', 'Insufficient data', 'Fixed', 'Duplicate'] self.session.add(item) self.session.commit() # Get on with testing user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() data = { 'csrf_token': csrf_token, } output = self.app.post('/pv/test/family', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) self.assertEqual( sorted(js_data.keys()), [u'code', u'family'] ) self.assertEqual(js_data['code'], 'OK') self.assertEqual( js_data['family'], [u'test', u'fork/foo/test', u'fork/ralph/test']) def test_get_pull_request_ready_branch_main_repo_no_branch(self): '''Test the get_pull_request_ready_branch from the internal API on the main repository ''' tests.create_projects(self.session) tests.create_projects_git( os.path.join(self.path, 'repos'), bare=True) # Get on with testing user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() # Query branches on the main repo data = { 'csrf_token': csrf_token, 'repo': 'test', } output = self.app.post('/pv/pull-request/ready', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) self.assertEqual( sorted(js_data.keys()), [u'code', u'message'] ) self.assertEqual(js_data['code'], 'OK') self.assertEqual( js_data['message'], {u'branch_w_pr': {}, u'new_branch': {}}) def test_get_pull_request_ready_branch_on_fork(self): '''Test the get_pull_request_ready_branch from the internal API on a fork ''' tests.create_projects(self.session) tests.create_projects_git( os.path.join(self.path, 'repos'), bare=True) tests.create_projects_git( os.path.join(self.path, 'repos', 'forks', 'foo'), bare=True) tests.add_content_git_repo( os.path.join(self.path, 'repos', 'forks', 'foo', 'test.git'), branch='feature') # Create foo's fork of the test project item = pagure.lib.model.Project( user_id=2, # foo name='test', is_fork=True, parent_id=1, # test description='test project #1', hook_token='aaabbbcccddd', ) item.close_status = ['Invalid', 'Insufficient data', 'Fixed', 'Duplicate'] self.session.add(item) self.session.commit() # Get on with testing user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() # Query branches on the Ralph's fork data = { 'csrf_token': csrf_token, 'repo': 'test', 'repouser': 'foo', } output = self.app.post('/pv/pull-request/ready', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) self.assertEqual( sorted(js_data.keys()), [u'code', u'message'] ) self.assertEqual(js_data['code'], 'OK') self.assertListEqual( sorted(js_data['message'].keys()), [u'branch_w_pr', u'new_branch']) self.assertEqual(js_data['message']['branch_w_pr'], {}) self.assertEqual(js_data['message']['new_branch'].keys(), ['feature']) self.assertEqual(len(js_data['message']['new_branch']['feature']), 2) def test_get_pull_request_ready_branch_on_fork_no_parent_no_pr(self): '''Test the get_pull_request_ready_branch from the internal API on a fork that has no parent repo (deleted) and doesn't allow PR ''' tests.create_projects(self.session) tests.create_projects_git( os.path.join(self.path, 'repos'), bare=True) tests.create_projects_git( os.path.join(self.path, 'repos', 'forks', 'foo'), bare=True) tests.add_content_git_repo( os.path.join(self.path, 'repos', 'forks', 'foo', 'test.git'), branch='feature') # Create foo's fork of the test project item = pagure.lib.model.Project( user_id=2, # foo name='test', is_fork=True, parent_id=1, # test description='test project #1', hook_token='aaabbbcccddd', ) item.close_status = ['Invalid', 'Insufficient data', 'Fixed', 'Duplicate'] self.session.add(item) self.session.commit() settings = item.settings settings['pull_requests'] = False item.settings = settings self.session.add(item) self.session.commit() # Delete the parent project project = pagure.lib.get_authorized_project(self.session, 'test') self.session.delete(project) self.session.commit() # Get on with testing user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() # Query branches on the Ralph's fork data = { 'csrf_token': csrf_token, 'repo': 'test', 'repouser': 'foo', } output = self.app.post('/pv/pull-request/ready', data=data) self.assertEqual(output.status_code, 400) js_data = json.loads(output.data.decode('utf-8')) self.assertEqual( sorted(js_data.keys()), [u'code', u'message'] ) self.assertEqual(js_data['code'], 'ERROR') self.assertEqual( js_data['message'], 'Pull-request have been disabled for this repo') def test_get_pull_request_ready_branch_on_fork_no_parent(self): '''Test the get_pull_request_ready_branch from the internal API on a fork that has no parent repo (deleted). ''' tests.create_projects(self.session) tests.create_projects_git( os.path.join(self.path, 'repos'), bare=True) tests.create_projects_git( os.path.join(self.path, 'repos', 'forks', 'foo'), bare=True) tests.add_content_git_repo( os.path.join(self.path, 'repos', 'forks', 'foo', 'test.git'), branch='feature') # Create foo's fork of the test project item = pagure.lib.model.Project( user_id=2, # foo name='test', is_fork=True, parent_id=1, # test description='test project #1', hook_token='aaabbbcccddd', ) item.close_status = ['Invalid', 'Insufficient data', 'Fixed', 'Duplicate'] self.session.add(item) self.session.commit() settings = item.settings settings['pull_requests'] = True item.settings = settings self.session.add(item) self.session.commit() # Delete the parent project project = pagure.lib.get_authorized_project(self.session, 'test') self.session.delete(project) self.session.commit() # Get on with testing user = tests.FakeUser() user.username = 'pingou' with tests.user_set(self.app.application, user): csrf_token = self.get_csrf() # Query branches on the Ralph's fork data = { 'csrf_token': csrf_token, 'repo': 'test', 'repouser': 'foo', } output = self.app.post('/pv/pull-request/ready', data=data) self.assertEqual(output.status_code, 200) js_data = json.loads(output.data.decode('utf-8')) self.assertEqual( sorted(js_data.keys()), [u'code', u'message'] ) self.assertEqual(js_data['code'], 'OK') self.assertEqual( sorted(js_data['message'].keys()), [u'branch_w_pr', u'new_branch']) self.assertEqual(js_data['message']['branch_w_pr'], {}) self.assertEqual(js_data['message']['new_branch'].keys(), ['feature']) self.assertEqual(len(js_data['message']['new_branch']['feature']), 2) if __name__ == '__main__': unittest.main(verbosity=2)