Function TestPreStart_SuccessTwoHooksInOrder

pkg/compose/pre_start_test.go:70–106

(t *testing.T)

}

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

	project := &types.Project{Name: "demo"}
	service := types.ServiceConfig{
		Name:  "web",
		Image: "alpine",
		PreStart: []types.ServiceHook{
			{Image: "alpine", Command: types.ShellCommand{"echo", "first"}},
			{Image: "alpine", Command: types.ShellCommand{"echo", "second"}},
		},
	}
	ctr := container.Summary{ID: "service-ctr-id"}

	// Hook 1: create → wait (subscribe) → logs (subscribe) → start.
	create1 := apiClient.EXPECT().ContainerCreate(gomock.Any(), gomock.Any()).
		Return(client.ContainerCreateResult{ID: "hook-1"}, nil)
	wait1 := apiClient.EXPECT().ContainerWait(gomock.Any(), "hook-1", gomock.Any()).
		Return(waitResultExit(0)).After(create1)
	logs1 := apiClient.EXPECT().ContainerLogs(gomock.Any(), "hook-1", gomock.Any()).
		Return(emptyLogs(), nil).After(wait1)
	start1 := apiClient.EXPECT().ContainerStart(gomock.Any(), "hook-1", gomock.Any()).
		Return(client.ContainerStartResult{}, nil).After(logs1)

	// Hook 2 is only created after hook 1 has been started (and waited on).
	create2 := apiClient.EXPECT().ContainerCreate(gomock.Any(), gomock.Any()).
		Return(client.ContainerCreateResult{ID: "hook-2"}, nil).After(start1)
	wait2 := apiClient.EXPECT().ContainerWait(gomock.Any(), "hook-2", gomock.Any()).
		Return(waitResultExit(0)).After(create2)
	logs2 := apiClient.EXPECT().ContainerLogs(gomock.Any(), "hook-2", gomock.Any()).
		Return(emptyLogs(), nil).After(wait2)
	apiClient.EXPECT().ContainerStart(gomock.Any(), "hook-2", gomock.Any()).
		Return(client.ContainerStartResult{}, nil).After(logs2)

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

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