new helper method to build json body in requests

This commit is contained in:
shockrah 2020-10-21 22:32:59 -07:00
parent 33234f8e18
commit 35dac99d88

View File

@ -44,30 +44,27 @@ class Test:
print(f'\t[Status Code]: {request.status_code}') print(f'\t[Status Code]: {request.status_code}')
print(f'\t[Body]: {request.text}') print(f'\t[Body]: {request.text}')
def post(self, url, **opts): def _build_req_body(self, **options):
rbody = self.body _map = self.body
for k in opts: for key in options:
rbody[k] = opts[k] _map[key] = options[key]
body = json.dumps(rbody) return json.dumps(_map)
r = requests.post(self.base + url, data=rbody)
def post(self, url, **opts):
body = self._build_req_body(**opts)
r = requests.post(self.base + url, data=body)
self.log(self.base + url, 'POST', r) self.log(self.base + url, 'POST', r)
def get(self, url, **opts): def get(self, url, **opts):
rbody = self.body body = self._build_req_body(**opts)
for k in opts:
rbody[k] = opts[k]
body = json.dumps(rbody)
r = requests.get(self.base + url, data=body) r = requests.get(self.base + url, data=body)
self.log(self.base + url, 'GET', r) self.log(self.base + url, 'GET', r)
def delete(self, url, **opts): def delete(self, url, **opts):
rbody = self.body body = self._build_req_body(**opts)
for k in opts:
rbody[k] = opts[k]
body =json.dumps(rbody)
r = requests.delete(self.base + url, data=body) r = requests.delete(self.base + url, data=body)
self.log(self.base + url, 'DELETE', r) self.log(self.base + url, 'DELETE', r)