mirror of
https://github.com/hanzoai/deploy.git
synced 2026-10-07 08:44:50 +00:00
zeekayandHanzo Dev
1ab6e799c1
go.mod still declared `module github.com/hanzoai/deploy` after the repo was renamed. A Go module path is not cosmetic — it is the resolvable location. So `go get github.com/hanzoai/cd` could not work, and anything importing this module had to spell a name that no longer exists at that address. The build log of v3.6.0 is where it surfaced: the compiler was still emitting github.com/hanzoai/deploy/cmd/cd/commands. 3541 import paths across 690 .go, 19 .proto, 4 .yaml, 3 .sh, 2 .mod, Procfile and Makefile. Plus the 62 ghcr.io/hanzoai/deploy image references, moved to ghcr.io/hanzoai/cd to match hanzo.yml's build target — a distinct string from the module path, replaced separately so neither could clobber the other. go mod tidy clean, CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build ./... clean. Co-authored-by: Hanzo Dev <[email protected]>
888 lines
28 KiB
Go
888 lines
28 KiB
Go
package generators
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/client/fake"
|
|
|
|
"github.com/hanzoai/cd/applicationset/utils"
|
|
argoprojiov1alpha1 "github.com/hanzoai/cd/pkg/apis/application/v1alpha1"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type possiblyErroringFakeCtrlRuntimeClient struct {
|
|
client.Client
|
|
shouldError bool
|
|
}
|
|
|
|
func (p *possiblyErroringFakeCtrlRuntimeClient) List(ctx context.Context, secretList client.ObjectList, opts ...client.ListOption) error {
|
|
if p.shouldError {
|
|
return errors.New("could not list Secrets")
|
|
}
|
|
return p.Client.List(ctx, secretList, opts...)
|
|
}
|
|
|
|
func TestGenerateParams(t *testing.T) {
|
|
clusters := []client.Object{
|
|
&corev1.Secret{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "Secret",
|
|
APIVersion: "v1",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "staging-01",
|
|
Namespace: "namespace",
|
|
Labels: map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "staging",
|
|
"org": "foo",
|
|
},
|
|
Annotations: map[string]string{
|
|
"foo.example.com": "staging",
|
|
},
|
|
},
|
|
Data: map[string][]byte{
|
|
"config": []byte("{}"),
|
|
"name": []byte("staging-01"),
|
|
"server": []byte("https://staging-01.example.com"),
|
|
},
|
|
Type: corev1.SecretType("Opaque"),
|
|
},
|
|
&corev1.Secret{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "Secret",
|
|
APIVersion: "v1",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "production-01",
|
|
Namespace: "namespace",
|
|
Labels: map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "production",
|
|
"org": "bar",
|
|
},
|
|
Annotations: map[string]string{
|
|
"foo.example.com": "production",
|
|
},
|
|
},
|
|
Data: map[string][]byte{
|
|
"config": []byte("{}"),
|
|
"name": []byte("production_01/west"),
|
|
"server": []byte("https://production-01.example.com"),
|
|
"project": []byte("prod-project"),
|
|
},
|
|
Type: corev1.SecretType("Opaque"),
|
|
},
|
|
}
|
|
testCases := []struct {
|
|
name string
|
|
selector metav1.LabelSelector
|
|
isFlatMode bool
|
|
values map[string]string
|
|
expected []map[string]any
|
|
// clientError is true if a k8s client error should be simulated
|
|
clientError bool
|
|
expectedError error
|
|
}{
|
|
{
|
|
name: "no label selector",
|
|
selector: metav1.LabelSelector{},
|
|
values: map[string]string{
|
|
"lol1": "lol",
|
|
"lol2": "{{values.lol1}}{{values.lol1}}",
|
|
"lol3": "{{values.lol2}}{{values.lol2}}{{values.lol2}}",
|
|
"foo": "bar",
|
|
"bar": "{{ metadata.annotations.foo.example.com }}",
|
|
"bat": "{{ metadata.labels.environment }}",
|
|
"aaa": "{{ server }}",
|
|
"no-op": "{{ this-does-not-exist }}",
|
|
}, expected: []map[string]any{
|
|
{"values.lol1": "lol", "values.lol2": "{{values.lol1}}{{values.lol1}}", "values.lol3": "{{values.lol2}}{{values.lol2}}{{values.lol2}}", "values.foo": "bar", "values.bar": "{{ metadata.annotations.foo.example.com }}", "values.no-op": "{{ this-does-not-exist }}", "values.bat": "{{ metadata.labels.environment }}", "values.aaa": "https://kubernetes.default.svc", "nameNormalized": "in-cluster", "name": "in-cluster", "server": "https://kubernetes.default.svc", "project": ""},
|
|
{
|
|
"values.lol1": "lol", "values.lol2": "{{values.lol1}}{{values.lol1}}", "values.lol3": "{{values.lol2}}{{values.lol2}}{{values.lol2}}", "values.foo": "bar", "values.bar": "production", "values.no-op": "{{ this-does-not-exist }}", "values.bat": "production", "values.aaa": "https://production-01.example.com", "name": "production_01/west", "nameNormalized": "production-01-west", "server": "https://production-01.example.com", "metadata.labels.environment": "production", "metadata.labels.org": "bar",
|
|
"metadata.labels.cd.hanzo.ai/secret-type": "cluster", "metadata.annotations.foo.example.com": "production", "project": "prod-project",
|
|
},
|
|
|
|
{
|
|
"values.lol1": "lol", "values.lol2": "{{values.lol1}}{{values.lol1}}", "values.lol3": "{{values.lol2}}{{values.lol2}}{{values.lol2}}", "values.foo": "bar", "values.bar": "staging", "values.no-op": "{{ this-does-not-exist }}", "values.bat": "staging", "values.aaa": "https://staging-01.example.com", "name": "staging-01", "nameNormalized": "staging-01", "server": "https://staging-01.example.com", "metadata.labels.environment": "staging", "metadata.labels.org": "foo",
|
|
"metadata.labels.cd.hanzo.ai/secret-type": "cluster", "metadata.annotations.foo.example.com": "staging", "project": "",
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "secret type label selector",
|
|
selector: metav1.LabelSelector{
|
|
MatchLabels: map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
},
|
|
},
|
|
values: nil,
|
|
expected: []map[string]any{
|
|
{
|
|
"name": "production_01/west", "nameNormalized": "production-01-west", "server": "https://production-01.example.com", "metadata.labels.environment": "production", "metadata.labels.org": "bar",
|
|
"metadata.labels.cd.hanzo.ai/secret-type": "cluster", "metadata.annotations.foo.example.com": "production", "project": "prod-project",
|
|
},
|
|
|
|
{
|
|
"name": "staging-01", "nameNormalized": "staging-01", "server": "https://staging-01.example.com", "metadata.labels.environment": "staging", "metadata.labels.org": "foo",
|
|
"metadata.labels.cd.hanzo.ai/secret-type": "cluster", "metadata.annotations.foo.example.com": "staging", "project": "",
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "production-only",
|
|
selector: metav1.LabelSelector{
|
|
MatchLabels: map[string]string{
|
|
"environment": "production",
|
|
},
|
|
},
|
|
values: map[string]string{
|
|
"foo": "bar",
|
|
},
|
|
expected: []map[string]any{
|
|
{
|
|
"values.foo": "bar", "name": "production_01/west", "nameNormalized": "production-01-west", "server": "https://production-01.example.com", "metadata.labels.environment": "production", "metadata.labels.org": "bar",
|
|
"metadata.labels.cd.hanzo.ai/secret-type": "cluster", "metadata.annotations.foo.example.com": "production", "project": "prod-project",
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "production or staging",
|
|
selector: metav1.LabelSelector{
|
|
MatchExpressions: []metav1.LabelSelectorRequirement{
|
|
{
|
|
Key: "environment",
|
|
Operator: "In",
|
|
Values: []string{
|
|
"production",
|
|
"staging",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
values: map[string]string{
|
|
"foo": "bar",
|
|
},
|
|
expected: []map[string]any{
|
|
{
|
|
"values.foo": "bar", "name": "staging-01", "nameNormalized": "staging-01", "server": "https://staging-01.example.com", "metadata.labels.environment": "staging", "metadata.labels.org": "foo",
|
|
"metadata.labels.cd.hanzo.ai/secret-type": "cluster", "metadata.annotations.foo.example.com": "staging", "project": "",
|
|
},
|
|
{
|
|
"values.foo": "bar", "name": "production_01/west", "nameNormalized": "production-01-west", "server": "https://production-01.example.com", "metadata.labels.environment": "production", "metadata.labels.org": "bar",
|
|
"metadata.labels.cd.hanzo.ai/secret-type": "cluster", "metadata.annotations.foo.example.com": "production", "project": "prod-project",
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "production or staging with match labels",
|
|
selector: metav1.LabelSelector{
|
|
MatchExpressions: []metav1.LabelSelectorRequirement{
|
|
{
|
|
Key: "environment",
|
|
Operator: "In",
|
|
Values: []string{
|
|
"production",
|
|
"staging",
|
|
},
|
|
},
|
|
},
|
|
MatchLabels: map[string]string{
|
|
"org": "foo",
|
|
},
|
|
},
|
|
values: map[string]string{
|
|
"name": "baz",
|
|
},
|
|
expected: []map[string]any{
|
|
{
|
|
"values.name": "baz", "name": "staging-01", "nameNormalized": "staging-01", "server": "https://staging-01.example.com", "metadata.labels.environment": "staging", "metadata.labels.org": "foo",
|
|
"metadata.labels.cd.hanzo.ai/secret-type": "cluster", "metadata.annotations.foo.example.com": "staging", "project": "",
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "simulate client error",
|
|
selector: metav1.LabelSelector{},
|
|
values: nil,
|
|
expected: nil,
|
|
clientError: true,
|
|
expectedError: errors.New("error getting cluster secrets: could not list Secrets"),
|
|
},
|
|
{
|
|
name: "flat mode without selectors",
|
|
selector: metav1.LabelSelector{},
|
|
values: map[string]string{
|
|
"lol1": "lol",
|
|
"lol2": "{{values.lol1}}{{values.lol1}}",
|
|
"lol3": "{{values.lol2}}{{values.lol2}}{{values.lol2}}",
|
|
"foo": "bar",
|
|
"bar": "{{ metadata.annotations.foo.example.com }}",
|
|
"bat": "{{ metadata.labels.environment }}",
|
|
"aaa": "{{ server }}",
|
|
"no-op": "{{ this-does-not-exist }}",
|
|
},
|
|
expected: []map[string]any{
|
|
{
|
|
"clusters": []map[string]any{
|
|
{"values.lol1": "lol", "values.lol2": "{{values.lol1}}{{values.lol1}}", "values.lol3": "{{values.lol2}}{{values.lol2}}{{values.lol2}}", "values.foo": "bar", "values.bar": "{{ metadata.annotations.foo.example.com }}", "values.no-op": "{{ this-does-not-exist }}", "values.bat": "{{ metadata.labels.environment }}", "values.aaa": "https://kubernetes.default.svc", "nameNormalized": "in-cluster", "name": "in-cluster", "server": "https://kubernetes.default.svc", "project": ""},
|
|
{
|
|
"values.lol1": "lol", "values.lol2": "{{values.lol1}}{{values.lol1}}", "values.lol3": "{{values.lol2}}{{values.lol2}}{{values.lol2}}", "values.foo": "bar", "values.bar": "production", "values.no-op": "{{ this-does-not-exist }}", "values.bat": "production", "values.aaa": "https://production-01.example.com", "name": "production_01/west", "nameNormalized": "production-01-west", "server": "https://production-01.example.com", "metadata.labels.environment": "production", "metadata.labels.org": "bar",
|
|
"metadata.labels.cd.hanzo.ai/secret-type": "cluster", "metadata.annotations.foo.example.com": "production", "project": "prod-project",
|
|
},
|
|
|
|
{
|
|
"values.lol1": "lol", "values.lol2": "{{values.lol1}}{{values.lol1}}", "values.lol3": "{{values.lol2}}{{values.lol2}}{{values.lol2}}", "values.foo": "bar", "values.bar": "staging", "values.no-op": "{{ this-does-not-exist }}", "values.bat": "staging", "values.aaa": "https://staging-01.example.com", "name": "staging-01", "nameNormalized": "staging-01", "server": "https://staging-01.example.com", "metadata.labels.environment": "staging", "metadata.labels.org": "foo",
|
|
"metadata.labels.cd.hanzo.ai/secret-type": "cluster", "metadata.annotations.foo.example.com": "staging", "project": "",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
isFlatMode: true,
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "production or staging with flat mode",
|
|
selector: metav1.LabelSelector{
|
|
MatchExpressions: []metav1.LabelSelectorRequirement{
|
|
{
|
|
Key: "environment",
|
|
Operator: "In",
|
|
Values: []string{
|
|
"production",
|
|
"staging",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
isFlatMode: true,
|
|
values: map[string]string{
|
|
"foo": "bar",
|
|
},
|
|
expected: []map[string]any{
|
|
{
|
|
"clusters": []map[string]any{
|
|
{
|
|
"values.foo": "bar", "name": "production_01/west", "nameNormalized": "production-01-west", "server": "https://production-01.example.com", "metadata.labels.environment": "production", "metadata.labels.org": "bar",
|
|
"metadata.labels.cd.hanzo.ai/secret-type": "cluster", "metadata.annotations.foo.example.com": "production", "project": "prod-project",
|
|
},
|
|
{
|
|
"values.foo": "bar", "name": "staging-01", "nameNormalized": "staging-01", "server": "https://staging-01.example.com", "metadata.labels.environment": "staging", "metadata.labels.org": "foo",
|
|
"metadata.labels.cd.hanzo.ai/secret-type": "cluster", "metadata.annotations.foo.example.com": "staging", "project": "",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
}
|
|
|
|
for _, testCase := range testCases {
|
|
t.Run(testCase.name, func(t *testing.T) {
|
|
fakeClient := fake.NewClientBuilder().WithObjects(clusters...).Build()
|
|
cl := &possiblyErroringFakeCtrlRuntimeClient{
|
|
fakeClient,
|
|
testCase.clientError,
|
|
}
|
|
|
|
clusterGenerator := NewClusterGenerator(cl, "namespace")
|
|
|
|
applicationSetInfo := argoprojiov1alpha1.ApplicationSet{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "set",
|
|
},
|
|
Spec: argoprojiov1alpha1.ApplicationSetSpec{},
|
|
}
|
|
|
|
got, err := clusterGenerator.GenerateParams(&argoprojiov1alpha1.ApplicationSetGenerator{
|
|
Clusters: &argoprojiov1alpha1.ClusterGenerator{
|
|
Selector: testCase.selector,
|
|
Values: testCase.values,
|
|
FlatList: testCase.isFlatMode,
|
|
},
|
|
}, &applicationSetInfo, nil)
|
|
|
|
if testCase.expectedError != nil {
|
|
require.EqualError(t, err, testCase.expectedError.Error())
|
|
} else {
|
|
require.NoError(t, err)
|
|
assertEqualParamsFlat(t, testCase.expected, got, testCase.isFlatMode)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func assertEqualParamsFlat(t *testing.T, expected, got []map[string]any, isFlatMode bool) {
|
|
t.Helper()
|
|
if isFlatMode && len(expected) == 1 && len(got) == 1 {
|
|
expectedClusters, ok1 := expected[0]["clusters"].([]map[string]any)
|
|
gotClusters, ok2 := got[0]["clusters"].([]map[string]any)
|
|
if ok1 && ok2 {
|
|
assert.ElementsMatch(t, expectedClusters, gotClusters)
|
|
return
|
|
}
|
|
}
|
|
assert.ElementsMatch(t, expected, got)
|
|
}
|
|
|
|
func TestGenerateParamsGoTemplate(t *testing.T) {
|
|
clusters := []client.Object{
|
|
&corev1.Secret{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "Secret",
|
|
APIVersion: "v1",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "staging-01",
|
|
Namespace: "namespace",
|
|
Labels: map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "staging",
|
|
"org": "foo",
|
|
},
|
|
Annotations: map[string]string{
|
|
"foo.example.com": "staging",
|
|
},
|
|
},
|
|
Data: map[string][]byte{
|
|
"config": []byte("{}"),
|
|
"name": []byte("staging-01"),
|
|
"server": []byte("https://staging-01.example.com"),
|
|
},
|
|
Type: corev1.SecretType("Opaque"),
|
|
},
|
|
&corev1.Secret{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "Secret",
|
|
APIVersion: "v1",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "production-01",
|
|
Namespace: "namespace",
|
|
Labels: map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "production",
|
|
"org": "bar",
|
|
},
|
|
Annotations: map[string]string{
|
|
"foo.example.com": "production",
|
|
},
|
|
},
|
|
Data: map[string][]byte{
|
|
"config": []byte("{}"),
|
|
"name": []byte("production_01/west"),
|
|
"server": []byte("https://production-01.example.com"),
|
|
},
|
|
Type: corev1.SecretType("Opaque"),
|
|
},
|
|
}
|
|
testCases := []struct {
|
|
name string
|
|
selector metav1.LabelSelector
|
|
values map[string]string
|
|
isFlatMode bool
|
|
expected []map[string]any
|
|
// clientError is true if a k8s client error should be simulated
|
|
clientError bool
|
|
expectedError error
|
|
}{
|
|
{
|
|
name: "no label selector",
|
|
selector: metav1.LabelSelector{},
|
|
values: map[string]string{
|
|
"lol1": "lol",
|
|
"lol2": "{{ .values.lol1 }}{{ .values.lol1 }}",
|
|
"lol3": "{{ .values.lol2 }}{{ .values.lol2 }}{{ .values.lol2 }}",
|
|
"foo": "bar",
|
|
"bar": "{{ if not (empty .metadata) }}{{index .metadata.annotations \"foo.example.com\" }}{{ end }}",
|
|
"bat": "{{ if not (empty .metadata) }}{{.metadata.labels.environment}}{{ end }}",
|
|
"aaa": "{{ .server }}",
|
|
"no-op": "{{ .thisDoesNotExist }}",
|
|
}, expected: []map[string]any{
|
|
{
|
|
"name": "production_01/west",
|
|
"nameNormalized": "production-01-west",
|
|
"server": "https://production-01.example.com",
|
|
"project": "",
|
|
"metadata": map[string]any{
|
|
"labels": map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "production",
|
|
"org": "bar",
|
|
},
|
|
"annotations": map[string]string{
|
|
"foo.example.com": "production",
|
|
},
|
|
},
|
|
"values": map[string]string{
|
|
"lol1": "lol",
|
|
"lol2": "<no value><no value>",
|
|
"lol3": "<no value><no value><no value>",
|
|
"foo": "bar",
|
|
"bar": "production",
|
|
"bat": "production",
|
|
"aaa": "https://production-01.example.com",
|
|
"no-op": "<no value>",
|
|
},
|
|
},
|
|
{
|
|
"name": "staging-01",
|
|
"nameNormalized": "staging-01",
|
|
"server": "https://staging-01.example.com",
|
|
"project": "",
|
|
"metadata": map[string]any{
|
|
"labels": map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "staging",
|
|
"org": "foo",
|
|
},
|
|
"annotations": map[string]string{
|
|
"foo.example.com": "staging",
|
|
},
|
|
},
|
|
"values": map[string]string{
|
|
"lol1": "lol",
|
|
"lol2": "<no value><no value>",
|
|
"lol3": "<no value><no value><no value>",
|
|
"foo": "bar",
|
|
"bar": "staging",
|
|
"bat": "staging",
|
|
"aaa": "https://staging-01.example.com",
|
|
"no-op": "<no value>",
|
|
},
|
|
},
|
|
{
|
|
"nameNormalized": "in-cluster",
|
|
"name": "in-cluster",
|
|
"server": "https://kubernetes.default.svc",
|
|
"project": "",
|
|
"values": map[string]string{
|
|
"lol1": "lol",
|
|
"lol2": "<no value><no value>",
|
|
"lol3": "<no value><no value><no value>",
|
|
"foo": "bar",
|
|
"bar": "",
|
|
"bat": "",
|
|
"aaa": "https://kubernetes.default.svc",
|
|
"no-op": "<no value>",
|
|
},
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "secret type label selector",
|
|
selector: metav1.LabelSelector{
|
|
MatchLabels: map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
},
|
|
},
|
|
values: nil,
|
|
expected: []map[string]any{
|
|
{
|
|
"name": "production_01/west",
|
|
"nameNormalized": "production-01-west",
|
|
"server": "https://production-01.example.com",
|
|
"project": "",
|
|
"metadata": map[string]any{
|
|
"labels": map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "production",
|
|
"org": "bar",
|
|
},
|
|
"annotations": map[string]string{
|
|
"foo.example.com": "production",
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"name": "staging-01",
|
|
"nameNormalized": "staging-01",
|
|
"server": "https://staging-01.example.com",
|
|
"project": "",
|
|
"metadata": map[string]any{
|
|
"labels": map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "staging",
|
|
"org": "foo",
|
|
},
|
|
"annotations": map[string]string{
|
|
"foo.example.com": "staging",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "production-only",
|
|
selector: metav1.LabelSelector{
|
|
MatchLabels: map[string]string{
|
|
"environment": "production",
|
|
},
|
|
},
|
|
values: map[string]string{
|
|
"foo": "bar",
|
|
},
|
|
expected: []map[string]any{
|
|
{
|
|
"name": "production_01/west",
|
|
"nameNormalized": "production-01-west",
|
|
"server": "https://production-01.example.com",
|
|
"project": "",
|
|
"metadata": map[string]any{
|
|
"labels": map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "production",
|
|
"org": "bar",
|
|
},
|
|
"annotations": map[string]string{
|
|
"foo.example.com": "production",
|
|
},
|
|
},
|
|
"values": map[string]string{
|
|
"foo": "bar",
|
|
},
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "production or staging",
|
|
selector: metav1.LabelSelector{
|
|
MatchExpressions: []metav1.LabelSelectorRequirement{
|
|
{
|
|
Key: "environment",
|
|
Operator: "In",
|
|
Values: []string{
|
|
"production",
|
|
"staging",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
values: map[string]string{
|
|
"foo": "bar",
|
|
},
|
|
expected: []map[string]any{
|
|
{
|
|
"name": "production_01/west",
|
|
"nameNormalized": "production-01-west",
|
|
"server": "https://production-01.example.com",
|
|
"project": "",
|
|
"metadata": map[string]any{
|
|
"labels": map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "production",
|
|
"org": "bar",
|
|
},
|
|
"annotations": map[string]string{
|
|
"foo.example.com": "production",
|
|
},
|
|
},
|
|
"values": map[string]string{
|
|
"foo": "bar",
|
|
},
|
|
},
|
|
{
|
|
"name": "staging-01",
|
|
"nameNormalized": "staging-01",
|
|
"server": "https://staging-01.example.com",
|
|
"project": "",
|
|
"metadata": map[string]any{
|
|
"labels": map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "staging",
|
|
"org": "foo",
|
|
},
|
|
"annotations": map[string]string{
|
|
"foo.example.com": "staging",
|
|
},
|
|
},
|
|
"values": map[string]string{
|
|
"foo": "bar",
|
|
},
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "production or staging with match labels",
|
|
selector: metav1.LabelSelector{
|
|
MatchExpressions: []metav1.LabelSelectorRequirement{
|
|
{
|
|
Key: "environment",
|
|
Operator: "In",
|
|
Values: []string{
|
|
"production",
|
|
"staging",
|
|
},
|
|
},
|
|
},
|
|
MatchLabels: map[string]string{
|
|
"org": "foo",
|
|
},
|
|
},
|
|
values: map[string]string{
|
|
"name": "baz",
|
|
},
|
|
expected: []map[string]any{
|
|
{
|
|
"name": "staging-01",
|
|
"nameNormalized": "staging-01",
|
|
"server": "https://staging-01.example.com",
|
|
"project": "",
|
|
"metadata": map[string]any{
|
|
"labels": map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "staging",
|
|
"org": "foo",
|
|
},
|
|
"annotations": map[string]string{
|
|
"foo.example.com": "staging",
|
|
},
|
|
},
|
|
"values": map[string]string{
|
|
"name": "baz",
|
|
},
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "simulate client error",
|
|
selector: metav1.LabelSelector{},
|
|
values: nil,
|
|
expected: nil,
|
|
clientError: true,
|
|
expectedError: errors.New("error getting cluster secrets: could not list Secrets"),
|
|
},
|
|
{
|
|
name: "Clusters with flat list mode and no selector",
|
|
selector: metav1.LabelSelector{},
|
|
isFlatMode: true,
|
|
values: map[string]string{
|
|
"lol1": "lol",
|
|
"lol2": "{{ .values.lol1 }}{{ .values.lol1 }}",
|
|
"lol3": "{{ .values.lol2 }}{{ .values.lol2 }}{{ .values.lol2 }}",
|
|
"foo": "bar",
|
|
"bar": "{{ if not (empty .metadata) }}{{index .metadata.annotations \"foo.example.com\" }}{{ end }}",
|
|
"bat": "{{ if not (empty .metadata) }}{{.metadata.labels.environment}}{{ end }}",
|
|
"aaa": "{{ .server }}",
|
|
"no-op": "{{ .thisDoesNotExist }}",
|
|
},
|
|
expected: []map[string]any{
|
|
{
|
|
"clusters": []map[string]any{
|
|
{
|
|
"nameNormalized": "in-cluster",
|
|
"name": "in-cluster",
|
|
"server": "https://kubernetes.default.svc",
|
|
"project": "",
|
|
"values": map[string]string{
|
|
"lol1": "lol",
|
|
"lol2": "<no value><no value>",
|
|
"lol3": "<no value><no value><no value>",
|
|
"foo": "bar",
|
|
"bar": "",
|
|
"bat": "",
|
|
"aaa": "https://kubernetes.default.svc",
|
|
"no-op": "<no value>",
|
|
},
|
|
},
|
|
{
|
|
"name": "production_01/west",
|
|
"nameNormalized": "production-01-west",
|
|
"server": "https://production-01.example.com",
|
|
"project": "",
|
|
"metadata": map[string]any{
|
|
"labels": map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "production",
|
|
"org": "bar",
|
|
},
|
|
"annotations": map[string]string{
|
|
"foo.example.com": "production",
|
|
},
|
|
},
|
|
"values": map[string]string{
|
|
"lol1": "lol",
|
|
"lol2": "<no value><no value>",
|
|
"lol3": "<no value><no value><no value>",
|
|
"foo": "bar",
|
|
"bar": "production",
|
|
"bat": "production",
|
|
"aaa": "https://production-01.example.com",
|
|
"no-op": "<no value>",
|
|
},
|
|
},
|
|
{
|
|
"name": "staging-01",
|
|
"nameNormalized": "staging-01",
|
|
"server": "https://staging-01.example.com",
|
|
"project": "",
|
|
"metadata": map[string]any{
|
|
"labels": map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "staging",
|
|
"org": "foo",
|
|
},
|
|
"annotations": map[string]string{
|
|
"foo.example.com": "staging",
|
|
},
|
|
},
|
|
"values": map[string]string{
|
|
"lol1": "lol",
|
|
"lol2": "<no value><no value>",
|
|
"lol3": "<no value><no value><no value>",
|
|
"foo": "bar",
|
|
"bar": "staging",
|
|
"bat": "staging",
|
|
"aaa": "https://staging-01.example.com",
|
|
"no-op": "<no value>",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "production or staging with flat mode",
|
|
selector: metav1.LabelSelector{
|
|
MatchExpressions: []metav1.LabelSelectorRequirement{
|
|
{
|
|
Key: "environment",
|
|
Operator: "In",
|
|
Values: []string{
|
|
"production",
|
|
"staging",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
isFlatMode: true,
|
|
values: map[string]string{
|
|
"foo": "bar",
|
|
},
|
|
expected: []map[string]any{
|
|
{
|
|
"clusters": []map[string]any{
|
|
{
|
|
"name": "production_01/west",
|
|
"nameNormalized": "production-01-west",
|
|
"server": "https://production-01.example.com",
|
|
"project": "",
|
|
"metadata": map[string]any{
|
|
"labels": map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "production",
|
|
"org": "bar",
|
|
},
|
|
"annotations": map[string]string{
|
|
"foo.example.com": "production",
|
|
},
|
|
},
|
|
"values": map[string]string{
|
|
"foo": "bar",
|
|
},
|
|
},
|
|
{
|
|
"name": "staging-01",
|
|
"nameNormalized": "staging-01",
|
|
"server": "https://staging-01.example.com",
|
|
"project": "",
|
|
"metadata": map[string]any{
|
|
"labels": map[string]string{
|
|
"cd.hanzo.ai/secret-type": "cluster",
|
|
"environment": "staging",
|
|
"org": "foo",
|
|
},
|
|
"annotations": map[string]string{
|
|
"foo.example.com": "staging",
|
|
},
|
|
},
|
|
"values": map[string]string{
|
|
"foo": "bar",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
clientError: false,
|
|
expectedError: nil,
|
|
},
|
|
}
|
|
|
|
for _, testCase := range testCases {
|
|
t.Run(testCase.name, func(t *testing.T) {
|
|
fakeClient := fake.NewClientBuilder().WithObjects(clusters...).Build()
|
|
cl := &possiblyErroringFakeCtrlRuntimeClient{
|
|
fakeClient,
|
|
testCase.clientError,
|
|
}
|
|
|
|
clusterGenerator := NewClusterGenerator(cl, "namespace")
|
|
|
|
applicationSetInfo := argoprojiov1alpha1.ApplicationSet{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "set",
|
|
},
|
|
Spec: argoprojiov1alpha1.ApplicationSetSpec{
|
|
GoTemplate: true,
|
|
},
|
|
}
|
|
|
|
got, err := clusterGenerator.GenerateParams(&argoprojiov1alpha1.ApplicationSetGenerator{
|
|
Clusters: &argoprojiov1alpha1.ClusterGenerator{
|
|
Selector: testCase.selector,
|
|
Values: testCase.values,
|
|
FlatList: testCase.isFlatMode,
|
|
},
|
|
}, &applicationSetInfo, nil)
|
|
|
|
if testCase.expectedError != nil {
|
|
require.EqualError(t, err, testCase.expectedError.Error())
|
|
} else {
|
|
require.NoError(t, err)
|
|
assertEqualParamsFlat(t, testCase.expected, got, testCase.isFlatMode)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestSanitizeClusterName(t *testing.T) {
|
|
t.Run("valid DNS-1123 subdomain name", func(t *testing.T) {
|
|
assert.Equal(t, "cluster-name", utils.SanitizeName("cluster-name"))
|
|
})
|
|
t.Run("invalid DNS-1123 subdomain name", func(t *testing.T) {
|
|
invalidName := "-.--CLUSTER/name -./.-"
|
|
assert.Equal(t, "cluster-name", utils.SanitizeName(invalidName))
|
|
})
|
|
}
|