-
Notifications
You must be signed in to change notification settings - Fork 1.1k
/
finalizer_test.go
92 lines (74 loc) · 2.66 KB
/
finalizer_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package finalizer
import (
"context"
"testing"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
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/envtest/printer"
)
type mockFinalizer struct{}
func (f mockFinalizer) Finalize(context.Context, client.Object) (needsUpdate bool, err error) {
return true, nil
}
func TestFinalizer(t *testing.T) {
RegisterFailHandler(Fail)
suiteName := "Finalizer Suite"
RunSpecsWithDefaultAndCustomReporters(t, suiteName, []Reporter{printer.NewlineReporter{}, printer.NewProwReporter(suiteName)})
}
var _ = Describe("TestFinalizer", func() {
var err error
var pod *corev1.Pod
var finalizers Finalizers
var f mockFinalizer
BeforeEach(func() {
pod = &corev1.Pod{
ObjectMeta: metav1.ObjectMeta{
Finalizers: []string{"finalizers.sigs.k8s.io/testfinalizer"},
},
}
Expect(pod).NotTo(BeNil())
finalizers = NewFinalizers()
Expect(finalizers).NotTo(BeNil())
f := mockFinalizer{}
Expect(f).NotTo(BeNil())
})
Describe("Finalizer helper library", func() {
It("Register should successfully register a finalizer", func() {
err = finalizers.Register("finalizers.sigs.k8s.io/testfinalizer", f)
Expect(err).To(BeNil())
})
It("Register should return an error when it is called twice with the same key", func() {
err = finalizers.Register("finalizers.sigs.k8s.io/testfinalizer", f)
Expect(err).To(BeNil())
// calling Register again with the same key should return an error
err = finalizers.Register("finalizers.sigs.k8s.io/testfinalizer", f)
Expect(err).NotTo(BeNil())
Expect(err.Error()).To(ContainSubstring("already registered"))
})
It("Finalize should return no error and indicate false for needsUpdate if a finalizer is not registered", func() {
ret, err := finalizers.Finalize(context.TODO(), pod)
Expect(err).To(BeNil())
Expect(ret).To(BeFalse())
})
It("Finalize should return no error when deletion timestamp is not nil and the finalizer exists", func() {
now := metav1.Now()
pod.DeletionTimestamp = &now
err = finalizers.Register("finalizers.sigs.k8s.io/testfinalizer", f)
Expect(err).To(BeNil())
_, err := finalizers.Finalize(context.TODO(), pod)
Expect(err).To(BeNil())
})
It("Finalize should return no error when deletion timestamp is nil and finalizer does not exist", func() {
err = finalizers.Register("finalizers.sigs.k8s.io/testfinalizer", f)
Expect(err).To(BeNil())
pod.DeletionTimestamp = nil
pod.Finalizers = []string{}
ret, err := finalizers.Finalize(context.TODO(), pod)
Expect(err).To(BeNil())
Expect(ret).To(BeTrue())
})
})
})