1229 test case more resilient

tests/issues/1229_test.go

@@ -9,10 +9,13 @@ import (
 
 	"github.com/ClickHouse/clickhouse-go/v2"
 	clickhouse_tests "github.com/ClickHouse/clickhouse-go/v2/tests"
+	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
 
 func Test1229(t *testing.T) {
+	const queryTimeout = 2 * time.Second
+
 	var (
 		conn, err = clickhouse_tests.GetConnection("issues", clickhouse.Settings{
 			"max_execution_time":             60,
@@ -32,7 +35,7 @@ func Test1229(t *testing.T) {
 
 	const insertQuery = "INSERT INTO test_1229 VALUES ('test1value%d', 'test2value%d')"
 	for i := 0; i < 100; i++ {
-		withTimeoutCtx, cancel := context.WithTimeout(ctx, time.Millisecond*100)
+		withTimeoutCtx, cancel := context.WithTimeout(ctx, queryTimeout)
 		require.NoError(t, conn.Exec(withTimeoutCtx, fmt.Sprintf(insertQuery, i, i)))
 		cancel()
 	}
@@ -43,14 +46,16 @@ func Test1229(t *testing.T) {
 		wg.Add(1)
 		go func() {
 			defer wg.Done()
-			withTimeoutCtx, cancel := context.WithTimeout(ctx, time.Millisecond*10)
+			withTimeoutCtx, cancel := context.WithTimeout(ctx, queryTimeout)
 			defer cancel()
 			_, _ = conn.Query(withTimeoutCtx, selectQuery)
 		}()
 	}
 
 	wg.Wait()
 
-	openConnections := conn.Stats().Open
-	require.Equal(t, 0, openConnections)
+	assert.EventuallyWithT(t, func(ct *assert.CollectT) {
+		openConnections := conn.Stats().Open
+		assert.Zerof(ct, openConnections, "open connections: %d", openConnections)
+	}, time.Second*5, time.Millisecond*10)
 }