@@ -436,6 +436,58 @@ def test_list_indexes(self):
436436
437437 assert result == ["index1" , "index2" ]
438438
439+ def test_list_vector_indexes (self ):
440+ """Test _list_vector_indexes correctly unpacks _get_schema tuple return."""
441+ mock_handler = MagicMock ()
442+ mock_handler .get_server_type .return_value = "milvus"
443+
444+ schema_dict = {
445+ "fields" : [
446+ {"name" : "id" , "type" : DataType .INT64 },
447+ {"name" : "vector" , "type" : DataType .FLOAT_VECTOR },
448+ ]
449+ }
450+ mock_handler ._get_schema .return_value = (schema_dict , 12345 )
451+
452+ mock_index = MagicMock ()
453+ mock_index .field_name = "vector"
454+ mock_index .index_name = "vec_index"
455+ mock_handler .list_indexes .return_value = [mock_index ]
456+ mock_handler .describe_index .return_value = {
457+ "field_name" : "vector" ,
458+ "index_name" : "vec_index" ,
459+ }
460+
461+ with patch (
462+ "pymilvus.milvus_client.milvus_client.create_connection" , return_value = "test"
463+ ), patch ("pymilvus.orm.connections.Connections._fetch_handler" , return_value = mock_handler ):
464+ client = MilvusClient ()
465+ result = client ._list_vector_indexes ("test_collection" )
466+
467+ assert result == ["vec_index" ]
468+ mock_handler ._get_schema .assert_called_once ()
469+
470+ def test_list_vector_indexes_no_vector_fields (self ):
471+ """Test _list_vector_indexes returns empty list when no vector fields."""
472+ mock_handler = MagicMock ()
473+ mock_handler .get_server_type .return_value = "milvus"
474+
475+ schema_dict = {
476+ "fields" : [
477+ {"name" : "id" , "type" : DataType .INT64 },
478+ {"name" : "text" , "type" : DataType .VARCHAR },
479+ ]
480+ }
481+ mock_handler ._get_schema .return_value = (schema_dict , 12345 )
482+
483+ with patch (
484+ "pymilvus.milvus_client.milvus_client.create_connection" , return_value = "test"
485+ ), patch ("pymilvus.orm.connections.Connections._fetch_handler" , return_value = mock_handler ):
486+ client = MilvusClient ()
487+ result = client ._list_vector_indexes ("test_collection" )
488+
489+ assert result == []
490+
439491
440492# ============================================================
441493# MilvusClient Alias Operations Tests
0 commit comments