fix: Fix test_get_and_cache_all_types_dict (#5054)

src/backend/base/langflow/main.py

@@ -101,7 +101,7 @@ def get_lifespan(*, fix_migration=False, version=None):
             rprint("[bold green]Starting Langflow...[/bold green]")
         try:
             await initialize_services(fix_migration=fix_migration)
-            await asyncio.to_thread(setup_llm_caching)
+            setup_llm_caching()
             await initialize_super_user_if_needed()
             all_types_dict = await get_and_cache_all_types_dict(get_settings_service())
             await asyncio.to_thread(create_or_update_starter_projects, all_types_dict)

src/backend/tests/conftest.py

@@ -57,6 +57,7 @@ def blockbuster():
         for func in bb.functions:
             if func.startswith("sqlite3."):
                 bb.functions[func].deactivate()
+        bb.functions["threading.Lock.acquire"].deactivate()
         yield bb
 
 

src/backend/tests/performance/test_server_init.py

@@ -25,7 +25,7 @@ async def test_initialize_services():
     from langflow.services.utils import initialize_services
 
     await initialize_services(fix_migration=False)
-    settings_service = await asyncio.to_thread(get_settings_service)
+    settings_service = get_settings_service()
     assert "test_performance.db" in settings_service.settings.database_url
 
 
@@ -34,8 +34,8 @@ async def test_setup_llm_caching():
     """Benchmark LLM caching setup."""
     from langflow.interface.utils import setup_llm_caching
 
-    await asyncio.to_thread(setup_llm_caching)
-    settings_service = await asyncio.to_thread(get_settings_service)
+    setup_llm_caching()
+    settings_service = get_settings_service()
     assert "test_performance.db" in settings_service.settings.database_url
 
 
@@ -47,7 +47,7 @@ async def test_initialize_super_user():
 
     await initialize_services(fix_migration=False)
     await initialize_super_user_if_needed()
-    settings_service = await asyncio.to_thread(get_settings_service)
+    settings_service = get_settings_service()
     assert "test_performance.db" in settings_service.settings.database_url
 
 
@@ -56,9 +56,9 @@ async def test_get_and_cache_all_types_dict():
     """Benchmark get_and_cache_all_types_dict function."""
     from langflow.interface.types import get_and_cache_all_types_dict
 
-    settings_service = await asyncio.to_thread(get_settings_service)
-    result = await asyncio.to_thread(get_and_cache_all_types_dict, settings_service)
-    assert result is not None
+    settings_service = get_settings_service()
+    result = await get_and_cache_all_types_dict(settings_service)
+    assert "vectorstores" in result
     assert "test_performance.db" in settings_service.settings.database_url
 
 
@@ -70,7 +70,7 @@ async def test_create_starter_projects():
     from langflow.services.utils import initialize_services
 
     await initialize_services(fix_migration=False)
-    settings_service = await asyncio.to_thread(get_settings_service)
+    settings_service = get_settings_service()
     types_dict = await get_and_cache_all_types_dict(settings_service)
     await asyncio.to_thread(create_or_update_starter_projects, types_dict)
     assert "test_performance.db" in settings_service.settings.database_url
@@ -82,5 +82,5 @@ async def test_load_flows():
     from langflow.initial_setup.setup import load_flows_from_directory
 
     await load_flows_from_directory()
-    settings_service = await asyncio.to_thread(get_settings_service)
+    settings_service = get_settings_service()
     assert "test_performance.db" in settings_service.settings.database_url