opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-07-21 10:38:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: use sqlite3 session in test_plugin_tool (#38781)

Browse Source
This commit is contained in:
Asuka Minato 2026-07-13 12:03:12 +09:00 committed by GitHub
parent 4c1c92eb2f
commit b893b53f0a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/tests/unit_tests/core/tools/test_plugin_tool.py
View File

@ -1,6 +1,9 @@
from __future__ import annotations
from unittest.mock import MagicMock, Mock, patch
from unittest.mock import Mock, patch
import pytest
from sqlalchemy.orm import Session
from core.app.entities.app_invoke_entities import InvokeFrom
from core.tools.__base.tool_runtime import ToolRuntime
@ -37,7 +40,8 @@ def _build_plugin_tool(*, has_runtime_parameters: bool) -> PluginTool:
)
def test_plugin_tool_invoke_and_fork_runtime():
@pytest.mark.parametrize("sqlite_session", [()], indirect=True)
def test_plugin_tool_invoke_and_fork_runtime(sqlite_session: Session):
tool = _build_plugin_tool(has_runtime_parameters=False)
manager = Mock()
manager.invoke.return_value = iter([tool.create_text_message("ok")])
@ -47,7 +51,7 @@ def test_plugin_tool_invoke_and_fork_runtime():
"core.tools.plugin_tool.tool.convert_parameters_to_plugin_format",
return_value={"converted": 1},
):
messages = list(tool.invoke(session=MagicMock(), user_id="user-1", tool_parameters={"raw": 1}))
messages = list(tool.invoke(session=sqlite_session, user_id="user-1", tool_parameters={"raw": 1}))
assert [m.message.text for m in messages] == ["ok"]
manager.invoke.assert_called_once()