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

load upgrade_db_url in test

Browse Source
This commit is contained in:
Min RK
2017-09-28 11:27:58 +02:00
parent 2625229847
commit bf3fa30a01
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
jupyterhub/tests/test_db.py
View File

@@ -4,6 +4,7 @@ import shutil
import pytest import pytest
from pytest import raises from pytest import raises
from traitlets.config import Config
from ..dbutil import upgrade from ..dbutil import upgrade
from ..app import NewToken, UpgradeDB, JupyterHub from ..app import NewToken, UpgradeDB, JupyterHub
@@ -29,9 +30,11 @@ def test_upgrade_entrypoint(tmpdir):
if not db_url: if not db_url:
# default: sqlite # default: sqlite
db_url = generate_old_db(str(tmpdir)) db_url = generate_old_db(str(tmpdir))
cfg = Config()
cfg.JupyterHub.db_url = db_url
tmpdir.chdir() tmpdir.chdir()
tokenapp = NewToken() tokenapp = NewToken(config=cfg)
tokenapp.initialize(['kaylee']) tokenapp.initialize(['kaylee'])
with raises(SystemExit): with raises(SystemExit):
tokenapp.start() tokenapp.start()
@@ -40,7 +43,7 @@ def test_upgrade_entrypoint(tmpdir):
sqlite_files = glob(os.path.join(str(tmpdir), 'jupyterhub.sqlite*')) sqlite_files = glob(os.path.join(str(tmpdir), 'jupyterhub.sqlite*'))
assert len(sqlite_files) == 1 assert len(sqlite_files) == 1
upgradeapp = UpgradeDB() upgradeapp = UpgradeDB(config=cfg)
yield upgradeapp.initialize([]) yield upgradeapp.initialize([])
upgradeapp.start() upgradeapp.start()