@@ -148,14 +148,12 @@ var _ = Describe("updateStatusByPullPods", func() {
148148 })
149149
150150 It ("pod failed, build config present" , func () {
151- pullPod := v1.Pod {
152- Status : v1.PodStatus {
153- Phase : v1 .PodFailed ,
154- },
155- }
151+ pullPod := v1.Pod {}
152+
156153 gomock .InOrder (
157154 mockImagePuller .EXPECT ().GetPullPodImage (pullPod ).Return ("image 1" ),
158155 micHelper .EXPECT ().GetModuleImageSpec (& testMic , "image 1" ).Return (& testMic .Spec .Images [0 ]),
156+ mockImagePuller .EXPECT ().GetPullPodStatus (& pullPod ).Return (pod .PullImageFailed ),
159157 micHelper .EXPECT ().SetImageStatus (& testMic , "image 1" , kmmv1beta1 .ImageNeedsBuilding ),
160158 clnt .EXPECT ().Status ().Return (statusWriter ),
161159 statusWriter .EXPECT ().Patch (ctx , & testMic , gomock .Any ()),
@@ -166,14 +164,12 @@ var _ = Describe("updateStatusByPullPods", func() {
166164 })
167165
168166 It ("pod failed, build config not present, sign config present" , func () {
169- pullPod := v1.Pod {
170- Status : v1.PodStatus {
171- Phase : v1 .PodFailed ,
172- },
173- }
167+ pullPod := v1.Pod {}
168+
174169 gomock .InOrder (
175170 mockImagePuller .EXPECT ().GetPullPodImage (pullPod ).Return ("image 2" ),
176171 micHelper .EXPECT ().GetModuleImageSpec (& testMic , "image 2" ).Return (& testMic .Spec .Images [1 ]),
172+ mockImagePuller .EXPECT ().GetPullPodStatus (& pullPod ).Return (pod .PullImageFailed ),
177173 micHelper .EXPECT ().SetImageStatus (& testMic , "image 2" , kmmv1beta1 .ImageNeedsSigning ),
178174 clnt .EXPECT ().Status ().Return (statusWriter ),
179175 statusWriter .EXPECT ().Patch (ctx , & testMic , gomock .Any ()),
@@ -183,15 +179,13 @@ var _ = Describe("updateStatusByPullPods", func() {
183179 Expect (err ).To (BeNil ())
184180 })
185181
186- It ("pod failed, build or sign config is not present" , func () {
187- pullPod := v1.Pod {
188- Status : v1.PodStatus {
189- Phase : v1 .PodFailed ,
190- },
191- }
182+ It ("pod failed, build or sign configs are not present" , func () {
183+ pullPod := v1.Pod {}
184+
192185 gomock .InOrder (
193186 mockImagePuller .EXPECT ().GetPullPodImage (pullPod ).Return ("image 3" ),
194187 micHelper .EXPECT ().GetModuleImageSpec (& testMic , "image 3" ).Return (& testMic .Spec .Images [2 ]),
188+ mockImagePuller .EXPECT ().GetPullPodStatus (& pullPod ).Return (pod .PullImageFailed ),
195189 clnt .EXPECT ().Status ().Return (statusWriter ),
196190 statusWriter .EXPECT ().Patch (ctx , & testMic , gomock .Any ()),
197191 mockImagePuller .EXPECT ().DeletePod (ctx , & pullPod ).Return (nil ),
@@ -201,14 +195,12 @@ var _ = Describe("updateStatusByPullPods", func() {
201195 })
202196
203197 It ("pod succeeded" , func () {
204- pullPod := v1.Pod {
205- Status : v1.PodStatus {
206- Phase : v1 .PodSucceeded ,
207- },
208- }
198+ pullPod := v1.Pod {}
199+
209200 gomock .InOrder (
210201 mockImagePuller .EXPECT ().GetPullPodImage (pullPod ).Return ("image 2" ),
211202 micHelper .EXPECT ().GetModuleImageSpec (& testMic , "image 2" ).Return (& testMic .Spec .Images [1 ]),
203+ mockImagePuller .EXPECT ().GetPullPodStatus (& pullPod ).Return (pod .PullImageSuccess ),
212204 micHelper .EXPECT ().SetImageStatus (& testMic , "image 2" , kmmv1beta1 .ImageExists ),
213205 clnt .EXPECT ().Status ().Return (statusWriter ),
214206 statusWriter .EXPECT ().Patch (ctx , & testMic , gomock .Any ()),
0 commit comments