use proper test connection

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

Author: kavirajk

tests/http_exception_test.go

@@ -10,10 +10,7 @@ import (
 )
 
 func TestHTTPExceptionHandling(t *testing.T) {
-	conn, err := clickhouse.Open(&clickhouse.Options{
-		Protocol: clickhouse.HTTP,
-		Addr:     []string{"localhost:8123"},
-	})
+	conn, err := GetNativeConnection(t, clickhouse.HTTP, nil, nil, nil)
 	require.NoError(t, err)
 
 	ctx := context.Background()
@@ -49,43 +46,3 @@ func TestHTTPExceptionHandling(t *testing.T) {
 
 	assert.True(t, occured, "execption not caught in the response chunks")
 }
-
-func TestHTTPExceptionHandlingDB(t *testing.T) {
-	conn := clickhouse.OpenDB(&clickhouse.Options{
-		Protocol: clickhouse.HTTP,
-		Addr:     []string{"localhost:8123"},
-	})
-
-	ctx := context.Background()
-
-	// These settings will make sure mid-stream exception most likely on the server
-	ctx = clickhouse.Context(ctx, clickhouse.WithSettings(clickhouse.Settings{
-		"max_threads":                           1,
-		"max_block_size":                        1,
-		"http_write_exception_in_output_format": 0,
-		"wait_end_of_query":                     0,
-		"http_response_buffer_size":             1,
-	}))
-
-	rows, err := conn.QueryContext(ctx, `SELECT throwIf(number=3, 'there is an exception') FROM system.numbers`)
-	require.NoError(t, err) // query shouldn't fail with 500 status code.
-
-	occured := false
-	// query should fail while scanning the rows mid-stream
-	for rows.Next() {
-		var result uint8
-		err := rows.Scan(&result)
-		if err != nil {
-			// should be an exception caught correctly
-			assert.Contains(t, err.Error(), "there is an exception", "Expected exception message not caught")
-			occured = true
-		}
-	}
-
-	if err := rows.Err(); err != nil {
-		assert.Contains(t, err.Error(), "there is an exception", "Expected exception message not caught")
-		occured = true
-	}
-
-	assert.True(t, occured, "execption not caught in the response chunks")
-}

tests/main_test.go

@@ -1,4 +1,3 @@
-
 package tests
 
 import (

tests/std/http_exception_test.go

@@ -0,0 +1,48 @@
+package std
+
+import (
+	"context"
+	"testing"
+
+	"github.com/ClickHouse/clickhouse-go/v2"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestHTTPExceptionHandlingDB(t *testing.T) {
+	conn, err := GetStdOpenDBConnection(clickhouse.HTTP, nil, nil, nil)
+	require.NoError(t, err)
+
+	ctx := context.Background()
+
+	// These settings will make sure mid-stream exception most likely on the server
+	ctx = clickhouse.Context(ctx, clickhouse.WithSettings(clickhouse.Settings{
+		"max_threads":                           1,
+		"max_block_size":                        1,
+		"http_write_exception_in_output_format": 0,
+		"wait_end_of_query":                     0,
+		"http_response_buffer_size":             1,
+	}))
+
+	rows, err := conn.QueryContext(ctx, `SELECT throwIf(number=3, 'there is an exception') FROM system.numbers`)
+	require.NoError(t, err) // query shouldn't fail with 500 status code.
+
+	occured := false
+	// query should fail while scanning the rows mid-stream
+	for rows.Next() {
+		var result uint8
+		err := rows.Scan(&result)
+		if err != nil {
+			// should be an exception caught correctly
+			assert.Contains(t, err.Error(), "there is an exception", "Expected exception message not caught")
+			occured = true
+		}
+	}
+
+	if err := rows.Err(); err != nil {
+		assert.Contains(t, err.Error(), "there is an exception", "Expected exception message not caught")
+		occured = true
+	}
+
+	assert.True(t, occured, "execption not caught in the response chunks")
+}

tests/std/utils.go

@@ -1,4 +1,3 @@
-
 package std
 
 import (