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

refactor(test): replace SimpleNamespace with typed mocks in schedule service tests (#38393)

Browse Source
This commit is contained in:
ojasarora.eth 2026-07-07 15:36:17 +01:00 committed by GitHub
parent 64aa142681
commit 09c5c5e5ed
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
api/tests/unit_tests/services/test_schedule_service.py
View File

@ -1,7 +1,7 @@
import json
import unittest
from datetime import UTC, datetime
from types import SimpleNamespace
from typing import Any, cast
from typing import Any
from unittest.mock import MagicMock, Mock
import pytest
@ -11,7 +11,7 @@ from core.trigger.constants import TRIGGER_SCHEDULE_NODE_TYPE
from core.workflow.nodes.trigger_schedule.entities import VisualConfig
from core.workflow.nodes.trigger_schedule.exc import ScheduleConfigError
from libs.schedule_utils import calculate_next_run_at, convert_12h_to_24h
from models.workflow import Workflow
from models.workflow import Workflow, WorkflowType
from services.trigger.schedule_service import ScheduleService
@ -503,7 +503,22 @@ def session_mock() -> MagicMock:
def _workflow(**kwargs: Any) -> Workflow:
return cast(Workflow, SimpleNamespace(**kwargs))
graph_dict = kwargs.pop("graph_dict", {})
workflow = Workflow.new(
tenant_id="tenant-1",
app_id="app-1",
type=WorkflowType.WORKFLOW,
version="draft",
graph=json.dumps(graph_dict),
features="{}",
created_by="account-1",
environment_variables=[],
conversation_variables=[],
rag_pipeline_variables=[],
)
for key, value in kwargs.items():
setattr(workflow, key, value)
return workflow
def test_to_schedule_config_should_build_from_cron_mode() -> None: