def test_validate_ok(self, req):
response = {'result': True}
req.get('https://bugzilla.redhat.com/rest/valid_login?' +
urlencode(self.fake_creds),
text=json.dumps(response))
bugzilla.Session()
def test_update_bug_single(self):
self.req.put('https://bugzilla.redhat.com/rest/bug/12345',
text=json.dumps('fake_response'))
session = bugzilla.Session()
r = session.update_bug('12345', {'cf_internal_whiteboard': 'test'})
bug_req = next((req for req in self.req.request_history
if req.path == '/rest/bug/12345'))
self.assertDictEqual({'cf_internal_whiteboard': 'test'},
bug_req.json())
self.assertEqual(r, 'fake_response')
def test_bugs_multiple(self):
query = dict(id='1,2,3,4,5', **self.fake_creds)
expected_url = ('https://bugzilla.redhat.com/rest/bug?' +
urlencode(query))
self.req.get(expected_url, text=json.dumps({'bugs': ['fake_bug']}))
session = bugzilla.Session()
r = session.get_bugs([1, 2, 3, 4, 5])
req = self._find_req_for_path('/rest/bug')
self.assertIsNotNone(req)
self._assert_query_match(query, req.qs)
self.assertIsInstance(r, types.GeneratorType)
self.assertListEqual(['fake_bug'], list(r))
def _test_update_bug_multiple(self):
self.req.put('https://bugzilla.redhat.com/rest/bug',
text=json.dumps('fake_response'))
session = bugzilla.Session()
r = session.update_bugs([1, 2, 3, 4, 5],
{'cf_internal_whiteboard': 'test'})
bug_req = next((req for req in self.req.request_history
if req.path == '/rest/bug'))
self.assertDictEqual(
{
'cf_internal_whiteboard': 'test',
'ids': [1, 2, 3, 4, 5]
}, bug_req.json())
self.assertEqual(r, 'fake_response')
def test_bugs_include_fields(self):
query = dict(id='1,2,3,4,5',
include_fields='cf_internal_whiteboard,id',
**self.fake_creds)
expected_url = ('https://bugzilla.redhat.com/rest/bug?' +
urlencode(query))
self.req.get(expected_url, text=json.dumps({'bugs': ['fake_bug']}))
session = bugzilla.Session()
r = session.get_bugs([1, 2, 3, 4, 5],
fields=['cf_internal_whiteboard'])
bug_req = next((req for req in self.req.request_history
if req.path == '/rest/bug'))
self._assert_query_match(query, bug_req.qs)
self.assertIsInstance(r, types.GeneratorType)
self.assertListEqual(['fake_bug'], list(r))