Tests: added getjson() helper.

This commit is contained in:
Tiago Natel
2019-11-26 16:00:11 +00:00
parent 01103c5005
commit 224787bbac
4 changed files with 29 additions and 10 deletions

View File

@@ -122,6 +122,9 @@ class TestHTTP(TestUnit):
encoding
)
if 'json' in kwargs:
resp = self._parse_json(resp)
if 'start' not in kwargs:
sock.close()
return resp
@@ -230,6 +233,23 @@ class TestHTTP(TestUnit):
return body
def _parse_json(self, resp):
headers = resp['headers']
self.assertIn('Content-Type', headers, 'Content-Type header set')
self.assertEqual(
headers['Content-Type'],
'application/json',
'Content-Type header is application/json',
)
resp['body'] = json.loads(resp['body'])
return resp
def getjson(self, **kwargs):
return self.get(json=True, **kwargs)
def waitforsocket(self, port):
ret = False