Add failing test TestBulkcopyColNameEscaping

Author: dlapko

bulkcopy_test.go

@@ -7,6 +7,7 @@ import (
 	"context"
 	"database/sql"
 	"encoding/hex"
+	"fmt"
 	"math"
 	"reflect"
 	"strings"
@@ -111,6 +112,72 @@ func TestBulkcopyWithInvalidNullableType(t *testing.T) {
 	}
 }
 
+func TestBulkcopyColNameEscaping(t *testing.T) {
+	tableName := "#table_test"
+	columnName := `test_[]{}?@!#$%^&*()_+-=~'\";:/.,<>|\ `
+
+	pool, logger := open(t)
+	defer pool.Close()
+	defer logger.StopLogging()
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	conn, err := pool.Conn(ctx)
+	if err != nil {
+		t.Fatal("failed to pull connection from pool", err)
+	}
+	defer conn.Close()
+
+	q := TSQLQuoter{}
+	tablesql := fmt.Sprintf(`CREATE TABLE %s (%s [int] NULL);`, tableName, q.ID(columnName))
+	_, err = conn.ExecContext(ctx, tablesql)
+	if err != nil {
+		t.Fatal("tablesql failed:", err)
+	}
+
+	stmt, err := conn.PrepareContext(ctx, CopyIn(tableName, BulkOptions{}, columnName))
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer stmt.Close()
+
+	_, err = stmt.Exec(1)
+	if err != nil {
+		t.Fatal("AddRow failed: ", err.Error())
+	}
+
+	result, err := stmt.Exec()
+	if err != nil {
+		t.Fatal("bulkcopy failed: ", err.Error())
+	}
+
+	insertedRowCount, _ := result.RowsAffected()
+	if insertedRowCount == 0 {
+		t.Fatal("0 row inserted!")
+	}
+
+	//data verification
+	rows, err := conn.QueryContext(ctx, "select "+q.ID(columnName)+" from "+tableName)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer rows.Close()
+
+	for rows.Next() {
+		var val int
+		if err := rows.Scan(&val); err != nil {
+			t.Fatal(err)
+		}
+		if val != 1 {
+			t.Errorf("column %s : expected: 1, got: %d", columnName, val)
+		}
+	}
+	if err := rows.Err(); err != nil {
+		t.Error(err)
+	}
+}
+
 func testBulkcopy(t *testing.T, guidConversion bool) {
 	// TDS level Bulk Insert is not supported on Azure SQL Server.
 	if dsn := makeConnStrSettingGuidConversion(t, guidConversion); strings.HasSuffix(strings.Split(dsn.Host, ":")[0], ".database.windows.net") {