hazza/jupyterhub
1
0
Fork 0
mirror of https://github.com/jupyterhub/jupyterhub.git synced 2025-10-12 12:33:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

test spawner failure mid-progress

Browse Source
This commit is contained in:
Min RK
2018-04-09 14:46:08 +02:00
parent 2952f62726
commit 307ad636dc
1 changed files with 31 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
jupyterhub/tests/test_api.py
View File

@@ -696,7 +696,7 @@ def next_event(it):
@mark.gen_test
def test_spawn_progress(request, app, no_patience, slow_spawn):
def test_progress(request, app, no_patience, slow_spawn):
db = app.db
name = 'martin'
app_user = add_user(db, app=app, name=name)
@@ -729,7 +729,7 @@ def test_spawn_progress(request, app, no_patience, slow_spawn):
@mark.gen_test
def test_spawn_progress_not_started(request, app):
def test_progress_not_started(request, app):
db = app.db
name = 'nope'
app_user = add_user(db, app=app, name=name)
@@ -742,7 +742,7 @@ def test_spawn_progress_not_started(request, app):
@mark.gen_test
def test_spawn_progress_not_found(request, app):
def test_progress_not_found(request, app):
db = app.db
name = 'noserver'
r = yield api_request(app, 'users', 'nosuchuser', 'server/progress')
@@ -753,7 +753,7 @@ def test_spawn_progress_not_found(request, app):
@mark.gen_test
def test_spawn_progress_ready(request, app):
def test_progress_ready(request, app):
"""Test progress API when spawner is already started
e.g. a race between requesting progress and progress already being complete
@@ -775,11 +775,8 @@ def test_spawn_progress_ready(request, app):
@mark.gen_test
def test_spawn_progress_bad(request, app, no_patience, bad_spawn):
"""Test progress API when spawner is already started
e.g. a race between requesting progress and progress already being complete
"""
def test_progress_bad(request, app, no_patience, bad_spawn):
"""Test progress API when spawner has already failed"""
db = app.db
name = 'simon'
app_user = add_user(db, app=app, name=name)
@@ -798,6 +795,31 @@ def test_spawn_progress_bad(request, app, no_patience, bad_spawn):
}
@mark.gen_test
def test_progress_bad_slow(request, app, no_patience, slow_bad_spawn):
"""Test progress API when spawner fails while watching"""
db = app.db
name = 'eugene'
app_user = add_user(db, app=app, name=name)
r = yield api_request(app, 'users', name, 'server', method='post')
assert r.status_code == 202
r = yield api_request(app, 'users', name, 'server/progress', stream=True)
r.raise_for_status()
request.addfinalizer(r.close)
ex = async_requests.executor
line_iter = iter(r.iter_lines(decode_unicode=True))
evt = yield ex.submit(next_event, line_iter)
assert evt['progress'] == 0
evt = yield ex.submit(next_event, line_iter)
assert evt['progress'] == 50
evt = yield ex.submit(next_event, line_iter)
assert evt == {
'progress': 100,
'failed': True,
'message': "Spawn failed: I don't work!",
}
@async_generator
async def progress_forever():
"""progress function that yields messages forever"""