Skip to content

Commit 3a340cd

Browse files
yevgeny-shnaidmank8s-ci-robot
authored andcommitted
Changing the GarbageCollector API of the buildsign manager
Currently the GarbageCollector API receives the PodType as an input parameter. The PodType is defined in the pod package, which should not be used by the callee, since it is a porting layer. This commit changes the API to use kmmv1beta1.BuildOrSignAction, and the translation to the PodType will be done inside the manager code
1 parent 768a613 commit 3a340cd

File tree

4 files changed

+14
-10
lines changed

4 files changed

+14
-10
lines changed

internal/buildsign/manager.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,5 +15,5 @@ type Manager interface {
1515
GetStatus(ctx context.Context, name, namespace, kernelVersion string,
1616
action kmmv1beta1.BuildOrSignAction, owner metav1.Object) (kmmv1beta1.BuildOrSignStatus, error)
1717
Sync(ctx context.Context, mld *api.ModuleLoaderData, pushImage bool, owner metav1.Object, action kmmv1beta1.BuildOrSignAction) error
18-
GarbageCollect(ctx context.Context, name, namespace, podType string, owner metav1.Object) ([]string, error)
18+
GarbageCollect(ctx context.Context, name, namespace string, action kmmv1beta1.BuildOrSignAction, owner metav1.Object) ([]string, error)
1919
}

internal/buildsign/mock_manager.go

Lines changed: 4 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

internal/buildsign/pod/manager.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,11 @@ func (pm *podManager) Sync(ctx context.Context, mld *api.ModuleLoaderData, pushI
120120
return nil
121121
}
122122

123-
func (pm *podManager) GarbageCollect(ctx context.Context, name, namespace, podType string, owner metav1.Object) ([]string, error) {
123+
func (pm *podManager) GarbageCollect(ctx context.Context, name, namespace string, action kmmv1beta1.BuildOrSignAction, owner metav1.Object) ([]string, error) {
124+
podType := PodTypeBuild
125+
if action == kmmv1beta1.SignImage {
126+
podType = PodTypeSign
127+
}
124128
pods, err := pm.buildSignPodManager.GetModulePods(ctx, name, namespace, podType, owner)
125129
if err != nil {
126130
return nil, fmt.Errorf("failed to get %s pods for mbsc %s/%s: %v", podType, namespace, name, err)

internal/buildsign/pod/manager_test.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -265,7 +265,7 @@ var _ = Describe("GarbageCollect", func() {
265265
It("failed to get module pods", func() {
266266
mockBuildSignPodManager.EXPECT().GetModulePods(ctx, mbscName, mbscNamespace, PodTypeBuild, &testMBSC).Return(nil, fmt.Errorf("some error"))
267267

268-
_, err := mgr.GarbageCollect(ctx, mbscName, mbscNamespace, PodTypeBuild, &testMBSC)
268+
_, err := mgr.GarbageCollect(ctx, mbscName, mbscNamespace, kmmv1beta1.BuildImage, &testMBSC)
269269
Expect(err).To(HaveOccurred())
270270
})
271271

@@ -276,12 +276,12 @@ var _ = Describe("GarbageCollect", func() {
276276
},
277277
}
278278
gomock.InOrder(
279-
mockBuildSignPodManager.EXPECT().GetModulePods(ctx, mbscName, mbscNamespace, PodTypeBuild, &testMBSC).
279+
mockBuildSignPodManager.EXPECT().GetModulePods(ctx, mbscName, mbscNamespace, PodTypeSign, &testMBSC).
280280
Return([]v1.Pod{testPod}, nil),
281281
mockBuildSignPodManager.EXPECT().DeletePod(ctx, &testPod).Return(fmt.Errorf("some error")),
282282
)
283283

284-
_, err := mgr.GarbageCollect(ctx, mbscName, mbscNamespace, PodTypeBuild, &testMBSC)
284+
_, err := mgr.GarbageCollect(ctx, mbscName, mbscNamespace, kmmv1beta1.SignImage, &testMBSC)
285285
Expect(err).To(HaveOccurred())
286286
})
287287

@@ -313,7 +313,7 @@ var _ = Describe("GarbageCollect", func() {
313313
mockBuildSignPodManager.EXPECT().DeletePod(ctx, &testPodSuccess).Return(nil),
314314
)
315315

316-
res, err := mgr.GarbageCollect(ctx, mbscName, mbscNamespace, PodTypeBuild, &testMBSC)
316+
res, err := mgr.GarbageCollect(ctx, mbscName, mbscNamespace, kmmv1beta1.BuildImage, &testMBSC)
317317
Expect(err).To(BeNil())
318318
Expect(res).To(Equal([]string{"podSuccess"}))
319319
})

0 commit comments

Comments
 (0)