Files
argo-cd/util/hook/helm/delete_policy_test.go
2019-09-30 15:04:13 -07:00

24 lines
979 B
Go

package helm
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/argoproj/argo-cd/pkg/apis/application/v1alpha1"
. "github.com/argoproj/argo-cd/test"
)
func TestDeletePolicies(t *testing.T) {
assert.Nil(t, DeletePolicies(NewPod()))
assert.Equal(t, []DeletePolicy{BeforeHookCreation}, DeletePolicies(Annotate(NewPod(), "helm.sh/hook-delete-policy", "before-hook-creation")))
assert.Equal(t, []DeletePolicy{HookSucceeded}, DeletePolicies(Annotate(NewPod(), "helm.sh/hook-delete-policy", "hook-succeeded")))
assert.Equal(t, []DeletePolicy{HookFailed}, DeletePolicies(Annotate(NewPod(), "helm.sh/hook-delete-policy", "hook-failed")))
}
func TestDeletePolicy_DeletePolicy(t *testing.T) {
assert.Equal(t, v1alpha1.HookDeletePolicyBeforeHookCreation, BeforeHookCreation.DeletePolicy())
assert.Equal(t, v1alpha1.HookDeletePolicyHookSucceeded, HookSucceeded.DeletePolicy())
assert.Equal(t, v1alpha1.HookDeletePolicyHookFailed, HookFailed.DeletePolicy())
}