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

replace existing get_token tests

Browse Source
This commit is contained in:
Min RK
2017-02-08 16:46:45 +01:00
parent f3b7fda4a8
commit 2a25e3cb89
1 changed files with 18 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
jupyterhub/tests/test_api.py
View File

@@ -131,26 +131,6 @@ def test_auth_api(app):
assert r.status_code == 403 assert r.status_code == 403
@mark.parametrize("headers, data, status", [
({}, None, 200),
({'Authorization': ''}, None, 403),
({}, {'username': 'fake', 'password': 'fake'}, 200),
])
def test_request_token(app, headers, data, status):
if data:
data = json.dumps(data)
# request a new token
r = api_request(app, 'authorizations', 'token', method='post', data=data, headers=headers)
assert r.status_code == status
if status != 200:
return
reply = r.json()
assert 'token' in reply
r = api_request(app, 'authorizations', 'token', reply['token'])
r.raise_for_status()
assert 'name' in r.json()
def test_referer_check(app, io_loop): def test_referer_check(app, io_loop):
url = ujoin(public_host(app), app.hub.server.base_url) url = ujoin(public_host(app), app.hub.server.base_url)
host = urlparse(url).netloc host = urlparse(url).netloc
@@ -588,28 +568,24 @@ def test_token(app):
assert r.status_code == 404 assert r.status_code == 404
def test_get_token(app): @mark.parametrize("headers, data, status", [
name = 'user' ({}, None, 200),
user = add_user(app.db, app=app, name=name) ({'Authorization': ''}, None, 403),
r = api_request(app, 'authorizations/token', method='post', data=json.dumps({ ({}, {'username': 'fake', 'password': 'fake'}, 200),
'username': name, ])
'password': name, def test_get_new_token(app, headers, data, status):
})) if data:
assert r.status_code == 200 data = json.dumps(data)
data = r.content.decode("utf-8") # request a new token
token = json.loads(data) r = api_request(app, 'authorizations', 'token', method='post', data=data, headers=headers)
assert not token['Authentication'] is None assert r.status_code == status
if status != 200:
return
def test_bad_get_token(app): reply = r.json()
name = 'user' assert 'token' in reply
password = 'fake' r = api_request(app, 'authorizations', 'token', reply['token'])
user = add_user(app.db, app=app, name=name) r.raise_for_status()
r = api_request(app, 'authorizations/token', method='post', data=json.dumps({ assert 'name' in r.json()
'username': name,
'password': password,
}))
assert r.status_code == 403
# --------------- # ---------------