Add tests for stripLeadingComments and getCommandType

Tests cover:
- Block comments (/* */)
- Line comments (--)
- Multiple/nested comments
- Fivetran-style comments (/*Fivetran*/)
- Edge cases (empty, whitespace, unclosed comments)
- All command types including CREATE/DROP SCHEMA

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: EDsCODE

server/conn_test.go

@@ -0,0 +1,226 @@
+package server
+
+import (
+	"testing"
+)
+
+func TestStripLeadingComments(t *testing.T) {
+	tests := []struct {
+		name     string
+		input    string
+		expected string
+	}{
+		{
+			name:     "no comments",
+			input:    "SELECT * FROM users",
+			expected: "SELECT * FROM users",
+		},
+		{
+			name:     "block comment at start",
+			input:    "/*Fivetran*/CREATE SCHEMA test",
+			expected: "CREATE SCHEMA test",
+		},
+		{
+			name:     "block comment with spaces",
+			input:    "/* comment */ SELECT 1",
+			expected: "SELECT 1",
+		},
+		{
+			name:     "multiple block comments",
+			input:    "/* first */ /* second */ INSERT INTO t",
+			expected: "INSERT INTO t",
+		},
+		{
+			name:     "line comment at start",
+			input:    "-- comment\nSELECT 1",
+			expected: "SELECT 1",
+		},
+		{
+			name:     "mixed comments",
+			input:    "/* block */ -- line\nUPDATE t SET x=1",
+			expected: "UPDATE t SET x=1",
+		},
+		{
+			name:     "whitespace before comment",
+			input:    "  /* comment */ DELETE FROM t",
+			expected: "DELETE FROM t",
+		},
+		{
+			name:     "unclosed block comment",
+			input:    "/* unclosed SELECT",
+			expected: "/* unclosed SELECT",
+		},
+		{
+			name:     "line comment without newline",
+			input:    "-- only comment",
+			expected: "",
+		},
+		{
+			name:     "empty string",
+			input:    "",
+			expected: "",
+		},
+		{
+			name:     "only whitespace",
+			input:    "   ",
+			expected: "",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			result := stripLeadingComments(tt.input)
+			if result != tt.expected {
+				t.Errorf("stripLeadingComments(%q) = %q, want %q", tt.input, result, tt.expected)
+			}
+		})
+	}
+}
+
+func TestGetCommandType(t *testing.T) {
+	// Create a minimal clientConn for testing
+	c := &clientConn{}
+
+	tests := []struct {
+		name     string
+		query    string
+		expected string
+	}{
+		// Basic commands without comments
+		{
+			name:     "SELECT",
+			query:    "SELECT * FROM users",
+			expected: "SELECT",
+		},
+		{
+			name:     "INSERT",
+			query:    "INSERT INTO users VALUES (1)",
+			expected: "INSERT",
+		},
+		{
+			name:     "UPDATE",
+			query:    "UPDATE users SET name='test'",
+			expected: "UPDATE",
+		},
+		{
+			name:     "DELETE",
+			query:    "DELETE FROM users",
+			expected: "DELETE",
+		},
+		{
+			name:     "CREATE TABLE",
+			query:    "CREATE TABLE users (id INT)",
+			expected: "CREATE TABLE",
+		},
+		{
+			name:     "CREATE SCHEMA",
+			query:    "CREATE SCHEMA myschema",
+			expected: "CREATE SCHEMA",
+		},
+		{
+			name:     "DROP TABLE",
+			query:    "DROP TABLE users",
+			expected: "DROP TABLE",
+		},
+		{
+			name:     "DROP SCHEMA",
+			query:    "DROP SCHEMA myschema",
+			expected: "DROP SCHEMA",
+		},
+		{
+			name:     "DROP SCHEMA IF EXISTS CASCADE",
+			query:    "DROP SCHEMA IF EXISTS myschema CASCADE",
+			expected: "DROP SCHEMA",
+		},
+
+		// Commands with Fivetran-style comments
+		{
+			name:     "CREATE SCHEMA with Fivetran comment",
+			query:    "/*Fivetran*/CREATE SCHEMA test_schema",
+			expected: "CREATE SCHEMA",
+		},
+		{
+			name:     "DROP SCHEMA with Fivetran comment",
+			query:    "/*Fivetran*/DROP SCHEMA IF EXISTS test_schema CASCADE",
+			expected: "DROP SCHEMA",
+		},
+		{
+			name:     "CREATE TABLE with Fivetran comment",
+			query:    "/*Fivetran*/CREATE TABLE test_table (id INT)",
+			expected: "CREATE TABLE",
+		},
+		{
+			name:     "INSERT with Fivetran comment",
+			query:    "/*Fivetran*/INSERT INTO test_table VALUES (1)",
+			expected: "INSERT",
+		},
+
+		// Commands with other comment styles
+		{
+			name:     "SELECT with block comment",
+			query:    "/* query */ SELECT 1",
+			expected: "SELECT",
+		},
+		{
+			name:     "UPDATE with line comment",
+			query:    "-- update query\nUPDATE t SET x=1",
+			expected: "UPDATE",
+		},
+
+		// Transaction commands
+		{
+			name:     "BEGIN",
+			query:    "BEGIN",
+			expected: "BEGIN",
+		},
+		{
+			name:     "COMMIT",
+			query:    "COMMIT",
+			expected: "COMMIT",
+		},
+		{
+			name:     "ROLLBACK",
+			query:    "ROLLBACK",
+			expected: "ROLLBACK",
+		},
+
+		// Other commands
+		{
+			name:     "SET",
+			query:    "SET search_path TO myschema",
+			expected: "SET",
+		},
+		{
+			name:     "TRUNCATE",
+			query:    "TRUNCATE TABLE users",
+			expected: "TRUNCATE TABLE",
+		},
+		{
+			name:     "ALTER",
+			query:    "ALTER TABLE users ADD COLUMN name TEXT",
+			expected: "ALTER TABLE",
+		},
+
+		// Edge cases
+		{
+			name:     "lowercase command",
+			query:    "select * from users",
+			expected: "SELECT",
+		},
+		{
+			name:     "mixed case with comment",
+			query:    "/*Test*/Select * From Users",
+			expected: "SELECT",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			// getCommandType expects uppercase input
+			result := c.getCommandType(tt.query)
+			if result != tt.expected {
+				t.Errorf("getCommandType(%q) = %q, want %q", tt.query, result, tt.expected)
+			}
+		})
+	}
+}