Adding new functions to MIC interface

Adding 2 function to the interface, that will be later used by
controllers, among then MIC controller

Author: yevgeny-shnaidman

internal/mic/mic.go

@@ -18,6 +18,9 @@ import (
 type MIC interface {
 	ApplyMIC(ctx context.Context, name, ns string, images []kmmv1beta1.ModuleImageSpec,
 		imageRepoSecret *v1.LocalObjectReference, owner metav1.Object) error
+	GetModuleImageSpec(micObj *kmmv1beta1.ModuleImagesConfig, image string) *kmmv1beta1.ModuleImageSpec
+	SetImageStatus(micObj *kmmv1beta1.ModuleImagesConfig, image string, status kmmv1beta1.ImageState)
+	GetImageState(micObj *kmmv1beta1.ModuleImagesConfig, image string) kmmv1beta1.ImageState
 }
 
 type micImpl struct {
@@ -61,3 +64,35 @@ func (mici *micImpl) ApplyMIC(ctx context.Context, name, ns string, images []kmm
 
 	return nil
 }
+
+func (mici *micImpl) GetModuleImageSpec(micObj *kmmv1beta1.ModuleImagesConfig, image string) *kmmv1beta1.ModuleImageSpec {
+	for i, imageSpec := range micObj.Spec.Images {
+		if imageSpec.Image == image {
+			return &micObj.Spec.Images[i]
+		}
+	}
+	return nil
+}
+
+func (mici *micImpl) SetImageStatus(micObj *kmmv1beta1.ModuleImagesConfig, image string, status kmmv1beta1.ImageState) {
+	imageState := kmmv1beta1.ModuleImageState{
+		Image:  image,
+		Status: status,
+	}
+	for i, imageStatus := range micObj.Status.ImagesStates {
+		if imageStatus.Image == image {
+			micObj.Status.ImagesStates[i] = imageState
+			return
+		}
+	}
+	micObj.Status.ImagesStates = append(micObj.Status.ImagesStates, imageState)
+}
+
+func (mici *micImpl) GetImageState(micObj *kmmv1beta1.ModuleImagesConfig, image string) kmmv1beta1.ImageState {
+	for _, imageState := range micObj.Status.ImagesStates {
+		if imageState.Image == image {
+			return imageState.Status
+		}
+	}
+	return ""
+}

internal/mic/mic_test.go

@@ -131,3 +131,112 @@ var _ = Describe("ApplyMIC", func() {
 		Expect(err).NotTo(HaveOccurred())
 	})
 })
+
+var _ = Describe("GetModuleImageSpec", func() {
+	var (
+		micAPI MIC
+	)
+
+	BeforeEach(func() {
+		micAPI = NewModuleImagesConfigAPI(nil, nil)
+	})
+
+	testMic := kmmv1beta1.ModuleImagesConfig{
+		Spec: kmmv1beta1.ModuleImagesConfigSpec{
+			Images: []kmmv1beta1.ModuleImageSpec{
+				{
+					Image: "image 1",
+				},
+				{
+					Image: "image 2",
+				},
+			},
+		},
+	}
+
+	It("check image present and not present scenarious", func() {
+
+		By("image spec is present")
+		res := micAPI.GetModuleImageSpec(&testMic, "image 1")
+		Expect(res).ToNot(BeNil())
+		Expect(res.Image).To(Equal("image 1"))
+
+		By("image spec is not present")
+		res = micAPI.GetModuleImageSpec(&testMic, "image 3")
+		Expect(res).To(BeNil())
+	})
+})
+
+var _ = Describe("SetImageStatus", func() {
+	var (
+		micAPI MIC
+	)
+
+	BeforeEach(func() {
+		micAPI = NewModuleImagesConfigAPI(nil, nil)
+	})
+
+	testMic := kmmv1beta1.ModuleImagesConfig{
+		Status: kmmv1beta1.ModuleImagesConfigStatus{
+			ImagesStates: []kmmv1beta1.ModuleImageState{
+				{
+					Image:  "image 1",
+					Status: kmmv1beta1.ImageDoesNotExist,
+				},
+				{
+					Image:  "image 2",
+					Status: kmmv1beta1.ImageExists,
+				},
+			},
+		},
+	}
+
+	It("set images status for both present and not present statuses", func() {
+
+		By("image status is present")
+		micAPI.SetImageStatus(&testMic, "image 1", kmmv1beta1.ImageExists)
+		Expect(testMic.Status.ImagesStates[0].Image).To(Equal("image 1"))
+		Expect(testMic.Status.ImagesStates[0].Status).To(Equal(kmmv1beta1.ImageExists))
+
+		By("image status is not present")
+		micAPI.SetImageStatus(&testMic, "image 3", kmmv1beta1.ImageDoesNotExist)
+		Expect(testMic.Status.ImagesStates[2].Image).To(Equal("image 3"))
+		Expect(testMic.Status.ImagesStates[2].Status).To(Equal(kmmv1beta1.ImageDoesNotExist))
+	})
+})
+
+var _ = Describe("GetImageState", func() {
+	var (
+		micAPI MIC
+	)
+
+	BeforeEach(func() {
+		micAPI = NewModuleImagesConfigAPI(nil, nil)
+	})
+
+	testMic := kmmv1beta1.ModuleImagesConfig{
+		Status: kmmv1beta1.ModuleImagesConfigStatus{
+			ImagesStates: []kmmv1beta1.ModuleImageState{
+				{
+					Image:  "image 1",
+					Status: kmmv1beta1.ImageDoesNotExist,
+				},
+				{
+					Image:  "image 2",
+					Status: kmmv1beta1.ImageExists,
+				},
+			},
+		},
+	}
+
+	It("get images status for both present and not present statuses", func() {
+
+		By("image status is present")
+		res := micAPI.GetImageState(&testMic, "image 1")
+		Expect(res).To(Equal(kmmv1beta1.ImageDoesNotExist))
+
+		By("image status is not present")
+		res = micAPI.GetImageState(&testMic, "image 3")
+		Expect(res).To(BeEmpty())
+	})
+})