mirror of
https://github.com/jupyterhub/jupyterhub.git
synced 2025-10-10 11:33:01 +00:00
5c87461605
basic testing framework to get starting writing and testing the REST API including tests for the authorizations API, the only API URL defined so far.
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
"""Tests for the REST API"""
|
|
|
|
import requests
|
|
|
|
from ..utils import url_path_join as ujoin
|
|
from .. import orm
|
|
|
|
|
|
def api_request(app, *api_path, **kwargs):
|
|
"""Make an API request"""
|
|
base_url = app.hub.server.url
|
|
token = app.db.query(orm.APIToken).first()
|
|
kwargs.setdefault('headers', {})
|
|
kwargs['headers'].setdefault('Authorization', 'token %s' % token.token)
|
|
|
|
url = ujoin(base_url, 'api', *api_path)
|
|
method = kwargs.pop('method', 'get')
|
|
f = getattr(requests, method)
|
|
return f(url, **kwargs)
|
|
|
|
def test_auth_api(app):
|
|
db = app.db
|
|
r = api_request(app, 'authorizations', 'gobbledygook')
|
|
assert r.status_code == 404
|
|
|
|
# make a new cookie token
|
|
user = db.query(orm.User).first()
|
|
cookie_token = user.new_cookie_token()
|
|
db.add(cookie_token)
|
|
db.commit()
|
|
|
|
# check success:
|
|
r = api_request(app, 'authorizations', cookie_token.token)
|
|
assert r.status_code == 200
|
|
reply = r.json()
|
|
assert reply['user'] == user.name
|
|
|
|
# check fail
|
|
r = api_request(app, 'authorizations', cookie_token.token,
|
|
headers={'Authorization': 'no sir'},
|
|
)
|
|
assert r.status_code == 403
|
|
|
|
r = api_request(app, 'authorizations', cookie_token.token,
|
|
headers={'Authorization': 'token: %s' % cookie_token.token},
|
|
)
|
|
assert r.status_code == 403
|
|
|
|
|