Function TestUserN1QLQueriesInvalid

db/functions/n1ql_function_test.go:224–257

Unit test to ensure only SELECT queries can run

(t *testing.T)

// Unit test to ensure only SELECT queries can run
func TestUserN1QLQueriesInvalid(t *testing.T) {
	var kBadN1QLFunctionsConfig = FunctionsConfig{
		Definitions: FunctionsDefs{
			"evil_mutant": &FunctionConfig{
				Type:  "query",
				Code:  `DROP COLLECTION Students`,
				Allow: &Allow{Channels: []string{"*"}},
			},
		},
	}

	_, err := CompileFunctions(base.TestCtx(t), kBadN1QLFunctionsConfig)
	assert.ErrorContains(t, err, "only SELECT queries are allowed") // See fn validateN1QLQuery

	var kOKN1QLFunctionsConfig = FunctionsConfig{
		Definitions: FunctionsDefs{
			"lowercase": &FunctionConfig{
				Type:  "query",
				Code:  `seleCt $args.city AS city`,
				Args:  []string{"city"},
				Allow: &Allow{Channels: []string{"*"}},
			},
			"parens": &FunctionConfig{
				Type:  "query",
				Code:  `  (select $args.city AS city)`,
				Args:  []string{"city"},
				Allow: &Allow{Channels: []string{"*"}},
			},
		},
	}

	_, err = CompileFunctions(base.TestCtx(t), kOKN1QLFunctionsConfig)
	assert.NoError(t, err)
}