Skip to content

Commit

Permalink
WIP: unit tests
Browse files Browse the repository at this point in the history
Signed-off-by: Rashmi Gottipati <chowdary.grashmi@gmail.com>
  • Loading branch information
rashmigottipati committed Apr 20, 2021
1 parent 5e5ea56 commit 97f97b2
Show file tree
Hide file tree
Showing 2 changed files with 52 additions and 2 deletions.
4 changes: 2 additions & 2 deletions pkg/finalizer/finalizer.go
Expand Up @@ -21,8 +21,8 @@ import (
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
)

func NewFinalizers() (Finalizers, error) {
return finalizers(map[string]Finalizer{}), nil
func NewFinalizers() Finalizers {
return finalizers(map[string]Finalizer{})
}

func (f finalizers) Register(key string, finalizer Finalizer) error {
Expand Down
50 changes: 50 additions & 0 deletions pkg/finalizer/finalizer_test.go
@@ -0,0 +1,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())
})
})
})

0 comments on commit 97f97b2

Please sign in to comment.