Function TestPreStart_ImageFallsBackToBuiltImage

pkg/compose/pre_start_test.go:154–183

(t *testing.T)

}

func TestPreStart_ImageFallsBackToBuiltImage(t *testing.T) {
	tested, apiClient := newPreStartTestService(t)

	project := &types.Project{Name: "demo"}
	// Service with no explicit image (build-only); hook image also empty.
	service := types.ServiceConfig{
		Name: "web",
		PreStart: []types.ServiceHook{
			{Command: types.ShellCommand{"echo", "hi"}},
		},
	}
	ctr := container.Summary{ID: "service-ctr-id"}

	var gotImage string
	apiClient.EXPECT().ContainerCreate(gomock.Any(), gomock.Any()).
		DoAndReturn(func(_ any, opts client.ContainerCreateOptions) (client.ContainerCreateResult, error) {
			gotImage = opts.Config.Image
			return client.ContainerCreateResult{ID: "hook-1"}, nil
		})
	apiClient.EXPECT().ContainerStart(gomock.Any(), "hook-1", gomock.Any()).
		Return(client.ContainerStartResult{}, nil)
	apiClient.EXPECT().ContainerLogs(gomock.Any(), "hook-1", gomock.Any()).
		Return(emptyLogs(), nil)
	apiClient.EXPECT().ContainerWait(gomock.Any(), "hook-1", gomock.Any()).
		Return(waitResultExit(0))

	err := tested.runPreStart(t.Context(), project, service, ctr, func(api.ContainerEvent) {})
	assert.NilError(t, err)
	assert.Equal(t, gotImage, api.GetImageNameOrDefault(service, project.Name))
}

func TestPreStart_ExplicitHookImageUsed(t *testing.T) {
	tested, apiClient := newPreStartTestService(t)