Function TestRowAccessPolicy

pkg/sql/ast/types_test.go:297–339

Test RowAccessPolicy

(t *testing.T)

// Test RowAccessPolicy
func TestRowAccessPolicy(t *testing.T) {
	tests := []struct {
		name         string
		policy       *RowAccessPolicy
		wantLiteral  string
		wantChildren int
	}{
		{
			name: "policy without filter",
			policy: &RowAccessPolicy{
				Name:    "admin_policy",
				Enabled: true,
			},
			wantLiteral:  "ROW ACCESS POLICY",
			wantChildren: 0,
		},
		{
			name: "policy with filter",
			policy: &RowAccessPolicy{
				Name:    "user_policy",
				Filter:  &mockExpr{},
				Enabled: true,
			},
			wantLiteral:  "ROW ACCESS POLICY",
			wantChildren: 1,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			// Test TokenLiteral
			if got := tt.policy.TokenLiteral(); got != tt.wantLiteral {
				t.Errorf("RowAccessPolicy.TokenLiteral() = %v, want %v", got, tt.wantLiteral)
			}

			// Test Children
			children := tt.policy.Children()
			if len(children) != tt.wantChildren {
				t.Errorf("RowAccessPolicy.Children() count = %d, want %d", len(children), tt.wantChildren)
			}
		})
	}
}

// Test StatementImpl
func TestStatementImpl(t *testing.T) {