public_git
/
helmfile
mirror of https://github.com/helmfile/helmfile.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add unittests for new preapply functions 

Signed-off-by: Anton Bretting <sajfer@gmail.com>
This commit is contained in:
Anton Bretting 2022-06-12 12:56:34 +02:00 committed by Yusuke Kuoka
parent ea56061609
commit fcdd852153
1 changed files with 176 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

177
pkg/app/app_apply_hooks_test.go
View File

@ -9,10 +9,14 @@ import (
"testing"
"github.com/google/go-cmp/cmp"
"github.com/helmfile/helmfile/pkg/event"
"github.com/helmfile/helmfile/pkg/exectest"
"github.com/helmfile/helmfile/pkg/helmexec"
"github.com/helmfile/helmfile/pkg/state"
"github.com/helmfile/helmfile/pkg/testhelper"
"github.com/stretchr/testify/require"
"github.com/variantdev/vals"
"k8s.io/utils/pointer"
)
func TestApply_hooks(t *testing.T) {
@ -263,5 +267,176 @@ releases:
logLevel: "info",
})
})
}
func TestGetReleasesWithPreApply(t *testing.T) {
tests := []struct {
releases []state.ReleaseSpec
preApplyreleases []state.ReleaseSpec
}{
{
[]state.ReleaseSpec{
{
Chart: "foo/bar",
Name: "foobar",
Namespace: "foobar",
Hooks: []event.Hook{
{
Events: []string{
"prepare",
"preapply",
},
},
},
},
{
Chart: "foo/foo",
Name: "foo",
Namespace: "foo",
Hooks: []event.Hook{
{
Events: []string{
"prepare",
"presync",
},
},
},
},
{
Chart: "bar/bar",
Name: "bar",
Namespace: "bar",
Hooks: []event.Hook{
{
Events: []string{
"preapply",
},
},
},
},
},
[]state.ReleaseSpec{
{
Chart: "foo/bar",
Name: "foobar",
Namespace: "foobar",
Hooks: []event.Hook{
{
Events: []string{
"prepare",
"preapply",
},
},
},
},
{
Chart: "bar/bar",
Name: "bar",
Namespace: "bar",
Hooks: []event.Hook{
{
Events: []string{
"preapply",
},
},
},
},
},
},
}
for _, test := range tests {
preApply := getReleasesWithPreApply(test.releases)
require.Equal(t, test.preApplyreleases, preApply)
}
}
func TestPreApplyInfoMsg(t *testing.T) {
tests := []struct {
preApplyreleases []state.ReleaseSpec
infoMsg *string
expected *string
}{
{
[]state.ReleaseSpec{
{
Chart: "foo/bar",
Name: "foobar",
Namespace: "foobar",
Hooks: []event.Hook{
{
Events: []string{
"prepare",
"preapply",
},
},
},
},
{
Chart: "bar/bar",
Name: "bar",
Namespace: "bar",
Hooks: []event.Hook{
{
Events: []string{
"preapply",
},
},
},
},
},
pointer.String("infoMsg\n"),
pointer.String(`infoMsg
Releases with preapply hooks:
foobar (foo/bar)
bar (bar/bar)
`),
},
{
[]state.ReleaseSpec{
{
Chart: "foo/bar",
Name: "foobar",
Namespace: "foobar",
Hooks: []event.Hook{
{
Events: []string{
"prepare",
"preapply",
},
},
},
},
{
Chart: "bar/bar",
Name: "bar",
Namespace: "bar",
Hooks: []event.Hook{
{
Events: []string{
"preapply",
},
},
},
},
},
pointer.String(""),
pointer.String(`
Releases with preapply hooks:
foobar (foo/bar)
bar (bar/bar)
`),
},
{
[]state.ReleaseSpec{},
pointer.String(""),
pointer.String(""),
},
}
for _, test := range tests {
infoMsg := preApplyInfoMsg(test.preApplyreleases, test.infoMsg)
require.Equal(t, *test.expected, *infoMsg)
}
}