Deadlock test for 548

tests/issues/548_test.go

@@ -0,0 +1,46 @@
+package issues
+
+import (
+ "context"
+ "testing"
+ "time"
+
+ "github.com/ClickHouse/clickhouse-go/v2"
+ "github.com/stretchr/testify/assert"
+)
+
+func Test548(t *testing.T) {
+ var (
+ ctx, cancel = context.WithTimeout(context.Background(), time.Second)
+ conn, err = clickhouse.Open(&clickhouse.Options{
+ Addr: []string{"127.0.0.1:9000"},
+ Auth: clickhouse.Auth{
+ Database: "default",
+ Username: "default",
+ Password: "",
+ },
+ DialTimeout: time.Second,
+ Compression: &clickhouse.Compression{
+ Method: clickhouse.CompressionLZ4,
+ },
+ //Debug: true,
+ })
+ )
+ defer cancel()
+ assert.NoError(t, err)
+ // give it plenty of time before we conclusively assume deadlock
+ timeout := time.After(5 * time.Second)
+ done := make(chan bool)
+ go func() {
+ // should take 1s
+ rows, _ := conn.Query(ctx, "SELECT sleepEachRow(0.001) as Col1 FROM system.numbers LIMIT 1000 SETTINGS max_block_size=10;")
+ rows.Close()
+ done <- true
+ }()
+
+ select {
+ case <-timeout:
+ t.Fatal("Close() deadlocked")
+ case <-done:
+ }
+}