mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-12-18 09:25:46 +08:00
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
import os
|
|
import threading
|
|
from pathlib import Path
|
|
from typing import TYPE_CHECKING, Set, Generator
|
|
|
|
import pytest
|
|
from nonebug import NONEBOT_INIT_KWARGS
|
|
from werkzeug.serving import BaseWSGIServer, make_server
|
|
|
|
import nonebot
|
|
from nonebot.drivers import URL
|
|
from fake_server import request_handler
|
|
|
|
os.environ["CONFIG_FROM_ENV"] = '{"test": "test"}'
|
|
os.environ["CONFIG_OVERRIDE"] = "new"
|
|
|
|
if TYPE_CHECKING:
|
|
from nonebot.plugin import Plugin
|
|
|
|
|
|
def pytest_configure(config: pytest.Config) -> None:
|
|
config.stash[NONEBOT_INIT_KWARGS] = {"config_from_init": "init"}
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def load_plugin(nonebug_init: None) -> Set["Plugin"]:
|
|
# preload global plugins
|
|
return nonebot.load_plugins(str(Path(__file__).parent / "plugins"))
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def load_builtin_plugin(nonebug_init: None) -> Set["Plugin"]:
|
|
# preload builtin plugins
|
|
return nonebot.load_builtin_plugins("echo", "single_session")
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def server() -> Generator[BaseWSGIServer, None, None]:
|
|
server = make_server("127.0.0.1", 0, app=request_handler)
|
|
thread = threading.Thread(target=server.serve_forever)
|
|
thread.start()
|
|
try:
|
|
yield server
|
|
finally:
|
|
server.shutdown()
|
|
thread.join()
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def server_url(server: BaseWSGIServer) -> URL:
|
|
return URL(f"http://{server.host}:{server.port}")
|