Function TestValidateSQLForEditor

backend/plugin/parser/plsql/query_test.go:11–96

(t *testing.T)

)

func TestValidateSQLForEditor(t *testing.T) {
	tests := []struct {
		name        string
		statement   string
		valid       bool
		gotAllQuery bool
		wantErr     bool
	}{
		{
			name:        "single select",
			statement:   "SELECT * FROM t1;",
			valid:       true,
			gotAllQuery: true,
		},
		{
			name:        "multiple selects",
			statement:   "SELECT * FROM t1; SELECT * FROM t2;",
			valid:       true,
			gotAllQuery: true,
		},
		{
			name:        "explain plan",
			statement:   "EXPLAIN PLAN FOR SELECT * FROM t1;",
			valid:       true,
			gotAllQuery: true,
		},
		{
			name:        "explain plan for update",
			statement:   "EXPLAIN PLAN FOR UPDATE t1 SET c1 = 1;",
			valid:       true,
			gotAllQuery: true,
		},
		{
			name:        "cte select",
			statement:   "WITH x AS (SELECT * FROM t1) SELECT * FROM x;",
			valid:       true,
			gotAllQuery: true,
		},
		{
			name:        "dml keywords in string and comment",
			statement:   "WITH x AS (SELECT 'update delete insert' AS c1 FROM t1 /* UPDATE */) SELECT c1 FROM x;",
			valid:       true,
			gotAllQuery: true,
		},
		{
			name:      "mixed select and update rejected",
			statement: "SELECT * FROM t1; UPDATE t1 SET c1 = 1;",
		},
		{
			name:      "cte update rejected",
			statement: "WITH x AS (SELECT * FROM t1) UPDATE t1 SET c1 = 1;",
			wantErr:   true,
		},
		{
			name:      "sqlplus command rejected",
			statement: "SET DEFINE OFF\nSELECT * FROM t1;",
		},
		{