mirror of
https://github.com/jupyterhub/the-littlest-jupyterhub.git
synced 2025-12-18 21:54:05 +08:00
28 lines
646 B
Python
28 lines
646 B
Python
|
"""pytest fixtures"""
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
from pytest import fixture
|
||
|
|
|
||
|
|
from tljh.config import CONFIG_FILE, reload_component
|
||
|
|
|
||
|
|
|
||
|
|
@fixture
|
||
|
|
def preserve_config(request):
|
||
|
|
"""Fixture to save and restore config around tests"""
|
||
|
|
if os.path.exists(CONFIG_FILE):
|
||
|
|
with open(CONFIG_FILE) as f:
|
||
|
|
save_config = f.read()
|
||
|
|
else:
|
||
|
|
save_config = None
|
||
|
|
try:
|
||
|
|
yield
|
||
|
|
finally:
|
||
|
|
if save_config:
|
||
|
|
with open(CONFIG_FILE, "w") as f:
|
||
|
|
f.write(save_config)
|
||
|
|
elif os.path.exists(CONFIG_FILE):
|
||
|
|
os.remove(CONFIG_FILE)
|
||
|
|
reload_component("hub")
|
||
|
|
reload_component("proxy")
|