Function TestCreateContainerImagePullPolicy

cli/command/container/create_test.go:93–168

(t *testing.T)

}

func TestCreateContainerImagePullPolicy(t *testing.T) {
	const (
		imageName   = "does-not-exist-locally"
		containerID = "abcdef"
	)
	config := &containerConfig{
		Config: &container.Config{
			Image: imageName,
		},
		HostConfig: &container.HostConfig{},
	}

	cases := []struct {
		PullPolicy      string
		ExpectedPulls   int
		ExpectedID      string
		ExpectedErrMsg  string
		ResponseCounter int
	}{
		{
			PullPolicy:    PullImageMissing,
			ExpectedPulls: 1,
			ExpectedID:    containerID,
		}, {
			PullPolicy:      PullImageAlways,
			ExpectedPulls:   1,
			ExpectedID:      containerID,
			ResponseCounter: 1, // This lets us return a container on the first pull
		}, {
			PullPolicy:     PullImageNever,
			ExpectedPulls:  0,
			ExpectedErrMsg: "error fake not found",
		},
	}
	for _, tc := range cases {
		t.Run(tc.PullPolicy, func(t *testing.T) {
			pullCounter := 0

			apiClient := &fakeClient{
				createContainerFunc: func(options client.ContainerCreateOptions) (client.ContainerCreateResult, error) {
					defer func() { tc.ResponseCounter++ }()
					switch tc.ResponseCounter {
					case 0:
						return client.ContainerCreateResult{}, fakeNotFound{}
					default:
						return client.ContainerCreateResult{ID: containerID}, nil
					}
				},
				imagePullFunc: func(ctx context.Context, parentReference string, options client.ImagePullOptions) (client.ImagePullResponse, error) {
					defer func() { pullCounter++ }()
					return fakeStreamResult{ReadCloser: io.NopCloser(strings.NewReader(""))}, nil
				},
				infoFunc: func() (client.SystemInfoResult, error) {
					return client.SystemInfoResult{
						Info: system.Info{IndexServerAddress: "https://indexserver.example.com"},
					}, nil
				},
			}