-
Notifications
You must be signed in to change notification settings - Fork 1.1k
/
finalizer_test.go
50 lines (41 loc) · 1.24 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
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() {
Describe("Finalizer", func() {
It("should return no error", func() {
var err error
now := metav1.Now()
pod := &corev1.Pod{
ObjectMeta: metav1.ObjectMeta{
Finalizers: []string{"finalizers.sigs.k8s.io/testfinalizer"},
DeletionTimestamp: &now,
},
}
finalizers := NewFinalizers()
f := mockFinalizer{}
err = finalizers.Register("finalizers.sigs.k8s.io/testfinalizer", f)
Expect(err).To(BeNil())
ret, err := finalizers.Finalize(context.TODO(), pod)
Expect(err).To(BeNil())
Expect(ret).To(BeTrue())
})
})
})