Function TestValidateQuery

backend/plugin/parser/trino/query_test.go:10–103

(t *testing.T)

)

func TestValidateQuery(t *testing.T) {
	tests := []struct {
		name         string
		sql          string
		wantReadOnly bool
		wantData     bool
		wantErr      bool
	}{
		{
			name:         "Simple SELECT",
			sql:          "SELECT * FROM users",
			wantReadOnly: true,
			wantData:     true,
			wantErr:      false,
		},
		{
			name:         "INSERT statement",
			sql:          "INSERT INTO users (id, name) VALUES (1, 'John')",
			wantReadOnly: false,
			wantData:     false,
			wantErr:      false,
		},
		{
			name:         "UPDATE statement",
			sql:          "UPDATE users SET name = 'John' WHERE id = 1",
			wantReadOnly: false,
			wantData:     false,
			wantErr:      false,
		},
		{
			name:         "DELETE statement",
			sql:          "DELETE FROM users WHERE id = 1",
			wantReadOnly: false,
			wantData:     false,
			wantErr:      false,
		},
		{
			name:         "CREATE TABLE statement",
			sql:          "CREATE TABLE users (id INT, name VARCHAR)",
			wantReadOnly: false,
			wantData:     false,
			wantErr:      false,
		},
		{
			name:         "EXPLAIN statement",
			sql:          "EXPLAIN SELECT * FROM users",
			wantReadOnly: true,
			wantData:     true,
			wantErr:      false,
		},
		{
			name:         "EXPLAIN ANALYZE SELECT",
			sql:          "EXPLAIN ANALYZE SELECT * FROM users",
			wantReadOnly: true,
			wantData:     true,
			wantErr:      false,
		},
		{