fix the unit tests error of agent provider

This commit is contained in:
Willem Jiang
2026-04-26 22:09:12 +08:00
parent ac18b9c424
commit 60754f0c50
2 changed files with 8 additions and 8 deletions
+4 -4
View File
@@ -174,9 +174,9 @@ class TestGetCheckpointer:
with (
patch.dict(sys.modules, {"langgraph.checkpoint.sqlite": mock_module}),
patch("deerflow.agents.checkpointer.provider.ensure_sqlite_parent_dir") as mock_ensure,
patch("deerflow.runtime.checkpointer.provider.ensure_sqlite_parent_dir") as mock_ensure,
patch(
"deerflow.agents.checkpointer.provider.resolve_sqlite_conn_str",
"deerflow.runtime.checkpointer.provider.resolve_sqlite_conn_str",
return_value="/tmp/resolved/relative/test.db",
),
):
@@ -210,11 +210,11 @@ class TestGetCheckpointer:
with (
patch.dict(sys.modules, {"langgraph.checkpoint.sqlite": mock_module}),
patch(
"deerflow.agents.checkpointer.provider.ensure_sqlite_parent_dir",
"deerflow.runtime.checkpointer.provider.ensure_sqlite_parent_dir",
side_effect=record_ensure,
),
patch(
"deerflow.agents.checkpointer.provider.resolve_sqlite_conn_str",
"deerflow.runtime.checkpointer.provider.resolve_sqlite_conn_str",
return_value="/tmp/resolved/relative/test.db",
),
):