From 224787bbaccfc2d065cbc6fb507820fe8d30cf61 Mon Sep 17 00:00:00 2001 From: Tiago Natel Date: Tue, 26 Nov 2019 16:00:11 +0000 Subject: Tests: added getjson() helper. --- test/unit/http.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'test/unit/http.py') diff --git a/test/unit/http.py b/test/unit/http.py index d59c7b56..839e91a2 100644 --- a/test/unit/http.py +++ b/test/unit/http.py @@ -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 -- cgit