|
import dataclasses
|
|
from pathlib import Path
|
|
from typing import List
|
|
|
|
import pytest
|
|
|
|
import gradio
|
|
import gradio as gr
|
|
from gradio.cli.commands.reload import _setup_config
|
|
from gradio.http_server import Server
|
|
|
|
|
|
def build_demo():
|
|
with gr.Blocks() as demo:
|
|
gr.Textbox("")
|
|
|
|
return demo
|
|
|
|
|
|
@dataclasses.dataclass
|
|
class Config:
|
|
module_name: str
|
|
path: Path
|
|
watch_dirs: List[str]
|
|
demo_name: str
|
|
|
|
|
|
class TestReload:
|
|
@pytest.fixture(autouse=True)
|
|
def argv(self):
|
|
return ["demo/calculator/run.py"]
|
|
|
|
@pytest.fixture
|
|
def config(self, monkeypatch, argv) -> Config:
|
|
monkeypatch.setattr("sys.argv", ["gradio"] + argv)
|
|
name = argv[1].replace("--demo-name", "").strip() if len(argv) > 1 else "demo"
|
|
return Config(*_setup_config(argv[0], name))
|
|
|
|
@pytest.fixture(params=[{}])
|
|
def reloader(self, config):
|
|
reloader = Server(config)
|
|
reloader.should_exit = True
|
|
yield reloader
|
|
reloader.close()
|
|
|
|
def test_config_default_app(self, config):
|
|
assert config.module_name == "demo.calculator.run"
|
|
|
|
@pytest.mark.parametrize("argv", [["demo/calculator/run.py", "--demo-name test"]])
|
|
def test_config_custom_app(self, config):
|
|
assert config.module_name == "demo.calculator.run"
|
|
assert config.demo_name == "test"
|
|
|
|
def test_config_watch_gradio(self, config):
|
|
gradio_dir = str(Path(gradio.__file__).parent)
|
|
assert gradio_dir in config.watch_dirs
|
|
|
|
def test_config_watch_app(self, config):
|
|
demo_dir = str(Path("demo/calculator/run.py").resolve().parent)
|
|
assert demo_dir in config.watch_dirs
|
|
|
