fixup test_orm with new User wrapper

user.db is evaluated immediately, rather than on first request,
which means we can't do User(orm.User) before adding the orm.User to the db
This commit is contained in:
Min RK
2018-02-06 17:02:23 +01:00
parent d5f0137052
commit e9573b6e24
2 changed files with 14 additions and 15 deletions

View File

@@ -136,13 +136,10 @@ class User:
log = app_log
settings = None
def __init__(self, orm_user, settings=None, **kwargs):
self.orm_user = orm_user
self.db = inspect(orm_user).session
def __init__(self, orm_user, settings=None, db=None):
self.db = db or inspect(orm_user).session
self.settings = settings or {}
for key, attr in kwargs:
print('setting', key, attr)
setattr(self, key, attr)
self.orm_user = orm_user
self.allow_named_servers = self.settings.get('allow_named_servers', False)
@@ -225,7 +222,7 @@ class User:
@property
def spawner(self):
return self.spawners['']
@spawner.setter
def spawner(self, spawner):
self.spawners[''] = spawner