Function TestValidateSQLForEditor

backend/plugin/parser/snowflake/query_test.go:13–105

(t *testing.T)

)

func TestValidateSQLForEditor(t *testing.T) {
	tests := []struct {
		statement   string
		valid       bool
		gotAllQuery bool
		err         bool
	}{
		{
			statement:   "SHOW TABLES;",
			valid:       true,
			gotAllQuery: true,
		},
		{
			// Result-pipe: read-only source + trailing SELECT over $1 is a query.
			statement:   "SHOW TABLES ->> SELECT * FROM $1;",
			valid:       true,
			gotAllQuery: true,
		},
		{
			// Result-pipe with a non-read-only source must not classify as a query.
			statement:   "CALL P() ->> SELECT * FROM $1;",
			valid:       false,
			gotAllQuery: false,
		},
		{
			// A non-read-only statement piped FROM a SHOW must not classify as a query.
			statement:   "SHOW TABLES ->> DELETE FROM T1;",
			valid:       false,
			gotAllQuery: false,
		},
		{
			statement:   "DESC TABLE bytebase;",
			valid:       true,
			gotAllQuery: true,
		},
		{
			statement:   "SELECT * FROM t1 WHERE c1 = 1; SELECT * FROM t2;",
			valid:       true,
			gotAllQuery: true,
		},
		{
			statement:   "CREATE TABLE t1 (c1 INT);",
			valid:       false,
			gotAllQuery: false,
		},
		{
			statement:   "UPDATE t1 SET c1 = 1;",
			valid:       false,
			gotAllQuery: false,
		},
		{
			statement:   "EXPLAIN SELECT * FROM t1;",
			valid:       true,
			gotAllQuery: true,
		},
		{
			statement:   `select* from t`,
			valid:       true,