leaselock test

Signed-off-by: Frank Jogeleit <frank.jogeleit@lovoo.com>

pkg/leaderelection/cleant_test.go

@@ -0,0 +1,34 @@
+package leaderelection_test
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/kyverno/policy-reporter/pkg/leaderelection"
+	"k8s.io/client-go/kubernetes/typed/coordination/v1/fake"
+)
+
+func TestClient(t *testing.T) {
+	client := leaderelection.New(&fake.FakeCoordinationV1{}, "policy-reporter", "namespace", "pod-123", time.Second, time.Second, time.Second, false)
+
+	if client == nil {
+		t.Fatal("failed to create leaderelection client")
+	}
+
+	client.RegisterOnNew(func(currentID, lockID string) {})
+	client.RegisterOnStart(func(c context.Context) {})
+	client.RegisterOnStop(func() {})
+
+	lock := client.CreateLock()
+
+	if lock.LeaseMeta.Name != "policy-reporter" {
+		t.Error("unexpected lease name")
+	}
+	if lock.LeaseMeta.Namespace != "namespace" {
+		t.Error("unexpected lease namespace")
+	}
+	if lock.LockConfig.Identity != "pod-123" {
+		t.Error("unexpected lease identity")
+	}
+}

pkg/leaderelection/client.go

@@ -46,7 +46,7 @@ func (c *Client) RegisterOnNew(callback func(currentID string, lockID string)) *
 
 func (c *Client) Run(ctx context.Context) error {
 	k8sleaderelection.RunOrDie(ctx, k8sleaderelection.LeaderElectionConfig{
-		Lock:            c.createLock(),
+		Lock:            c.CreateLock(),
 		ReleaseOnCancel: c.releaseOnCancel,
 		LeaseDuration:   c.leaseDuration,
 		RenewDeadline:   c.renewDeadline,
@@ -63,7 +63,7 @@ func (c *Client) Run(ctx context.Context) error {
 	return errors.New("leaderelection stopped")
 }
 
-func (c *Client) createLock() *resourcelock.LeaseLock {
+func (c *Client) CreateLock() *resourcelock.LeaseLock {
 	return &resourcelock.LeaseLock{
 		LeaseMeta: metav1.ObjectMeta{
 			Name:      c.lockName,