Function TestStripFuncResultTextCasts

schema/normalize_policy_test.go:26–131

(t *testing.T)

}

func TestStripFuncResultTextCasts(t *testing.T) {
	testCases := []struct {
		name  string
		input string
		want  string
	}{
		{
			name:  "strips text cast on function result",
			input: "current_setting('app.tenant_id', true)::text = tenant_id::text",
			want:  "current_setting('app.tenant_id', true) = tenant_id::text",
		},
		{
			name:  "strips text cast on parenthesized function result",
			input: "(current_setting('app.tenant_id', true))::text = 'x'",
			want:  "current_setting('app.tenant_id', true) = 'x'",
		},
		{
			name:  "strips character varying cast on function result",
			input: "current_user::character varying = 'x'",
			want:  "current_user::character varying = 'x'", // current_user is a keyword, not a function call
		},
		{
			name:  "strips character varying cast on function call",
			input: "lower('X')::character varying = 'x'",
			want:  "lower('X') = 'x'",
		},
		{
			name:  "keeps non-text cast on function result",
			input: "current_setting('app.tenant_id', true)::integer = 1",
			want:  "current_setting('app.tenant_id', true)::integer = 1",
		},
		{
			name:  "keeps text cast on column",
			input: "tenant_id::text = 'x'",
			want:  "tenant_id::text = 'x'",
		},
		{
			name:  "keeps text cast on literal",
			input: "'x'::text = name",
			want:  "'x'::text = name",
		},
		{
			name:  "strips inner text cast under outer non-text cast",
			input: "lower('X')::text::integer = 1",
			want:  "lower('X')::integer = 1",
		},
		{
			name:  "recurses into NOT",
			input: "NOT (lower('X')::text = 'x')",
			want:  "NOT (lower('X') = 'x')",
		},
		{
			name:  "recurses into AND and OR",
			input: "lower('a')::text = 'a' AND (lower('b')::text = 'b' OR tenant_id = 1)",
			want:  "lower('a') = 'a' AND (lower('b') = 'b' OR tenant_id = 1)",
		},
		{
			name:  "recurses into string concatenation",