use User wrappers in test_spawner

This commit is contained in:
Min RK
2017-04-05 16:12:29 +02:00
parent 1f2218c875
commit c7ea106675

View File

@@ -16,6 +16,7 @@ import pytest
import requests import requests
from tornado import gen from tornado import gen
from ..user import User
from .. import spawner as spawnermod from .. import spawner as spawnermod
from ..spawner import LocalProcessSpawner from ..spawner import LocalProcessSpawner
from .. import orm from .. import orm
@@ -42,7 +43,7 @@ def setup():
def new_spawner(db, **kwargs): def new_spawner(db, **kwargs):
kwargs.setdefault('cmd', [sys.executable, '-c', _echo_sleep]) kwargs.setdefault('cmd', [sys.executable, '-c', _echo_sleep])
kwargs.setdefault('user', db.query(orm.User).first()) kwargs.setdefault('user', User(db.query(orm.User).first(), {}))
kwargs.setdefault('hub', db.query(orm.Hub).first()) kwargs.setdefault('hub', db.query(orm.Hub).first())
kwargs.setdefault('notebook_dir', os.getcwd()) kwargs.setdefault('notebook_dir', os.getcwd())
kwargs.setdefault('default_url', '/user/{username}/lab') kwargs.setdefault('default_url', '/user/{username}/lab')