Added tests

Signed-off-by: Jakub Dzon <[email protected]>

Author: Jakub Dzon

internal/edgeapi/backend/api.go

@@ -13,6 +13,7 @@ type Notification struct {
 	Retry     int32
 }
 
+//go:generate mockgen -package=backend -destination=mock_heartbeat-handler.go . HeartbeatHandler
 type HeartbeatHandler interface {
 	Process(ctx context.Context, notification Notification) (bool, error)
 }

internal/edgeapi/backend/factory/factoru_suite_test.go

@@ -0,0 +1,13 @@
+package factory_test
+
+import (
+	"testing"
+
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+)
+
+func TestBackendFactory(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "Backend Factory Suite")
+}

internal/edgeapi/backend/factory/factory_test.go

@@ -0,0 +1,29 @@
+package factory_test
+
+import (
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+	"go.uber.org/zap"
+	"k8s.io/client-go/tools/record"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+
+	"github.com/project-flotta/flotta-operator/internal/edgeapi/backend/factory"
+)
+
+const namespace = "some-ns"
+
+var _ = Describe("Backend factory", func() {
+
+	It("should create k8s factory", func() {
+		// given
+		logger := &zap.SugaredLogger{}
+		eventRecorder := record.NewFakeRecorder(1)
+		var c client.Client
+
+		// when
+		backend := factory.CreateBackend(namespace, c, logger, eventRecorder)
+
+		// then
+		Expect(backend).ToNot(BeNil())
+	})
+})

internal/edgeapi/backend/mock_heartbeat-handler.go

@@ -0,0 +1,87 @@
+package yggdrasil_test
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/golang/mock/gomock"
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+
+	"github.com/project-flotta/flotta-operator/internal/edgeapi/backend"
+	"github.com/project-flotta/flotta-operator/internal/edgeapi/yggdrasil"
+)
+
+var _ = Describe("Heartbeat handler", func() {
+
+	var (
+		mockCtrl     *gomock.Controller
+		mockDelegate *backend.MockHeartbeatHandler
+		handler      *yggdrasil.RetryingDelegatingHandler
+	)
+
+	BeforeEach(func() {
+		mockCtrl = gomock.NewController(GinkgoT())
+		mockDelegate = backend.NewMockHeartbeatHandler(mockCtrl)
+
+		handler = yggdrasil.NewRetryingDelegatingHandler(mockDelegate)
+	})
+
+	AfterEach(func() {
+		mockCtrl.Finish()
+	})
+
+	It("should call delegate", func() {
+		// given
+		ctx := context.TODO()
+		notification := backend.Notification{DeviceID: "1234"}
+
+		mockDelegate.EXPECT().
+			Process(ctx, notification).
+			Return(false, nil)
+
+		// when
+		err := handler.Process(ctx, notification)
+
+		// then
+		Expect(err).ToNot(HaveOccurred())
+		// mock verification
+	})
+
+	It("should retry calling delegate on error and succeed", func() {
+		// given
+		ctx := context.TODO()
+		notification := backend.Notification{DeviceID: "1234"}
+		retryNotification := backend.Notification{DeviceID: "1234", Retry: 1}
+		errorCall := mockDelegate.EXPECT().
+			Process(ctx, notification).
+			Return(true, fmt.Errorf("boom"))
+		mockDelegate.EXPECT().
+			Process(ctx, retryNotification).
+			Return(false, nil).
+			After(errorCall)
+
+		// when
+		err := handler.Process(ctx, notification)
+
+		// then
+		Expect(err).ToNot(HaveOccurred())
+	})
+
+	It("should retry calling delegate on error and eventually fail", func() {
+		// given
+		ctx := context.TODO()
+		notification := backend.Notification{DeviceID: "1234"}
+
+		mockDelegate.EXPECT().
+			Process(ctx, gomock.AssignableToTypeOf(notification)).
+			Return(true, fmt.Errorf("boom")).
+			Times(4)
+
+		// when
+		err := handler.Process(ctx, notification)
+
+		// then
+		Expect(err).To(HaveOccurred())
+	})
+})