♻️ Add Unitest and fix node.js server env read

Author: Phinease

test/backend/test_llm_integration.py

@@ -5,11 +5,52 @@
 import pytest
 import sys
 import os
+import types
 from unittest.mock import patch, MagicMock
 
 # Add backend to path
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..', 'backend'))
 
+# Mock database.client and MinioClient before any imports to avoid MinIO initialization
+class _MinioClient:
+    pass
+
+if "database.client" not in sys.modules:
+    database_client_mod = types.ModuleType("database.client")
+    database_client_mod.MinioClient = _MinioClient
+    sys.modules["database.client"] = database_client_mod
+
+# Mock backend.database.client as well
+if "backend.database.client" not in sys.modules:
+    backend_db_client_mod = types.ModuleType("backend.database.client")
+    backend_db_client_mod.MinioClient = _MinioClient
+    sys.modules["backend.database.client"] = backend_db_client_mod
+
+# Ensure database module exists as a package (needs __path__ attribute)
+if "database" not in sys.modules:
+    database_mod = types.ModuleType("database")
+    database_mod.__path__ = []  # Make it a package
+    sys.modules["database"] = database_mod
+
+# Mock database.model_management_db module to avoid MinIO initialization
+if "database.model_management_db" not in sys.modules:
+    model_mgmt_db_mod = types.ModuleType("database.model_management_db")
+    model_mgmt_db_mod.get_model_by_model_id = MagicMock(return_value=None)
+    sys.modules["database.model_management_db"] = model_mgmt_db_mod
+    setattr(sys.modules["database"], "model_management_db", model_mgmt_db_mod)
+
+# Mock database.tenant_config_db to avoid import errors
+if "database.tenant_config_db" not in sys.modules:
+    tenant_config_db_mod = types.ModuleType("database.tenant_config_db")
+    # Mock all functions that config_utils imports
+    tenant_config_db_mod.delete_config_by_tenant_config_id = MagicMock()
+    tenant_config_db_mod.get_all_configs_by_tenant_id = MagicMock()
+    tenant_config_db_mod.get_single_config_info = MagicMock()
+    tenant_config_db_mod.insert_config = MagicMock()
+    tenant_config_db_mod.update_config_by_tenant_config_id_and_data = MagicMock()
+    sys.modules["database.tenant_config_db"] = tenant_config_db_mod
+    setattr(sys.modules["database"], "tenant_config_db", tenant_config_db_mod)
+
 from utils.document_vector_utils import summarize_document, summarize_cluster
 
 
@@ -33,19 +74,21 @@ def test_summarize_document_with_llm_params_no_config(self):
         content = "This is a test document with some content about machine learning and AI."
         filename = "test_doc.txt"
 
-        # Mock get_model_by_model_id to return None (no config found) to avoid MinIO initialization
-        # Patch at the source module since it's imported inside the function
-        with patch('database.model_management_db.get_model_by_model_id', return_value=None):
-            # Test with model_id and tenant_id but no actual LLM call (will fallback due to missing config)
-            result = summarize_document(
-                content, filename, language="zh", max_words=50, 
-                model_id=1, tenant_id="test_tenant"
-            )
-            
-            # Should return placeholder summary when model config not found (fallback behavior)
-            assert "[Document Summary: test_doc.txt]" in result
-            assert "max 50 words" in result
-            assert "Content:" in result
+        # Mock get_model_by_model_id to return None (no config found)
+        # Use the already mocked module and just ensure it returns None
+        import database.model_management_db as model_mgmt_db
+        model_mgmt_db.get_model_by_model_id = MagicMock(return_value=None)
+        
+        # Test with model_id and tenant_id but no actual LLM call (will fallback due to missing config)
+        result = summarize_document(
+            content, filename, language="zh", max_words=50, 
+            model_id=1, tenant_id="test_tenant"
+        )
+        
+        # Should return placeholder summary when model config not found (fallback behavior)
+        assert "[Document Summary: test_doc.txt]" in result
+        assert "max 50 words" in result
+        assert "Content:" in result
 
     def test_summarize_cluster_without_llm(self):
         """Test cluster summarization without LLM (fallback mode)"""
@@ -69,18 +112,20 @@ def test_summarize_cluster_with_llm_params_no_config(self):
             "Document 2 discusses neural networks and deep learning."
         ]
 
-        # Mock get_model_by_model_id to return None (no config found) to avoid MinIO initialization
-        # Patch at the source module since it's imported inside the function
-        with patch('database.model_management_db.get_model_by_model_id', return_value=None):
-            result = summarize_cluster(
-                document_summaries, language="zh", max_words=100,
-                model_id=1, tenant_id="test_tenant"
-            )
-            
-            # Should return placeholder summary when model config not found (fallback behavior)
-            assert "[Cluster Summary]" in result
-            assert "max 100 words" in result
-            assert "Based on 2 documents" in result
+        # Mock get_model_by_model_id to return None (no config found)
+        # Use the already mocked module and just ensure it returns None
+        import database.model_management_db as model_mgmt_db
+        model_mgmt_db.get_model_by_model_id = MagicMock(return_value=None)
+        
+        result = summarize_cluster(
+            document_summaries, language="zh", max_words=100,
+            model_id=1, tenant_id="test_tenant"
+        )
+        
+        # Should return placeholder summary when model config not found (fallback behavior)
+        assert "[Cluster Summary]" in result
+        assert "max 100 words" in result
+        assert "Based on 2 documents" in result
 
     def test_summarize_document_english(self):
         """Test document summarization in English"""