@@ -87,6 +87,7 @@ var _ = Describe("UpdateDockerfileCmd", func() {
8787 c .Opts .Baseimage = "workspace-agent-24.04.tar"
8888 c .Opts .Force = false
8989
90+ mockPackageManager .EXPECT ().Extract (false ).Return (nil )
9091 mockPackageManager .EXPECT ().GetBaseimageName ("workspace-agent-24.04.tar" ).Return ("" , errors .New ("failed to extract image" ))
9192
9293 err := c .UpdateDockerfile (mockPackageManager , mockImageManager , []string {})
@@ -103,6 +104,7 @@ var _ = Describe("UpdateDockerfileCmd", func() {
103104 c .Opts .Baseimage = ""
104105 c .Opts .Force = false
105106
107+ mockPackageManager .EXPECT ().Extract (false ).Return (nil )
106108 mockPackageManager .EXPECT ().GetBaseimageName ("" ).Return ("ubuntu:24.04" , nil )
107109 mockPackageManager .EXPECT ().GetBaseimagePath ("" , false ).Return ("/test/workdir/deps/codesphere/images/workspace-agent-24.04.tar" , nil )
108110 mockImageManager .EXPECT ().LoadImage ("/test/workdir/deps/codesphere/images/workspace-agent-24.04.tar" ).Return (nil )
@@ -122,6 +124,7 @@ var _ = Describe("UpdateDockerfileCmd", func() {
122124 c .Opts .Baseimage = ""
123125 c .Opts .Force = false
124126
127+ mockPackageManager .EXPECT ().Extract (false ).Return (nil )
125128 mockPackageManager .EXPECT ().GetBaseimageName ("" ).Return ("ubuntu:24.04" , nil )
126129 mockPackageManager .EXPECT ().GetBaseimagePath ("" , false ).Return ("/test/workdir/deps/codesphere/images/workspace-agent-24.04.tar" , nil )
127130 mockImageManager .EXPECT ().LoadImage ("/test/workdir/deps/codesphere/images/workspace-agent-24.04.tar" ).Return (errors .New ("load failed" ))
@@ -152,6 +155,7 @@ var _ = Describe("UpdateDockerfileCmd", func() {
152155 c .Opts .Baseimage = ""
153156 c .Opts .Force = false
154157
158+ mockPackageManager .EXPECT ().Extract (false ).Return (nil )
155159 mockPackageManager .EXPECT ().GetBaseimageName ("" ).Return ("ubuntu:24.04" , nil )
156160 mockPackageManager .EXPECT ().GetBaseimagePath ("" , false ).Return ("/test/workdir/deps/codesphere/images/workspace-agent-24.04.tar" , nil )
157161 mockImageManager .EXPECT ().LoadImage ("/test/workdir/deps/codesphere/images/workspace-agent-24.04.tar" ).Return (nil )
@@ -185,6 +189,7 @@ var _ = Describe("UpdateDockerfileCmd", func() {
185189 c .Opts .Baseimage = ""
186190 c .Opts .Force = false
187191
192+ mockPackageManager .EXPECT ().Extract (false ).Return (nil )
188193 mockPackageManager .EXPECT ().GetBaseimageName ("" ).Return ("ubuntu:24.04" , nil )
189194 mockPackageManager .EXPECT ().GetBaseimagePath ("" , false ).Return ("/test/workdir/deps/codesphere/images/workspace-agent-24.04.tar" , nil )
190195 mockPackageManager .EXPECT ().FileIO ().Return (mockFileIO )
@@ -217,6 +222,7 @@ var _ = Describe("UpdateDockerfileCmd", func() {
217222 c .Opts .Baseimage = "workspace-agent-20.04.tar"
218223 c .Opts .Force = true
219224
225+ mockPackageManager .EXPECT ().Extract (true ).Return (nil )
220226 mockPackageManager .EXPECT ().GetBaseimageName ("workspace-agent-20.04.tar" ).Return ("ubuntu:20.04" , nil )
221227 mockPackageManager .EXPECT ().GetBaseimagePath ("workspace-agent-20.04.tar" , true ).Return ("/test/workdir/deps/codesphere/images/workspace-agent-20.04.tar" , nil )
222228 mockPackageManager .EXPECT ().FileIO ().Return (mockFileIO )
@@ -249,6 +255,7 @@ var _ = Describe("UpdateDockerfileCmd", func() {
249255 c .Opts .Baseimage = "workspace-agent-24.04.tar"
250256 c .Opts .Force = false
251257
258+ mockPackageManager .EXPECT ().Extract (false ).Return (nil )
252259 mockPackageManager .EXPECT ().GetBaseimageName ("workspace-agent-24.04.tar" ).Return ("registry.example.com/workspace-agent:24.04" , nil )
253260 mockPackageManager .EXPECT ().GetBaseimagePath ("workspace-agent-24.04.tar" , false ).Return ("/test/workdir/deps/codesphere/images/workspace-agent-24.04.tar" , nil )
254261 mockPackageManager .EXPECT ().FileIO ().Return (mockFileIO )
0 commit comments