opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-11 23:18:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
Yansong Zhang 2026-04-15 13:39:29 +08:00
parent 43e4e161b5
commit f82f235d5a
1 changed files with 17 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
api/providers/vdb/vdb-tidb-on-qdrant/tests/unit_tests/test_tidb_on_qdrant_vector.py
View File

@ -193,19 +193,25 @@ class TestInitVectorEndpointSelection:
b.cluster_id = cluster_id
return b
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.current_app")
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.dify_config")
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.db")
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.qdrant_client.QdrantClient")
def test_uses_binding_endpoint_when_present(self, mock_qc, mock_db, mock_config, mock_app):
binding = self._make_binding(qdrant_endpoint="https://qdrant-custom.tidb.com")
mock_db.session.scalars.return_value.one_or_none.return_value = binding
def _setup_config(self, mock_config, mock_app):
mock_config.TIDB_ON_QDRANT_URL = "https://qdrant-global.tidb.com"
mock_config.TIDB_ON_QDRANT_CLIENT_TIMEOUT = 20
mock_config.TIDB_ON_QDRANT_GRPC_PORT = 6334
mock_config.TIDB_ON_QDRANT_GRPC_ENABLED = False
mock_config.QDRANT_REPLICATION_FACTOR = 1
mock_app.config = {"root_path": "/app"}
flask_config = MagicMock()
flask_config.root_path = "/app"
mock_app.config = flask_config
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.current_app")
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.dify_config")
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.select")
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.db")
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.qdrant_client.QdrantClient")
def test_uses_binding_endpoint_when_present(self, mock_qc, mock_db, mock_select, mock_config, mock_app):
binding = self._make_binding(qdrant_endpoint="https://qdrant-custom.tidb.com")
mock_db.session.scalars.return_value.one_or_none.return_value = binding
self._setup_config(mock_config, mock_app)
ds = self._make_dataset(index_struct_dict={"type": "tidb_on_qdrant", "vector_store": {"class_prefix": "col"}})
factory = TidbOnQdrantVectorFactory()
@ -215,17 +221,13 @@ class TestInitVectorEndpointSelection:
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.current_app")
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.dify_config")
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.select")
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.db")
@patch("dify_vdb_tidb_on_qdrant.tidb_on_qdrant_vector.qdrant_client.QdrantClient")
def test_falls_back_to_global_when_binding_endpoint_is_none(self, mock_qc, mock_db, mock_config, mock_app):
def test_falls_back_to_global_when_binding_endpoint_is_none(self, mock_qc, mock_db, mock_select, mock_config, mock_app):
binding = self._make_binding(qdrant_endpoint=None)
mock_db.session.scalars.return_value.one_or_none.return_value = binding
mock_config.TIDB_ON_QDRANT_URL = "https://qdrant-global.tidb.com"
mock_config.TIDB_ON_QDRANT_CLIENT_TIMEOUT = 20
mock_config.TIDB_ON_QDRANT_GRPC_PORT = 6334
mock_config.TIDB_ON_QDRANT_GRPC_ENABLED = False
mock_config.QDRANT_REPLICATION_FACTOR = 1
mock_app.config = {"root_path": "/app"}
self._setup_config(mock_config, mock_app)
ds = self._make_dataset(index_struct_dict={"type": "tidb_on_qdrant", "vector_store": {"class_prefix": "col"}})
factory = TidbOnQdrantVectorFactory()