ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2024-12-19 04:57:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

move response-check to function

Browse Source
This commit is contained in:
meejah 2019-08-07 13:12:52 -06:00
parent 7738f9f4e3
commit d939ed5042
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
integration/util.py
View File

@ -371,6 +371,16 @@ def node_url(node_dir, uri_fragment):
return url
def _check_status(response):
"""
Check the response code is a 2xx (raise an exception otherwise)
"""
if response.status_code < 200 or response.status_code >= 300:
raise RuntimeError(
"Expected a 2xx code, got {}".format(response.status_code)
)
def web_get(node_dir, uri_fragment, **kw):
"""
Make a web-request to the webport of `node_dir`. This will look
@ -378,10 +388,7 @@ def web_get(node_dir, uri_fragment, **kw):
"""
url = node_url(node_dir, uri_fragment)
resp = requests.get(url, **kw)
if resp.status_code < 200 or resp.status_code >= 300:
raise RuntimeError(
"Expected a 200 code, got {}".format(resp.status_code)
)
_check_status(resp)
return resp.content
@ -392,10 +399,7 @@ def web_post(node_dir, uri_fragment, **kw):
"""
url = node_url(node_dir, uri_fragment)
resp = requests.post(url, **kw)
if resp.status_code < 200 or resp.status_code >= 300:
raise RuntimeError(
"Expected a 200 code, got {}".format(resp.status_code)
)
_check_status(resp)
return resp.content