add test case

Author: iyuroch

conn_http_batch.go

@@ -73,8 +73,8 @@ func (h *httpConnect) prepareBatch(ctx context.Context, query string, opts drive
 			return nil, err
 		}
 		// these column types cannot be specified in INSERT queries
-                if default_type == "MATERIALIZED" || default_type == "ALIAS" {
-		 	continue
+		if default_type == "MATERIALIZED" || default_type == "ALIAS" {
+			continue
 		}
 		colNames = append(colNames, colName)
 		columns[colName] = colType

tests/std/mat_cols_test.go

@@ -0,0 +1,77 @@
+// Licensed to ClickHouse, Inc. under one or more contributor
+// license agreements. See the NOTICE file distributed with
+// this work for additional information regarding copyright
+// ownership. ClickHouse, Inc. licenses this file to you under
+// the Apache License, Version 2.0 (the "License"); you may
+// not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+package std
+
+import (
+	"fmt"
+	"math/big"
+	"strconv"
+	"testing"
+
+	"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 TestMatColInsert(t *testing.T) {
+	dsns := map[string]clickhouse.Protocol{"Native": clickhouse.Native, "Http": clickhouse.HTTP}
+	useSSL, err := strconv.ParseBool(clickhouse_tests.GetEnv("CLICKHOUSE_USE_SSL", "false"))
+	require.NoError(t, err)
+	for name, protocol := range dsns {
+		t.Run(fmt.Sprintf("%s Protocol", name), func(t *testing.T) {
+			if conn, err := GetStdDSNConnection(protocol, useSSL, nil); assert.NoError(t, err) {
+				if !CheckMinServerVersion(conn, 21, 12, 0) {
+					t.Skip(fmt.Errorf("unsupported clickhouse version"))
+					return
+				}
+				const ddl = `
+		CREATE TABLE test_mat_cols (
+			  Col1 Int128
+			, Col2 MATERIALIZED Col1 * 2
+		) Engine MergeTree() ORDER BY tuple()
+		`
+				defer func() {
+					conn.Exec("DROP TABLE test_mat_cols")
+				}()
+				_, err := conn.Exec(ddl)
+				require.NoError(t, err)
+				scope, err := conn.Begin()
+				require.NoError(t, err)
+				batch, err := scope.Prepare("INSERT INTO test_mat_cols")
+				require.NoError(t, err)
+				var (
+					col1Data = big.NewInt(128)
+					col2Data = big.NewInt(128 * 2)
+				)
+				_, err = batch.Exec(col1Data)
+				require.NoError(t, err)
+				require.NoError(t, scope.Commit())
+				var (
+					col1 big.Int
+					col2 big.Int
+				)
+				require.NoError(t, conn.QueryRow("SELECT * FROM test_mat_cols").Scan(&col1))
+				assert.Equal(t, *col1Data, col1)
+				require.NoError(t, conn.QueryRow("SELECT Col1, Col2 from test_mat_cols").Scan(&col1, &col2))
+				assert.Equal(t, *col1Data, col1)
+				assert.Equal(t, *col2Data, col2)
+			}
+		})
+	}
+}