youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 18:08:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix unittests

Browse Source
This commit is contained in:
hjlarry
2026-01-19 21:53:56 +08:00
parent 060c7f2b45
commit faa88dc2f3
1 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
api/tests/unit_tests/services/test_workflow_comment_service.py
View File

@ -1,4 +1,4 @@
from unittest.mock import Mock, patch from unittest.mock import MagicMock, Mock, patch
import pytest import pytest
from werkzeug.exceptions import Forbidden, NotFound from werkzeug.exceptions import Forbidden, NotFound
@ -10,10 +10,13 @@ from services.workflow_comment_service import WorkflowCommentService
@pytest.fixture @pytest.fixture
def mock_session(monkeypatch: pytest.MonkeyPatch) -> Mock: def mock_session(monkeypatch: pytest.MonkeyPatch) -> Mock:
session = Mock() session = Mock()
context_manager = Mock() context_manager = MagicMock()
context_manager.__enter__.return_value = session context_manager.__enter__.return_value = session
context_manager.__exit__.return_value = False context_manager.__exit__.return_value = False
mock_db = MagicMock()
mock_db.engine = Mock()
monkeypatch.setattr(service_module, "Session", Mock(return_value=context_manager)) monkeypatch.setattr(service_module, "Session", Mock(return_value=context_manager))
monkeypatch.setattr(service_module, "db", mock_db)
return session return session
@ -92,10 +95,7 @@ class TestWorkflowCommentService:
existing_mentions = [Mock(), Mock()] existing_mentions = [Mock(), Mock()]
mock_session.scalars.return_value = _mock_scalars(existing_mentions) mock_session.scalars.return_value = _mock_scalars(existing_mentions)
with ( with patch.object(service_module, "uuid_value", side_effect=[True, False]):
patch.object(service_module, "WorkflowCommentMention", return_value=Mock()),
patch.object(service_module, "uuid_value", side_effect=[True, False]),
):
result = WorkflowCommentService.update_comment( result = WorkflowCommentService.update_comment(
tenant_id="tenant-1", tenant_id="tenant-1",
app_id="app-1", app_id="app-1",
@ -211,10 +211,7 @@ class TestWorkflowCommentService:
mock_session.get.return_value = reply mock_session.get.return_value = reply
mock_session.scalars.return_value = _mock_scalars([Mock()]) mock_session.scalars.return_value = _mock_scalars([Mock()])
with ( with patch.object(service_module, "uuid_value", side_effect=[True, False]):
patch.object(service_module, "WorkflowCommentMention", return_value=Mock()),
patch.object(service_module, "uuid_value", side_effect=[True, False]),
):
result = WorkflowCommentService.update_reply( result = WorkflowCommentService.update_reply(
reply_id="reply-1", reply_id="reply-1",
user_id="owner", user_id="owner",