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

Fix minor formatting

Browse Source
This commit is contained in:
Carol Willing
2017-01-20 23:58:38 -08:00
parent 025977f19a
commit 0f283e088e
1 changed files with 31 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
jupyterhub/tests/test_api.py
View File

@@ -105,14 +105,10 @@ def test_auth_api(app):
assert reply['name'] == user.name assert reply['name'] == user.name
# check fail # check fail
r = api_request(app, 'authorizations/token', api_token, r = api_request(app, 'authorizations/token', api_token, headers={'Authorization': 'no sir'},)
headers={'Authorization': 'no sir'},
)
assert r.status_code == 403 assert r.status_code == 403
r = api_request(app, 'authorizations/token', api_token, r = api_request(app, 'authorizations/token', api_token, headers={'Authorization': 'token: %s' % user.cookie_id},)
headers={'Authorization': 'token: %s' % user.cookie_id},
)
assert r.status_code == 403 assert r.status_code == 403
@@ -125,22 +121,17 @@ def test_referer_check(app, io_loop):
cookies = app.login_user('admin') cookies = app.login_user('admin')
app_user = get_app_user(app, 'admin') app_user = get_app_user(app, 'admin')
# stop the admin's server so we don't mess up future tests # stop the admin's server so we don't mess up future tests
io_loop.run_sync(lambda : app.proxy.delete_user(app_user)) io_loop.run_sync(lambda: app.proxy.delete_user(app_user))
io_loop.run_sync(app_user.stop) io_loop.run_sync(app_user.stop)
r = api_request(app, 'users', r = api_request(app, 'users',
headers={ headers={'Authorization': '', 'Referer': 'null', },
'Authorization': '', cookies=cookies, )
'Referer': 'null',
}, cookies=cookies,
)
assert r.status_code == 403 assert r.status_code == 403
r = api_request(app, 'users', r = api_request(app, 'users', headers={
headers={
'Authorization': '', 'Authorization': '',
'Referer': 'http://attack.com/csrf/vulnerability', 'Referer': 'http://attack.com/csrf/vulnerability',
}, cookies=cookies, }, cookies=cookies,)
)
assert r.status_code == 403 assert r.status_code == 403
r = api_request(app, 'users', r = api_request(app, 'users',
headers={ headers={
@@ -159,7 +150,10 @@ def test_referer_check(app, io_loop):
) )
assert r.status_code == 200 assert r.status_code == 200
# user API tests
# --------------
# User API tests
# --------------
@mark.user @mark.user
@@ -412,7 +406,7 @@ def test_spawn(app, io_loop):
assert user.server is None assert user.server is None
after_servers = sorted(db.query(orm.Server), key=lambda s: s.url) after_servers = sorted(db.query(orm.Server), key=lambda s: s.url)
assert before_servers == after_servers assert before_servers == after_servers
tokens = list(db.query(orm.APIToken).filter(orm.APIToken.user_id==user.id)) tokens = list(db.query(orm.APIToken).filter(orm.APIToken.user_id == user.id))
assert tokens == [] assert tokens == []
@@ -559,6 +553,11 @@ def test_bad_get_token(app):
assert r.status_code == 403 assert r.status_code == 403
# ---------------
# Group API tests
# ---------------
@mark.group @mark.group
def test_groups_list(app): def test_groups_list(app):
r = api_request(app, 'groups') r = api_request(app, 'groups')
@@ -680,7 +679,11 @@ def test_group_delete_users(app):
assert sorted([ u.name for u in group.users ]) == sorted(names[2:]) assert sorted([ u.name for u in group.users ]) == sorted(names[2:])
# service API # -----------------
# Service API tests
# -----------------
@mark.services @mark.services
def test_get_services(app, mockservice): def test_get_services(app, mockservice):
db = app.db db = app.db
@@ -770,7 +773,10 @@ def test_info(app):
} }
# general API tests # -----------------
# General API tests
# -----------------
def test_options(app): def test_options(app):
r = api_request(app, 'users', method='options') r = api_request(app, 'users', method='options')
@@ -783,7 +789,11 @@ def test_bad_json_body(app):
assert r.status_code == 400 assert r.status_code == 400
# shutdown must be last # ---------------------------------
# Shutdown MUST always be last test
# ---------------------------------
def test_shutdown(app): def test_shutdown(app):
r = api_request(app, 'shutdown', method='post', data=json.dumps({ r = api_request(app, 'shutdown', method='post', data=json.dumps({
'servers': True, 'servers': True,
@@ -797,4 +807,3 @@ def test_shutdown(app):
else: else:
break break
assert not app.io_loop._running assert not app.io_loop._running