Skip to content

Commit 16427e2

Browse files
committed
Add tests for vectordb core
1 parent dcfde3c commit 16427e2

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed

test/backend/test_config_service.py

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,31 @@
4444
patch('nexent.storage.minio_config.MinIOStorageConfig.validate', lambda self: None).start()
4545
patch('backend.database.client.MinioClient', return_value=minio_client_mock).start()
4646

47+
# Create stub vector database modules to satisfy imports
48+
vector_db_module = types.ModuleType("nexent.vector_database")
49+
vector_db_module.__path__ = [] # Mark as package
50+
vector_db_base_module = types.ModuleType("nexent.vector_database.base")
51+
52+
class MockVectorDatabaseCore:
53+
def __init__(self, *args, **kwargs):
54+
pass
55+
56+
vector_db_base_module.VectorDatabaseCore = MockVectorDatabaseCore
57+
58+
vector_db_es_module = types.ModuleType("nexent.vector_database.elasticsearch_core")
59+
60+
class MockElasticSearchCore:
61+
def __init__(self, *args, **kwargs):
62+
pass
63+
64+
vector_db_es_module.ElasticSearchCore = MockElasticSearchCore
65+
66+
sys.modules['nexent.vector_database'] = vector_db_module
67+
sys.modules['nexent.vector_database.base'] = vector_db_base_module
68+
sys.modules['nexent.vector_database.elasticsearch_core'] = vector_db_es_module
69+
setattr(vector_db_module, "base", vector_db_base_module)
70+
setattr(vector_db_module, "elasticsearch_core", vector_db_es_module)
71+
4772
# Pre-inject a stubbed base_app to avoid import side effects
4873
backend_pkg = types.ModuleType("backend")
4974
apps_pkg = types.ModuleType("backend.apps")

0 commit comments

Comments
 (0)