mirror of
https://github.com/langgenius/dify.git
synced 2026-11-11 05:41:09 +08:00
53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
import pytest
|
|
|
|
from controllers.openapi.app_release import passed_on_draft
|
|
from graphon.enums import WorkflowExecutionStatus
|
|
from models.workflow import Workflow, WorkflowRun, WorkflowType
|
|
|
|
_GRAPH = '{"nodes": [{"id": "start", "data": {"type": "start"}}], "edges": []}'
|
|
_OTHER_GRAPH = '{"nodes": [{"id": "llm", "data": {"type": "llm"}}], "edges": []}'
|
|
_RUN_STARTED = datetime(2026, 10, 15, 12, 0, 0)
|
|
|
|
|
|
def _draft(*, saved_at: datetime, graph: str = _GRAPH) -> Workflow:
|
|
draft = Workflow(
|
|
tenant_id="workspace-1",
|
|
app_id="app-1",
|
|
type=WorkflowType.WORKFLOW,
|
|
version=Workflow.VERSION_DRAFT,
|
|
graph=graph,
|
|
_features="{}",
|
|
created_by="account-1",
|
|
)
|
|
draft.updated_at = saved_at
|
|
return draft
|
|
|
|
|
|
def _run(*, status: WorkflowExecutionStatus = WorkflowExecutionStatus.SUCCEEDED, graph: str = _GRAPH) -> WorkflowRun:
|
|
run = WorkflowRun(status=status, graph=graph)
|
|
run.created_at = _RUN_STARTED
|
|
return run
|
|
|
|
|
|
def test_a_passing_run_on_the_unchanged_draft_counts() -> None:
|
|
assert passed_on_draft(_run(), _draft(saved_at=_RUN_STARTED - timedelta(minutes=1))) is True
|
|
|
|
|
|
def test_a_draft_saved_after_the_run_started_needs_a_new_test() -> None:
|
|
"""Environment variables, conversation variables and features live outside the run's graph."""
|
|
assert passed_on_draft(_run(), _draft(saved_at=_RUN_STARTED + timedelta(seconds=1))) is False
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"run",
|
|
[
|
|
pytest.param(None, id="no-run"),
|
|
pytest.param(_run(status=WorkflowExecutionStatus.FAILED), id="failed"),
|
|
pytest.param(_run(graph=_OTHER_GRAPH), id="other-graph"),
|
|
],
|
|
)
|
|
def test_no_passing_run_on_this_graph_does_not_count(run: WorkflowRun | None) -> None:
|
|
assert passed_on_draft(run, _draft(saved_at=_RUN_STARTED - timedelta(minutes=1))) is False
|