Function TestGetCommandType

server/conn_test.go:200–435

(t *testing.T)

	}
}

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 TEMPORARY TABLE",
			query:    "CREATE TEMPORARY TABLE temp_users (id INT)",
			expected: "CREATE TABLE",
		},
		{
			name:     "CREATE TEMP TABLE",
			query:    "CREATE TEMP TABLE temp_users (id INT)",
			expected: "CREATE TABLE",
		},
		{
			name:     "CREATE TEMPORARY TABLE with Fivetran comment",
			query:    "/*Fivetran*/CREATE TEMPORARY TABLE temp_users (id INT)",
			expected: "CREATE TABLE",
		},
		{
			name:     "CREATE SCHEMA",
			query:    "CREATE SCHEMA myschema",
			expected: "CREATE SCHEMA",
		},
		{
			name:     "DROP TABLE",