Skip to content

Commit 3755b54

Browse files
committed
🔨 use arm host for unitest runner and update monitor auto disable
1 parent 04cbab6 commit 3755b54

File tree

2 files changed

+7
-16
lines changed

2 files changed

+7
-16
lines changed

‎test/backend/app/test_agent_app.py‎

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,8 @@
1919
sys.modules['boto3'] = boto3_mock
2020

2121
# Import target endpoints with all external dependencies patched
22-
with patch('backend.database.client.MinioClient') as minio_mock:
22+
with patch('backend.database.client.MinioClient') as minio_mock, \
23+
patch('elasticsearch.Elasticsearch', return_value=MagicMock()) as es_mock:
2324
minio_mock.return_value = MagicMock()
2425

2526
from apps.agent_app import router
@@ -28,9 +29,7 @@
2829

2930
patches = [
3031
# Mock database sessions
31-
patch('backend.database.client.get_db_session', return_value=Mock()),
32-
# Mock Elasticsearch to prevent connection errors
33-
patch('elasticsearch.Elasticsearch', return_value=Mock())
32+
patch('backend.database.client.get_db_session', return_value=Mock())
3433
]
3534

3635
for p in patches:

‎test/backend/services/test_agent_service.py‎

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,9 @@
1414
boto3_mock = MagicMock()
1515
sys.modules['boto3'] = boto3_mock
1616

17-
# Mock MinioClient before importing backend modules that might initialize it
18-
with patch('backend.database.client.MinioClient') as minio_mock:
17+
# Mock external dependencies before importing backend modules that might initialize them
18+
with patch('backend.database.client.MinioClient') as minio_mock, \
19+
patch('elasticsearch.Elasticsearch', return_value=MagicMock()) as es_mock:
1920
minio_mock.return_value = MagicMock()
2021

2122
import backend.services.agent_service as agent_service
@@ -43,17 +44,8 @@
4344
)
4445
from consts.model import ExportAndImportAgentInfo, ExportAndImportDataFormat, MCPInfo, AgentRequest
4546

46-
# Mock Elasticsearch
47+
# Mock Elasticsearch (already done in the import section above, but keeping for reference)
4748
elasticsearch_client_mock = MagicMock()
48-
patch('elasticsearch._sync.client.Elasticsearch',
49-
return_value=elasticsearch_client_mock).start()
50-
patch('elasticsearch.Elasticsearch',
51-
return_value=elasticsearch_client_mock).start()
52-
53-
# Mock ElasticSearchCore
54-
elasticsearch_core_mock = MagicMock()
55-
patch('sdk.nexent.vector_database.elasticsearch_core.ElasticSearchCore',
56-
return_value=elasticsearch_core_mock).start()
5749

5850

5951
# Mock memory-related modules

0 commit comments

Comments
 (0)