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

15
jupyterhub/tests/test_api.py
View File

@@ -13,6 +13,7 @@ from pytest import mark
import requests
from tornado import gen
from tornado.concurrent import Future as tFuture
import jupyterhub
from .. import orm
@@ -430,17 +431,27 @@ def test_set_auth_state(app):
user = find_user(db, name)
assert user is not None
with mock.patch('jupyterhub.user.encrypt') as mock_encrypt:
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
user = find_user(db, name)
assert user is not None
assert user.name == name
encrypted_auth = user.encrypted_auth_state
assert encrypted_auth is not None
assert user.encrypted_auth_state == b"encrypted_state"
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 user.name == name
assert r.json()['auth_state'] == auth_state