Function TestBasicFunctionality

api/service/vectorstore/test_basic.go:14–154

TestBasicFunctionality 测试基本功能

()

// TestBasicFunctionality 测试基本功能
func TestBasicFunctionality() error {
	logger.SysLogf("=== 测试向量数据库驱动基本功能 ===")

	// 1. 测试配置加载
	logger.SysLogf("1. 测试配置加载...")
	config := LoadFromEnv()
	if err := config.Validate(); err != nil {
		return fmt.Errorf("配置验证失败: %v", err)
	}
	logger.SysLogf("   配置类型: %s, 端点: %s", config.Type, config.Endpoint)

	// 2. 测试向量存储创建
	logger.SysLogf("2. 测试向量存储创建...")
	store, err := NewVectorStore(config)
	if err != nil {
		return fmt.Errorf("创建向量存储失败: %v", err)
	}
	logger.SysLogf("   向量存储类型: %s", store.Type())

	// 3. 测试连接
	logger.SysLogf("3. 测试连接...")
	ctx := context.Background()
	if err := store.Connect(ctx); err != nil {
		return fmt.Errorf("连接失败: %v", err)
	}
	logger.SysLogf("   连接成功")

	// 4. 测试健康检查
	logger.SysLogf("4. 测试健康检查...")
	if err := store.HealthCheck(ctx); err != nil {
		return fmt.Errorf("健康检查失败: %v", err)
	}
	logger.SysLogf("   健康检查通过")

	// 5. 测试集合操作
	logger.SysLogf("5. 测试集合操作...")
	testCollection := "test_collection_" + fmt.Sprintf("%d", time.Now().Unix())

	// 创建测试集合
	collectionConfig := CollectionConfig{
		Name:      testCollection,
		Dimension: 3,
		Metric:    "cosine",
		IndexType: "HNSW",
	}

	if err := store.CreateCollection(ctx, collectionConfig); err != nil {
		return fmt.Errorf("创建集合失败: %v", err)
	}
	logger.SysLogf("   创建集合成功: %s", testCollection)

	// 检查集合是否存在（通过获取集合信息的方式）
	_, err = store.GetCollectionInfo(ctx, testCollection)
	if err != nil {
		if vsErr, ok := err.(*VectorStoreError); ok && vsErr.Code == ErrCodeCollectionNotFound {
			return fmt.Errorf("集合应该存在但获取信息失败")
		}
		return fmt.Errorf("获取集合信息失败: %v", err)