add more test

Signed-off-by: zzzk1 <[email protected]>

Author: zhengkezhou1

internal/storage/integration/package_test.go

@@ -11,9 +11,14 @@ import (
 )
 
 func TestMain(m *testing.M) {
-	if os.Getenv("STORAGE") == "elasticsearch" || os.Getenv("STORAGE") == "opensearch" {
+	switch os.Getenv("STORAGE") {
+	case "elasticsearch":
 		testutils.VerifyGoLeaksForES(m)
-	} else {
+	case "opensearch":
+		testutils.VerifyGoLeaksForES(m)
+	case "clickhouse":
+		testutils.VerifyGoLeaksForCH(m)
+	default:
 		testutils.VerifyGoLeaks(m)
 	}
 }

internal/storage/v2/clickhouse/factory_test.go

@@ -6,7 +6,6 @@ package clickhouse
 import (
 	"testing"
 
-	"github.com/ClickHouse/ch-go/cht"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"go.uber.org/zap"
@@ -24,21 +23,14 @@ func TestClickHouseFactoryWithConfig(t *testing.T) {
 		},
 		ConnectionPoolConfig: config.ConnectionPoolConfig{},
 	}
-	// provide a tcp server for testing.
-	cht.New(t,
-		cht.WithLog(zap.NewNop()),
-	)
+
 	f, err := NewFactoryWithConfig(&cfg, zap.NewNop())
 	require.NoError(t, err)
 	defer f.Close()
 }
 
 func TestCreateTraceWriter(t *testing.T) {
 	cfg := config.DefaultConfiguration()
-	// provide a tcp server for testing.
-	cht.New(t,
-		cht.WithLog(zap.NewNop()),
-	)
 
 	f, err := NewFactoryWithConfig(&cfg, zap.NewNop())
 	require.NoError(t, err)

internal/storage/v2/clickhouse/package_test.go

@@ -0,0 +1,14 @@
+// Copyright (c) 2024 The Jaeger Authors.
+// SPDX-License-Identifier: Apache-2.0
+
+package clickhouse
+
+import (
+	"testing"
+
+	"github.com/jaegertracing/jaeger/pkg/testutils"
+)
+
+func TestMain(m *testing.M) {
+	testutils.VerifyGoLeaksForCH(m)
+}

pkg/clickhouse/config/config_test.go

@@ -6,7 +6,6 @@ package config
 import (
 	"testing"
 
-	"github.com/ClickHouse/ch-go/cht"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"go.uber.org/zap"
@@ -23,11 +22,8 @@ func TestNewClientWithDefaults(t *testing.T) {
 	cfg := DefaultConfiguration()
 	logger := zap.NewNop()
 
-	cht.New(t,
-		cht.WithLog(logger),
-	)
-
 	client, err := cfg.NewClient(logger)
+
 	require.NoError(t, err)
 	assert.NotEmpty(t, client)
 	defer client.Close()
@@ -36,11 +32,7 @@ func TestNewClientWithDefaults(t *testing.T) {
 func TestNewPool(t *testing.T) {
 	cfg := DefaultConfiguration()
 	logger := zap.NewNop()
-	conn, err := cfg.newConn()
-
-	cht.New(t,
-		cht.WithLog(logger),
-	)
+	conn, err := cfg.newPool(logger)
 
 	require.NoError(t, err)
 	assert.NotEmpty(t, conn)
@@ -52,19 +44,12 @@ func TestNewPoolFail(t *testing.T) {
 	logger := zap.NewNop()
 	pool, err := cfg.newPool(logger)
 
-	cht.New(t,
-		cht.WithLog(logger),
-	)
-
 	require.Error(t, err)
 	assert.Nil(t, pool)
 }
 
 func TestNewConnection(t *testing.T) {
 	cfg := DefaultConfiguration()
-	logger := zap.NewNop()
-
-	cht.New(t, cht.WithLog(logger))
 
 	conn, err := cfg.newConn()
 	require.NoError(t, err)

pkg/clickhouse/config/package_test.go

@@ -0,0 +1,14 @@
+// Copyright (c) 2024 The Jaeger Authors.
+// SPDX-License-Identifier: Apache-2.0
+
+package config
+
+import (
+	"testing"
+
+	"github.com/jaegertracing/jaeger/pkg/testutils"
+)
+
+func TestMain(m *testing.M) {
+	testutils.VerifyGoLeaksForCH(m)
+}

pkg/testutils/leakcheck.go

@@ -55,6 +55,14 @@ func ignoreHttpTransportReadLoopLeak() goleak.Option {
 	return goleak.IgnoreTopFunction("net/http.(*persistConn).readLoop")
 }
 
+func ignoreBackgroundHealthCheck() goleak.Option {
+	return goleak.IgnoreTopFunction("github.com/ClickHouse/ch-go/chpool.(*Pool).backgroundHealthCheck")
+}
+
+func ignoreStartAutoCloseIdleConnections() goleak.Option {
+	return goleak.IgnoreTopFunction("github.com/ClickHouse/clickhouse-go/v2.(*clickhouse).startAutoCloseIdleConnections")
+}
+
 // VerifyGoLeaks verifies that unit tests do not leak any goroutines.
 // It should be called in TestMain.
 func VerifyGoLeaks(m *testing.M) {
@@ -82,3 +90,7 @@ func VerifyGoLeaksOnceForES(t *testing.T) {
 func VerifyGoLeaksForES(m *testing.M) {
 	goleak.VerifyTestMain(m, ignoreHttpTransportWriteLoopLeak(), ignoreHttpTransportPollRuntimeLeak(), ignoreHttpTransportReadLoopLeak())
 }
+
+func VerifyGoLeaksForCH(m *testing.M) {
+	goleak.VerifyTestMain(m, ignoreHttpTransportWriteLoopLeak(), ignoreHttpTransportPollRuntimeLeak(), ignoreBackgroundHealthCheck(), ignoreStartAutoCloseIdleConnections())
+}