# -*- coding: utf-8 -*- """ (c) 2015 - Copyright Red Hat Inc Authors: Pierre-Yves Chibon """ __requires__ = ['SQLAlchemy >= 0.8'] import pkg_resources import json import unittest import shutil import sys import os import pygit2 from mock import patch sys.path.insert(0, os.path.join(os.path.dirname( os.path.abspath(__file__)), '..')) import pagure.lib import tests 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.APP.config['TESTING'] = True pagure.APP.config['IP_ALLOWED_INTERNAL'] = list(set( pagure.APP.config['IP_ALLOWED_INTERNAL'] + [None])) pagure.SESSION = self.session pagure.internal.SESSION = self.session pagure.ui.repo.SESSION = self.session pagure.APP.config['GIT_FOLDER'] = tests.HERE pagure.APP.config['FORK_FOLDER'] = os.path.join( tests.HERE, 'forks') pagure.APP.config['REQUESTS_FOLDER'] = None pagure.APP.config['TICKETS_FOLDER'] = None pagure.APP.config['DOCS_FOLDER'] = None self.app = pagure.APP.test_client() @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_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'}) repo = pagure.lib.get_project(self.session, 'test') request = repo.requests[0] self.assertEqual(len(request.comments), 1) self.assertEqual(len(request.discussion), 1) # Check the @localonly pagure.APP.config['IP_ALLOWED_INTERNAL'].remove(None) output = self.app.put('/pv/pull-request/comment/', data=data) self.assertEqual(output.status_code, 403) @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_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'}) repo = pagure.lib.get_project(self.session, 'test') issue = repo.issues[0] self.assertEqual(len(issue.comments), 1) # Check the @localonly pagure.APP.config['IP_ALLOWED_INTERNAL'].remove(None) output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 403) @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_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'}) repo = pagure.lib.get_project(self.session, 'test') issue = repo.issues[0] self.assertEqual(len(issue.comments), 1) # Check the @localonly pagure.APP.config['IP_ALLOWED_INTERNAL'].remove(None) output = self.app.put('/pv/ticket/comment/', data=data) self.assertEqual(output.status_code, 403) @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 gitrepo = os.path.join(tests.HERE, '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/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') # Create a PR for these changes tests.create_projects(self.session) project = pagure.lib.get_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(pagure.APP, 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_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(tests.HERE, '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_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(pagure.APP, 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_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 gitrepo = os.path.join(tests.HERE, '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/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') 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] ) # Create a PR for these changes tests.create_projects(self.session) project = pagure.lib.get_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(pagure.APP, 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_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 gitrepo = os.path.join(tests.HERE, '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/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, '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] ) # Create a PR for these changes tests.create_projects(self.session) project = pagure.lib.get_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(pagure.APP, 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_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) user = tests.FakeUser() user.username = 'pingou' with tests.user_set(pagure.APP, user): output = self.app.get('/test/adduser') 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 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': u'No git repo found with the information provided' } ) # Create a git repo to play with gitrepo = os.path.join(tests.HERE, '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/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'], } ) if __name__ == '__main__': SUITE = unittest.TestLoader().loadTestsFromTestCase( PagureFlaskInternaltests) unittest.TextTestRunner(verbosity=2).run(SUITE)