mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-24 17:15:05 +08:00
90 lines
2.1 KiB
Python
90 lines
2.1 KiB
Python
import sys
|
|
from typing import TYPE_CHECKING, Set
|
|
|
|
import pytest
|
|
from nonebug import App
|
|
|
|
if TYPE_CHECKING:
|
|
from nonebot.plugin import Plugin
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_load_plugin(load_plugin: Set["Plugin"]):
|
|
import nonebot
|
|
|
|
loaded_plugins = set(
|
|
plugin for plugin in nonebot.get_loaded_plugins() if not plugin.parent_plugin
|
|
)
|
|
assert loaded_plugins == load_plugin
|
|
plugin = nonebot.get_plugin("export")
|
|
assert plugin
|
|
assert plugin.module_name == "plugins.export"
|
|
assert "plugins.export" in sys.modules
|
|
|
|
try:
|
|
nonebot.load_plugin("plugins.export")
|
|
assert False
|
|
except RuntimeError:
|
|
assert True
|
|
|
|
assert nonebot.load_plugin("some_plugin_not_exist") is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_require_loaded(app: App, monkeypatch: pytest.MonkeyPatch):
|
|
import nonebot
|
|
|
|
def _patched_find(name: str):
|
|
assert False
|
|
|
|
monkeypatch.setattr("nonebot.plugin.load._find_manager_by_name", _patched_find)
|
|
|
|
nonebot.load_plugin("plugins.export")
|
|
|
|
nonebot.require("plugins.export")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_require_not_loaded(app: App, monkeypatch: pytest.MonkeyPatch):
|
|
import nonebot
|
|
from nonebot.plugin import _managers
|
|
from nonebot.plugin.manager import PluginManager
|
|
|
|
m = PluginManager(["plugins.export"])
|
|
_managers.append(m)
|
|
|
|
origin_load = PluginManager.load_plugin
|
|
|
|
def _patched_load(self: PluginManager, name: str):
|
|
assert self is m
|
|
return origin_load(self, name)
|
|
|
|
monkeypatch.setattr(PluginManager, "load_plugin", _patched_load)
|
|
|
|
nonebot.require("plugins.export")
|
|
|
|
assert len(_managers) == 1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_require_not_declared(app: App):
|
|
import nonebot
|
|
from nonebot.plugin import _managers
|
|
|
|
nonebot.require("plugins.export")
|
|
|
|
assert len(_managers) == 1
|
|
assert _managers[-1].plugins == {"plugins.export"}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_require_not_found(app: App):
|
|
import nonebot
|
|
from nonebot.plugin import _managers
|
|
|
|
try:
|
|
nonebot.require("some_plugin_not_exist")
|
|
assert False
|
|
except RuntimeError:
|
|
assert True
|