From 39de931555ce7919a6697e3d49c1ab8013d5916b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9C=A8=E4=B9=8B=E6=9C=AC=E6=BE=AA?=
 <kinomotomiovo@gmail.com>
Date: Thu, 26 Feb 2026 02:24:58 +0800
Subject: [PATCH] test: migrate restore_archived_workflow_run SQL tests to
 testcontainers (#32590)

Co-authored-by: KinomotoMio <200703522+KinomotoMio@users.noreply.github.com>
---
 .../test_restore_archived_workflow_run.py     | 53 +++++++++++++++++++
 .../test_restore_archived_workflow_run.py     | 28 ----------
 2 files changed, 53 insertions(+), 28 deletions(-)
 create mode 100644 api/tests/test_containers_integration_tests/services/test_restore_archived_workflow_run.py

diff --git a/api/tests/test_containers_integration_tests/services/test_restore_archived_workflow_run.py b/api/tests/test_containers_integration_tests/services/test_restore_archived_workflow_run.py
new file mode 100644
index 0000000000..ba4310e22e
--- /dev/null
+++ b/api/tests/test_containers_integration_tests/services/test_restore_archived_workflow_run.py
@@ -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
diff --git a/api/tests/unit_tests/services/test_restore_archived_workflow_run.py b/api/tests/unit_tests/services/test_restore_archived_workflow_run.py
index 68aa8c0fe1..a214ecf728 100644
--- a/api/tests/unit_tests/services/test_restore_archived_workflow_run.py
+++ b/api/tests/unit_tests/services/test_restore_archived_workflow_run.py
@@ -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()