From 60754f0c502ce2ec03899173252fe8bd1aa71819 Mon Sep 17 00:00:00 2001 From: Willem Jiang Date: Sun, 26 Apr 2026 22:09:12 +0800 Subject: [PATCH] fix the unit tests error of agent provider --- backend/tests/test_checkpointer.py | 8 ++++---- backend/tests/test_lead_agent_prompt.py | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/backend/tests/test_checkpointer.py b/backend/tests/test_checkpointer.py index 224beaee7..5a31cfb78 100644 --- a/backend/tests/test_checkpointer.py +++ b/backend/tests/test_checkpointer.py @@ -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", ), ): diff --git a/backend/tests/test_lead_agent_prompt.py b/backend/tests/test_lead_agent_prompt.py index 6817e7678..c63e32d99 100644 --- a/backend/tests/test_lead_agent_prompt.py +++ b/backend/tests/test_lead_agent_prompt.py @@ -84,7 +84,7 @@ def test_refresh_skills_system_prompt_cache_async_reloads_immediately(monkeypatc state = {"skills": [make_skill("first-skill")]} monkeypatch.setattr(prompt_module, "load_skills", lambda enabled_only=True: list(state["skills"])) - prompt_module._reset_skills_system_prompt_cache_state() + prompt_module.clear_skills_system_prompt_cache() try: prompt_module.warm_enabled_skills_cache() @@ -95,7 +95,7 @@ def test_refresh_skills_system_prompt_cache_async_reloads_immediately(monkeypatc assert [skill.name for skill in prompt_module._get_enabled_skills()] == ["second-skill"] finally: - prompt_module._reset_skills_system_prompt_cache_state() + prompt_module.clear_skills_system_prompt_cache() def test_clear_cache_does_not_spawn_parallel_refresh_workers(monkeypatch, tmp_path): @@ -137,7 +137,7 @@ def test_clear_cache_does_not_spawn_parallel_refresh_workers(monkeypatch, tmp_pa return [make_skill(f"skill-{current_call}")] monkeypatch.setattr(prompt_module, "load_skills", fake_load_skills) - prompt_module._reset_skills_system_prompt_cache_state() + prompt_module.clear_skills_system_prompt_cache() try: prompt_module.clear_skills_system_prompt_cache() @@ -151,7 +151,7 @@ def test_clear_cache_does_not_spawn_parallel_refresh_workers(monkeypatch, tmp_pa assert [skill.name for skill in prompt_module._get_enabled_skills()] == ["skill-2"] finally: release.set() - prompt_module._reset_skills_system_prompt_cache_state() + prompt_module.clear_skills_system_prompt_cache() def test_warm_enabled_skills_cache_logs_on_timeout(monkeypatch, caplog):