chore: update tests

tests/test_plugin_manager.py

@@ -0,0 +1,93 @@
+import pytest
+import os
+import shutil
+from astrbot.core.star.star_manager import PluginManager
+from astrbot.core.star.star_handler import star_handlers_registry
+from astrbot.core.star.star import star_registry
+from astrbot.core.star.context import Context
+from astrbot.core.config.astrbot_config import AstrBotConfig
+from astrbot.core.db.sqlite import SQLiteDatabase
+from asyncio import Queue
+
+@pytest.fixture
+def event_queue():
+    return Queue()
+
+@pytest.fixture
+def config():
+    return AstrBotConfig()
+
+@pytest.fixture
+def db():
+    return SQLiteDatabase("data/data_v3.db")
+
+@pytest.fixture
+def star_context(event_queue, config, db):
+    return Context(event_queue, config, db)
+
+@pytest.fixture
+def plugin_manager_pm(star_context, config):
+    return PluginManager(star_context, config)
+
+def test_plugin_manager_initialization(plugin_manager_pm: PluginManager):
+    assert plugin_manager_pm is not None
+    assert plugin_manager_pm.context is not None
+    assert plugin_manager_pm.config is not None
+
+def test_plugin_manager_reload(plugin_manager_pm: PluginManager):
+    success, err_message = plugin_manager_pm.reload()
+    assert success is True
+    assert err_message is None
+    assert len(star_handlers_registry) > 0 # package
+    
+@pytest.mark.asyncio
+async def test_plugin_crud(plugin_manager_pm: PluginManager):
+    '''测试插件安装和重载'''
+    os.makedirs("data/plugins", exist_ok=True)
+    test_repo = "https://github.com/Soulter/astrbot_plugin_essential"
+    plugin_path = await plugin_manager_pm.install_plugin(test_repo)
+    exists = False
+    for md in star_registry:
+        if md.name == "astrbot_plugin_essential":
+            exists = True
+            break
+    assert plugin_path is not None
+    assert os.path.exists(plugin_path)
+    assert exists is True, "插件 astrbot_plugin_essential 未成功载入" 
+    # shutil.rmtree(plugin_path)
+    
+    # install plugin which is not exists
+    with pytest.raises(Exception):
+        plugin_path = await plugin_manager_pm.install_plugin(test_repo + "haha")
+        
+    # update
+    await plugin_manager_pm.update_plugin("astrbot_plugin_essential")
+    
+    with pytest.raises(Exception):
+        await plugin_manager_pm.update_plugin("astrbot_plugin_essentialhaha")
+    
+    # uninstall
+    await plugin_manager_pm.uninstall_plugin("astrbot_plugin_essential")
+    assert not os.path.exists(plugin_path)
+    exists = False
+    for md in star_registry:
+        if md.name == "astrbot_plugin_essential":
+            exists = True
+            break
+    assert exists is False, "插件 astrbot_plugin_essential 未成功卸载"
+    exists = False
+    for md in star_handlers_registry:
+        if "astrbot_plugin_essential" in md.handler_module_path:
+            exists = True
+            break
+    assert exists is False, "插件 astrbot_plugin_essential 未成功卸载"
+    
+    with pytest.raises(Exception):
+        await plugin_manager_pm.uninstall_plugin("astrbot_plugin_essentialhaha")
+
+    # TODO: file installation
+
+
+
+
+