fix build image from exist tests

Author: CMGS

Makefile

@@ -34,7 +34,7 @@ mock: deps
 .ONESHELL:
 
 cloc:
-	cloc --exclude-dir=vendor,3rdmocks,mocks,tools --not-match-f=test .
+	cloc --exclude-dir=vendor,3rdmocks,mocks,tools,gen --not-match-f=test .
 
 unit-test:
 	go vet `go list ./... | grep -v '/vendor/' | grep -v '/tools'`

cluster/calcium/build_test.go

@@ -120,8 +120,9 @@ func TestBuild(t *testing.T) {
 	assert.Error(t, err)
 	// build from exist not implemented
 	opts.BuildMethod = types.BuildFromExist
+	opts.ExistID = "123"
 	engine.On("ImageBuildFromExist", mock.Anything, mock.Anything, mock.Anything).Return("", types.ErrEngineNotImplemented).Once()
-	store.On("GetWorkload", mock.Anything, mock.Anything).Return(&types.Workload{}, nil)
+	store.On("GetWorkload", mock.Anything, mock.Anything).Return(&types.Workload{Nodename: "123"}, nil)
 	store.On("GetNode", mock.Anything, mock.Anything).Return(&types.Node{Engine: engine}, nil)
 	ch, err = c.BuildImage(ctx, opts)
 	assert.NoError(t, err)

cluster/calcium/image_test.go

@@ -48,12 +48,12 @@ func TestRemoveImage(t *testing.T) {
 	engine.On("ImageRemove", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return([]string{"xx"}, nil)
 	// sucess remove but prune fail
 	engine.On("ImagesPrune", mock.Anything).Return(types.ErrBadStorage).Once()
-	ch, err = c.RemoveImage(ctx, &types.ImageOptions{Podname: "podname", Images: []string{"xx"}})
+	ch, err = c.RemoveImage(ctx, &types.ImageOptions{Podname: "podname", Images: []string{"xx"}, Prune: true})
 	for c := range ch {
 		assert.True(t, c.Success)
 	}
 	engine.On("ImagesPrune", mock.Anything).Return(nil)
-	ch, err = c.RemoveImage(ctx, &types.ImageOptions{Podname: "podname", Images: []string{"xx"}})
+	ch, err = c.RemoveImage(ctx, &types.ImageOptions{Podname: "podname", Images: []string{"xx"}, Prune: true})
 	for c := range ch {
 		assert.True(t, c.Success)
 	}