Function TestCompose_SourceInfo

policy/composer_test.go:13–53

(t *testing.T)

)

func TestCompose_SourceInfo(t *testing.T) {
	policyYAML := `name: test_policy
rule:
  match:
    - condition: "2 == 1"
      output: "'hi'"
    - output: "'hello' + ' world'"
`
	src := StringSource(policyYAML, "test_policy.yaml")
	parser, err := NewParser()
	if err != nil {
		t.Fatalf("NewParser() failed: %v", err)
	}
	policy, iss := parser.Parse(src)
	if iss.Err() != nil {
		t.Fatalf("parser.Parse() failed: %v", iss.Err())
	}

	env, err := cel.NewEnv(cel.OptionalTypes(), ext.Bindings())
	if err != nil {
		t.Fatalf("cel.NewEnv() failed: %v", err)
	}
	compiledRule, iss := CompileRule(env, policy)
	if iss.Err() != nil {
		t.Fatalf("CompileRule() failed: %v", iss.Err())
	}
	composer, err := NewRuleComposer(env)
	if err != nil {
		t.Fatalf("NewRuleComposer() failed: %v", err)
	}
	compAST, iss := composer.Compose(compiledRule)
	if iss.Err() != nil {
		t.Fatalf("composer.Compose() failed: %v", iss.Err())
	}

	si := compAST.SourceInfo()
	if si.Location != "test_policy.yaml" {
		t.Errorf("SourceInfo.Location got %q, wanted test_policy.yaml", si.Location)
	}
	verifySourceInfoTransfer(t, compiledRule, compAST)
}

func TestCompose_Unnest(t *testing.T) {
	policyYAML := `name: unnest