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

async fixes in test_init_tokens

Browse Source
This commit is contained in:
Min RK
2016-07-28 16:14:24 +02:00
parent 2a35d1c8a6
commit 39a80edb74
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
jupyterhub/tests/test_app.py
View File

@@ -55,8 +55,7 @@ def test_generate_config():
assert 'Spawner.cmd' in cfg_text assert 'Spawner.cmd' in cfg_text
assert 'Authenticator.whitelist' in cfg_text assert 'Authenticator.whitelist' in cfg_text
def test_init_tokens(io_loop):
def test_init_tokens():
with TemporaryDirectory() as td: with TemporaryDirectory() as td:
db_file = os.path.join(td, 'jupyterhub.sqlite') db_file = os.path.join(td, 'jupyterhub.sqlite')
tokens = { tokens = {
@@ -64,8 +63,8 @@ def test_init_tokens():
'also-super-secret': 'gordon', 'also-super-secret': 'gordon',
'boagasdfasdf': 'chell', 'boagasdfasdf': 'chell',
} }
app = MockHub(db_file=db_file, api_tokens=tokens) app = MockHub(db_url=db_file, api_tokens=tokens)
app.initialize([]) io_loop.run_sync(lambda : app.initialize([]))
db = app.db db = app.db
for token, username in tokens.items(): for token, username in tokens.items():
api_token = orm.APIToken.find(db, token) api_token = orm.APIToken.find(db, token)
@@ -74,8 +73,8 @@ def test_init_tokens():
assert user.name == username assert user.name == username
# simulate second startup, reloading same tokens: # simulate second startup, reloading same tokens:
app = MockHub(db_file=db_file, api_tokens=tokens) app = MockHub(db_url=db_file, api_tokens=tokens)
app.initialize([]) io_loop.run_sync(lambda : app.initialize([]))
db = app.db db = app.db
for token, username in tokens.items(): for token, username in tokens.items():
api_token = orm.APIToken.find(db, token) api_token = orm.APIToken.find(db, token)
@@ -85,9 +84,9 @@ def test_init_tokens():
# don't allow failed token insertion to create users: # don't allow failed token insertion to create users:
tokens['short'] = 'gman' tokens['short'] = 'gman'
app = MockHub(db_file=db_file, api_tokens=tokens) app = MockHub(db_url=db_file, api_tokens=tokens)
# with pytest.raises(ValueError): with pytest.raises(ValueError):
app.initialize([]) io_loop.run_sync(lambda : app.initialize([]))
assert orm.User.find(app.db, 'gman') is None assert orm.User.find(app.db, 'gman') is None