opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-03-11 20:01:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: migrate restore_archived_workflow_run SQL tests to testcontainers (#32590)

Browse Source 
Co-authored-by: KinomotoMio <200703522+KinomotoMio@users.noreply.github.com>
This commit is contained in:
木之本澪 2026-02-26 02:24:58 +08:00 committed by GitHub
parent 05c827606b
commit 39de931555
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 53 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
api/tests/test_containers_integration_tests/services/test_restore_archived_workflow_run.py Normal file
View File

@ -0,0 +1,53 @@
"""
Testcontainers integration tests for workflow run restore functionality.
"""
from uuid import uuid4
from sqlalchemy import select
from models.workflow import WorkflowPause
from services.retention.workflow_run.restore_archived_workflow_run import WorkflowRunRestore
class TestWorkflowRunRestore:
"""Tests for the WorkflowRunRestore class."""
def test_restore_table_records_returns_rowcount(self, db_session_with_containers):
"""Restore should return inserted rowcount."""
restore = WorkflowRunRestore()
record_id = str(uuid4())
records = [
{
"id": record_id,
"workflow_id": str(uuid4()),
"workflow_run_id": str(uuid4()),
"state_object_key": f"workflow-state-{uuid4()}.json",
"created_at": "2024-01-01T00:00:00",
"updated_at": "2024-01-01T00:00:00",
}
]
restored = restore._restore_table_records(
db_session_with_containers,
"workflow_pauses",
records,
schema_version="1.0",
)
assert restored == 1
restored_pause = db_session_with_containers.scalar(select(WorkflowPause).where(WorkflowPause.id == record_id))
assert restored_pause is not None
def test_restore_table_records_unknown_table(self, db_session_with_containers):
"""Unknown table names should be ignored gracefully."""
restore = WorkflowRunRestore()
restored = restore._restore_table_records(
db_session_with_containers,
"unknown_table",
[{"id": str(uuid4())}],
schema_version="1.0",
)
assert restored == 0

28
api/tests/unit_tests/services/test_restore_archived_workflow_run.py
View File

@ -3,7 +3,6 @@ Unit tests for workflow run restore functionality.
"""
from datetime import datetime
from unittest.mock import MagicMock
class TestWorkflowRunRestore:
@ -36,30 +35,3 @@ class TestWorkflowRunRestore:
assert result["created_at"].year == 2024
assert result["created_at"].month == 1
assert result["name"] == "test"
def test_restore_table_records_returns_rowcount(self):
"""Restore should return inserted rowcount."""
from services.retention.workflow_run.restore_archived_workflow_run import WorkflowRunRestore
session = MagicMock()
session.execute.return_value = MagicMock(rowcount=2)
restore = WorkflowRunRestore()
records = [{"id": "p1", "workflow_run_id": "r1", "created_at": "2024-01-01T00:00:00"}]
restored = restore._restore_table_records(session, "workflow_pauses", records, schema_version="1.0")
assert restored == 2
session.execute.assert_called_once()
def test_restore_table_records_unknown_table(self):
"""Unknown table names should be ignored gracefully."""
from services.retention.workflow_run.restore_archived_workflow_run import WorkflowRunRestore
session = MagicMock()
restore = WorkflowRunRestore()
restored = restore._restore_table_records(session, "unknown_table", [{"id": "x1"}], schema_version="1.0")
assert restored == 0
session.execute.assert_not_called()