| """ |
| (c) 2017 - Copyright Red Hat Inc |
| |
| Authors: |
| Clement Verna <cverna@tutanota.com> |
| |
| """ |
| |
| from __future__ import unicode_literals, absolute_import |
| |
| import unittest |
| import sys |
| import os |
| import json |
| |
| from mock import patch, MagicMock |
| |
| sys.path.insert(0, os.path.join(os.path.dirname( |
| os.path.abspath(__file__)), '..')) |
| |
| import pagure.lib.query |
| import tests |
| |
| |
| class PagureFlaskApiCustomFieldIssuetests(tests.Modeltests): |
| """ Tests for the flask API of pagure for issue's custom fields """ |
| |
| def setUp(self): |
| """ Set up the environnment, ran before every tests. """ |
| self.maxDiff = None |
| super(PagureFlaskApiCustomFieldIssuetests, self).setUp() |
| |
| pagure.config.config['TICKETS_FOLDER'] = None |
| |
| tests.create_projects(self.session) |
| tests.create_projects_git(os.path.join(self.path, 'tickets')) |
| tests.create_tokens(self.session) |
| tests.create_tokens_acl(self.session) |
| |
| |
| repo = pagure.lib.query.get_authorized_project(self.session, 'test') |
| pagure.lib.query.new_issue( |
| session=self.session, |
| repo=repo, |
| title='Test issue #1', |
| content='We should work on this', |
| user='pingou', |
| private=False, |
| ) |
| self.session.commit() |
| |
| def test_api_update_custom_field_bad_request(self): |
| """ Test the api_update_custom_field method of the flask api. |
| This test that a badly form request returns the correct error. |
| """ |
| |
| headers = {'Authorization': 'token aaabbbcccddd'} |
| |
| |
| payload = {} |
| output = self.app.post( |
| '/api/0/test/issue/1/custom', headers=headers, data=payload) |
| self.assertEqual(output.status_code, 400) |
| data = json.loads(output.get_data(as_text=True)) |
| self.assertDictEqual( |
| data, |
| { |
| "error": "Invalid or incomplete input submitted", |
| "error_code": "EINVALIDREQ", |
| } |
| ) |
| |
| def test_api_update_custom_field_wrong_field(self): |
| """ Test the api_update_custom_field method of the flask api. |
| This test that an invalid field retruns the correct error. |
| """ |
| |
| headers = {'Authorization': 'token aaabbbcccddd'} |
| |
| payload = {'foo': 'bar'} |
| output = self.app.post( |
| '/api/0/test/issue/1/custom', headers=headers, data=payload) |
| self.assertEqual(output.status_code, 400) |
| data = json.loads(output.get_data(as_text=True)) |
| self.assertDictEqual( |
| data, |
| { |
| "error": "Invalid custom field submitted", |
| "error_code": "EINVALIDISSUEFIELD", |
| } |
| ) |
| |
| @patch( |
| 'pagure.lib.query.set_custom_key_value', |
| MagicMock(side_effect=pagure.exceptions.PagureException('error'))) |
| def test_api_update_custom_field_raise_error(self): |
| """ Test the api_update_custom_field method of the flask api. |
| This test the successful requests scenarii. |
| """ |
| |
| headers = {'Authorization': 'token aaabbbcccddd'} |
| |
| |
| repo = pagure.lib.query.get_authorized_project(self.session, 'test') |
| msg = pagure.lib.query.set_custom_key_fields( |
| self.session, repo, |
| ['bugzilla', 'upstream', 'reviewstatus'], |
| ['link', 'boolean', 'list'], |
| ['unused data for non-list type', '', 'ack', 'nack', 'needs review'], |
| [None, None, None]) |
| self.session.commit() |
| self.assertEqual(msg, 'List of custom fields updated') |
| |
| payload = {'bugzilla': '', 'upstream': True} |
| output = self.app.post( |
| '/api/0/test/issue/1/custom', headers=headers, data=payload) |
| self.assertEqual(output.status_code, 400) |
| data = json.loads(output.get_data(as_text=True)) |
| self.assertDictEqual( |
| data, {u'error': u'error', u'error_code': u'ENOCODE'}) |
| |
| def test_api_update_custom_field(self): |
| """ Test the api_update_custom_field method of the flask api. |
| This test the successful requests scenarii. |
| """ |
| |
| headers = {'Authorization': 'token aaabbbcccddd'} |
| |
| |
| repo = pagure.lib.query.get_authorized_project(self.session, 'test') |
| msg = pagure.lib.query.set_custom_key_fields( |
| self.session, repo, |
| ['bugzilla', 'upstream', 'reviewstatus'], |
| ['link', 'boolean', 'list'], |
| ['unused data for non-list type', '', 'ack', 'nack', 'needs review'], |
| [None, None, None]) |
| self.session.commit() |
| self.assertEqual(msg, 'List of custom fields updated') |
| |
| payload = {'bugzilla': '', 'upstream': True} |
| output = self.app.post( |
| '/api/0/test/issue/1/custom', headers=headers, data=payload) |
| self.assertEqual(output.status_code, 200) |
| data = json.loads(output.get_data(as_text=True)) |
| data["messages"].sort(key=lambda d: list(d.keys())[0]) |
| self.assertDictEqual( |
| data, |
| { |
| "messages": [ |
| {"bugzilla": "No changes"}, |
| {"upstream": "Custom field upstream adjusted to True"}, |
| ] |
| } |
| ) |
| |
| self.session.commit() |
| repo = pagure.lib.query.get_authorized_project(self.session, 'test') |
| issue = pagure.lib.query.search_issues(self.session, repo, issueid=1) |
| self.assertEqual(len(issue.other_fields), 1) |
| |
| payload = {'bugzilla': 'https://bugzilla.redhat.com/1234', |
| 'upstream': False, |
| 'reviewstatus': 'ack'} |
| output = self.app.post( |
| '/api/0/test/issue/1/custom', headers=headers, |
| data=payload) |
| self.assertEqual(output.status_code, 200) |
| data = json.loads(output.get_data(as_text=True)) |
| data["messages"].sort(key=lambda d: list(d.keys())[0]) |
| self.assertDictEqual( |
| data, |
| { |
| "messages": [ |
| {"bugzilla": "Custom field bugzilla adjusted to " |
| "https://bugzilla.redhat.com/1234"}, |
| {"reviewstatus": "Custom field reviewstatus adjusted to ack"}, |
| {"upstream": "Custom field upstream adjusted to False (was: True)"}, |
| |
| ] |
| } |
| ) |
| |
| self.session.commit() |
| repo = pagure.lib.query.get_authorized_project(self.session, 'test') |
| issue = pagure.lib.query.search_issues(self.session, repo, issueid=1) |
| self.assertEqual(len(issue.other_fields), 3) |
| |
| |
| payload = {'bugzilla': '', 'upstream': '', 'reviewstatus': ''} |
| output = self.app.post( |
| '/api/0/test/issue/1/custom', headers=headers, |
| data=payload) |
| self.assertEqual(output.status_code, 200) |
| data = json.loads(output.get_data(as_text=True)) |
| data["messages"].sort(key=lambda d: list(d.keys())[0]) |
| self.assertDictEqual( |
| data, |
| { |
| "messages": [ |
| {"bugzilla": "Custom field bugzilla reset " |
| "(from https://bugzilla.redhat.com/1234)"}, |
| {"reviewstatus": "Custom field reviewstatus reset (from ack)"}, |
| {"upstream": "Custom field upstream reset (from False)"}, |
| ] |
| } |
| ) |
| |
| |
| if __name__ == '__main__': |
| unittest.main(verbosity=2) |