aboutsummaryrefslogtreecommitdiffstats
path: root/python
diff options
context:
space:
mode:
authorBryan Newbold <bnewbold@robocracy.org>2018-12-29 00:07:14 -0800
committerBryan Newbold <bnewbold@robocracy.org>2018-12-29 00:07:14 -0800
commitd95fa771f5c9a75dece2038ee50619a53f9c43f7 (patch)
tree5126a6b0163abc748c0d3656aae73ad684fe7a80 /python
parent0f7fb8806c1a39aa54d0cd051aa48478c1b1c070 (diff)
downloadfatcat-d95fa771f5c9a75dece2038ee50619a53f9c43f7.tar.gz
fatcat-d95fa771f5c9a75dece2038ee50619a53f9c43f7.zip
check request status codes idiomatically
Diffstat (limited to 'python')
-rw-r--r--python/fatcat_tools/harvest/doi_registrars.py4
-rw-r--r--python/fatcat_tools/workers/elasticsearch.py2
-rw-r--r--python/fatcat_web/search.py1
-rw-r--r--python/tests/routes.py40
-rw-r--r--python/tests/subentity_state.py38
5 files changed, 44 insertions, 41 deletions
diff --git a/python/fatcat_tools/harvest/doi_registrars.py b/python/fatcat_tools/harvest/doi_registrars.py
index d27389ba..4a0cb8db 100644
--- a/python/fatcat_tools/harvest/doi_registrars.py
+++ b/python/fatcat_tools/harvest/doi_registrars.py
@@ -95,11 +95,11 @@ class HarvestCrossrefWorker:
while True:
http_resp = requests.get(self.api_host_url, params, headers=headers)
if http_resp.status_code == 503:
- # crud backoff
+ # crude backoff
print("got HTTP {}, pausing for 30 seconds".format(http_resp.status_code))
time.sleep(30.0)
continue
- assert http_resp.status_code == 200
+ http_resp.raise_for_status()
resp = http_resp.json()
items = self.extract_items(resp)
count += len(items)
diff --git a/python/fatcat_tools/workers/elasticsearch.py b/python/fatcat_tools/workers/elasticsearch.py
index e7abd5ee..3adb088e 100644
--- a/python/fatcat_tools/workers/elasticsearch.py
+++ b/python/fatcat_tools/workers/elasticsearch.py
@@ -48,5 +48,5 @@ class ElasticsearchReleaseWorker(FatcatWorker):
release.ident)
print("Updating document: {}".format(elasticsearch_endpoint))
resp = requests.post(elasticsearch_endpoint, json=release_to_elasticsearch(release))
- assert resp.status_code in (200, 201)
+ resp.raise_for_status()
#consumer.commit_offsets()
diff --git a/python/fatcat_web/search.py b/python/fatcat_web/search.py
index 471e522a..5cebfde4 100644
--- a/python/fatcat_web/search.py
+++ b/python/fatcat_web/search.py
@@ -36,6 +36,7 @@ def do_search(q, limit=50, fulltext_only=True):
(app.config['ELASTICSEARCH_BACKEND'], app.config['ELASTICSEARCH_INDEX']),
json=search_request)
+ #rv.raise_for_status()
if resp.status_code != 200:
print("elasticsearch non-200 status code: " + str(resp.status_code))
print(resp.content)
diff --git a/python/tests/routes.py b/python/tests/routes.py
index bca7cc0f..7b4ac442 100644
--- a/python/tests/routes.py
+++ b/python/tests/routes.py
@@ -9,7 +9,7 @@ from fixtures import *
def test_static_routes(app):
for route in ('/health', '/robots.txt', '/', '/about'):
rv = app.get(route)
- assert rv.status_code == 200
+ rv.raise_for_status()
assert app.get("/static/bogus/route").status_code == 404
@@ -27,16 +27,16 @@ def test_all_views(app):
assert rv.status_code == 404
rv = app.get('/container/aaaaaaaaaaaaaeiraaaaaaaaai')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/container/aaaaaaaaaaaaaeiraaaaaaaaai/history')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/container/aaaaaaaaaaaaaeiraaaaaaaaai/edit')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/container/create')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/container/lookup')
assert rv.status_code == 400
@@ -48,34 +48,34 @@ def test_all_views(app):
assert rv.status_code == 302
rv = app.get('/creator/aaaaaaaaaaaaaircaaaaaaaaai')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/creator/aaaaaaaaaaaaaircaaaaaaaaai/history')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/creator/aaaaaaaaaaaaaircaaaaaaaaai/edit')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/creator/lookup?orcid=0000-0003-2088-7465')
assert rv.status_code == 302
rv = app.get('/file/aaaaaaaaaaaaamztaaaaaaaaai')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/file/lookup?sha1=7d97e98f8af710c7e7fe703abc8f639e0ee507c4')
assert rv.status_code == 302
rv = app.get('/release/aaaaaaaaaaaaarceaaaaaaaaai')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/release/aaaaaaaaaaaaarceaaaaaaaaai/history')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/release/aaaaaaaaaaaaarceaaaaaaaaai/edit')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/release/create')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/release/lookup?doi=10.123/abc')
assert rv.status_code == 302
@@ -84,22 +84,22 @@ def test_all_views(app):
assert rv.status_code == 302
rv = app.get('/release/search')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/work/aaaaaaaaaaaaavkvaaaaaaaaai')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/work/aaaaaaaaaaaaavkvaaaaaaaaai/history')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/work/aaaaaaaaaaaaavkvaaaaaaaaai/edit')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/work/create')
assert rv.status_code == 404
rv = app.get('/editgroup/aaaaaaaaaaaabo53aaaaaaaaae')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/editgroup/ccccccccccccccccccccccccca')
print(rv)
@@ -110,7 +110,7 @@ def test_all_views(app):
#assert rv.status_code == 302
rv = app.get('/editor/aaaaaaaaaaaabkvkaaaaaaaaae')
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/editor/aaaaaaaaaaaabkvkaaaaaaaaae/changelog')
- assert rv.status_code == 200
+ rv.raise_for_status()
diff --git a/python/tests/subentity_state.py b/python/tests/subentity_state.py
index 03362686..799f5e03 100644
--- a/python/tests/subentity_state.py
+++ b/python/tests/subentity_state.py
@@ -36,7 +36,7 @@ def test_relation_states(api, app):
eg = quick_eg(api)
j2 = api.get_container(api.create_container(j2, editgroup_id=eg.editgroup_id).ident)
rv = app.get('/container/{}'.format(j2.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
# create inter-related entities
eg = quick_eg(api)
@@ -57,14 +57,15 @@ def test_relation_states(api, app):
assert r1.contribs[0].creator.display_name == "test person"
assert r1.contribs[0].creator.state == "active"
rv = app.get('/release/{}'.format(r1.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
# delete creator
eg = quick_eg(api)
api.delete_creator(c1.ident, editgroup_id=eg.editgroup_id)
api.accept_editgroup(eg.editgroup_id)
rv = app.get('/creator/{}'.format(c1.ident))
- assert rv.status_code == 200 # TODO: HTTP status "Gone"?
+ # TODO: HTTP status "Gone"?
+ rv.raise_for_status()
c1_deleted = api.get_creator(c1.ident)
assert c1_deleted.state == "deleted"
@@ -77,7 +78,7 @@ def test_relation_states(api, app):
assert r1.contribs[0].creator.display_name is None
assert r1.contribs[0].creator.state == "deleted"
rv = app.get('/release/{}'.format(r1.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
# wip container
eg = quick_eg(api)
@@ -94,7 +95,7 @@ def test_relation_states(api, app):
assert r1.contribs[0].creator.display_name is None
assert r1.contribs[0].creator.state == "deleted"
rv = app.get('/release/{}'.format(r1.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
# redirect release
r2 = api.get_release(r2.ident, expand="container,creators,files")
@@ -116,7 +117,7 @@ def test_relation_states(api, app):
rv = app.get('/release/{}'.format(r2.ident))
assert rv.status_code == 302
rv = app.get('/file/{}'.format(f2.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
# delete release
eg = quick_eg(api)
@@ -128,10 +129,11 @@ def test_relation_states(api, app):
assert r2.files is None
assert r2.contribs is None
rv = app.get('/release/{}'.format(r2.ident))
- assert rv.status_code == 200 # TODO: HTTP Gone?
+ # TODO: HTTP Gone?
+ rv.raise_for_status()
rv = app.get('/file/{}'.format(f2.ident))
print(rv.data)
- assert rv.status_code == 200
+ rv.raise_for_status()
def test_app_entity_states(api, app):
@@ -171,23 +173,23 @@ def test_app_entity_states(api, app):
# all entities
rv = app.get('/container/{}'.format(j1.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/container/{}'.format(j2.ident))
assert rv.status_code == 302
rv = app.get('/creator/{}'.format(c1.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/creator/{}'.format(c2.ident))
assert rv.status_code == 302
rv = app.get('/file/{}'.format(f1.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/file/{}'.format(f2.ident))
assert rv.status_code == 302
rv = app.get('/release/{}'.format(r1.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/release/{}'.format(r2.ident))
assert rv.status_code == 302
rv = app.get('/work/{}'.format(r1.work_id))
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/work/{}'.format(r2.work_id))
assert rv.status_code == 302
@@ -202,23 +204,23 @@ def test_app_entity_states(api, app):
# all entities
rv = app.get('/container/{}'.format(j1.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/container/{}'.format(j2.ident))
assert rv.status_code == 302
rv = app.get('/creator/{}'.format(c1.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/creator/{}'.format(c2.ident))
assert rv.status_code == 302
rv = app.get('/file/{}'.format(f1.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/file/{}'.format(f2.ident))
assert rv.status_code == 302
rv = app.get('/release/{}'.format(r1.ident))
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/release/{}'.format(r2.ident))
assert rv.status_code == 302
rv = app.get('/work/{}'.format(r1.work_id))
- assert rv.status_code == 200
+ rv.raise_for_status()
rv = app.get('/work/{}'.format(r2.work_id))
assert rv.status_code == 302