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

Modify auth_state API test to use mocked crypto

Browse Source
This commit is contained in:
Tim Head
2018-03-22 16:45:40 +01:00
parent 1ee993c664
commit 2582f0bbe6
1 changed files with 17 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
jupyterhub/tests/test_api.py
View File

@@ -13,6 +13,7 @@ from pytest import mark
import requests import requests
from tornado import gen from tornado import gen
from tornado.concurrent import Future as tFuture
import jupyterhub import jupyterhub
from .. import orm from .. import orm
@@ -430,17 +431,27 @@ def test_set_auth_state(app):
user = find_user(db, name) user = find_user(db, name)
assert user is not None assert user is not None
r = yield api_request(app, 'users', name, method='patch', with mock.patch('jupyterhub.user.encrypt') as mock_encrypt:
data=json.dumps({'auth_state': auth_state}) encrypt_future = tFuture()
) encrypt_future.set_result(b"encrypted_state")
mock_encrypt.return_value = encrypt_future
r = yield api_request(app, 'users', name, method='patch',
data=json.dumps({'auth_state': auth_state})
)
assert mock_encrypt.call_count == 1
assert r.status_code == 200 assert r.status_code == 200
user = find_user(db, name) user = find_user(db, name)
assert user is not None assert user is not None
assert user.name == name assert user.name == name
encrypted_auth = user.encrypted_auth_state assert user.encrypted_auth_state == b"encrypted_state"
assert encrypted_auth is not None
r = yield api_request(app, 'users', name) with mock.patch('jupyterhub.user.decrypt') as mock_decrypt:
decrypt_future = tFuture()
decrypt_future.set_result(auth_state)
mock_decrypt.return_value = decrypt_future
r = yield api_request(app, 'users', name)
assert mock_decrypt.call_count == 1
assert r.status_code == 200 assert r.status_code == 200
assert user.name == name assert user.name == name
assert r.json()['auth_state'] == auth_state assert r.json()['auth_state'] == auth_state