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

avoid creating duplicate users in test_api

Browse Source 
now that we check!
This commit is contained in:
Min RK
2016-07-28 16:27:32 +02:00
parent fd3b959771
commit dfa13cb2c5
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
jupyterhub/tests/test_api.py
View File

@@ -42,8 +42,13 @@ def find_user(db, name):
return db.query(orm.User).filter(orm.User.name==name).first() return db.query(orm.User).filter(orm.User.name==name).first()
def add_user(db, app=None, **kwargs): def add_user(db, app=None, **kwargs):
orm_user = orm.User(**kwargs) orm_user = find_user(db, name=kwargs.get('name'))
db.add(orm_user) if orm_user is None:
orm_user = orm.User(**kwargs)
db.add(orm_user)
else:
for attr, value in kwargs.items():
setattr(orm_user, attr, value)
db.commit() db.commit()
if app: if app:
user = app.users[orm_user.id] = User(orm_user, app.tornado_settings) user = app.users[orm_user.id] = User(orm_user, app.tornado_settings)