mirror of https://github.com/hashicorp/consul
119 lines
3.2 KiB
Go
119 lines
3.2 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package templatedpolicy
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/consul/agent/structs"
|
|
"github.com/hashicorp/consul/api"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// golden reads from the golden file returning the contents as a string.
|
|
func golden(t *testing.T, name string) string {
|
|
t.Helper()
|
|
|
|
golden := filepath.Join("testdata", name+".golden")
|
|
expected, err := os.ReadFile(golden)
|
|
require.NoError(t, err)
|
|
|
|
return string(expected)
|
|
}
|
|
|
|
func testFormatTemplatedPolicy(t *testing.T, dirPath string) {
|
|
type testCase struct {
|
|
templatedPolicy api.ACLTemplatedPolicyResponse
|
|
}
|
|
|
|
cases := map[string]testCase{
|
|
"node-templated-policy": {
|
|
templatedPolicy: api.ACLTemplatedPolicyResponse{
|
|
TemplateName: api.ACLTemplatedPolicyNodeName,
|
|
Schema: structs.ACLTemplatedPolicyNodeSchema,
|
|
Template: structs.ACLTemplatedPolicyNode,
|
|
},
|
|
},
|
|
"dns-templated-policy": {
|
|
templatedPolicy: api.ACLTemplatedPolicyResponse{
|
|
TemplateName: api.ACLTemplatedPolicyDNSName,
|
|
Schema: structs.ACLTemplatedPolicyNoRequiredVariablesSchema,
|
|
Template: structs.ACLTemplatedPolicyDNS,
|
|
},
|
|
},
|
|
"service-templated-policy": {
|
|
templatedPolicy: api.ACLTemplatedPolicyResponse{
|
|
TemplateName: api.ACLTemplatedPolicyServiceName,
|
|
Schema: structs.ACLTemplatedPolicyServiceSchema,
|
|
Template: structs.ACLTemplatedPolicyService,
|
|
},
|
|
},
|
|
}
|
|
|
|
formatters := map[string]Formatter{
|
|
"pretty": newPrettyFormatter(false),
|
|
"pretty-meta": newPrettyFormatter(true),
|
|
// the JSON formatter ignores the showMeta
|
|
"json": newJSONFormatter(false),
|
|
}
|
|
|
|
for name, tcase := range cases {
|
|
t.Run(name, func(t *testing.T) {
|
|
for fmtName, formatter := range formatters {
|
|
t.Run(fmtName, func(t *testing.T) {
|
|
actual, err := formatter.FormatTemplatedPolicy(tcase.templatedPolicy)
|
|
require.NoError(t, err)
|
|
|
|
gName := fmt.Sprintf("%s.%s", name, fmtName)
|
|
|
|
expected := golden(t, path.Join(dirPath, gName))
|
|
require.Equal(t, expected, actual)
|
|
})
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func testFormatTemplatedPolicyList(t *testing.T, dirPath string) {
|
|
// we don't consider the showMeta field for policy list
|
|
formatters := map[string]Formatter{
|
|
"pretty": newPrettyFormatter(false),
|
|
"json": newJSONFormatter(false),
|
|
}
|
|
|
|
policies := map[string]api.ACLTemplatedPolicyResponse{
|
|
"builtin/node": {
|
|
TemplateName: api.ACLTemplatedPolicyNodeName,
|
|
Schema: structs.ACLTemplatedPolicyNodeSchema,
|
|
Template: structs.ACLTemplatedPolicyNode,
|
|
},
|
|
"builtin/dns": {
|
|
TemplateName: api.ACLTemplatedPolicyDNSName,
|
|
Schema: structs.ACLTemplatedPolicyNoRequiredVariablesSchema,
|
|
Template: structs.ACLTemplatedPolicyDNS,
|
|
},
|
|
"builtin/service": {
|
|
TemplateName: api.ACLTemplatedPolicyServiceName,
|
|
Schema: structs.ACLTemplatedPolicyServiceSchema,
|
|
Template: structs.ACLTemplatedPolicyService,
|
|
},
|
|
}
|
|
|
|
for fmtName, formatter := range formatters {
|
|
t.Run(fmtName, func(t *testing.T) {
|
|
actual, err := formatter.FormatTemplatedPolicyList(policies)
|
|
require.NoError(t, err)
|
|
|
|
gName := fmt.Sprintf("list.%s", fmtName)
|
|
|
|
expected := golden(t, path.Join(dirPath, gName))
|
|
require.Equal(t, expected, actual)
|
|
})
|
|
}
|
|
}
|