/*
Copyright 2017 The Kubernetes Authors.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package polymorphichelpers

import (
	"fmt"
	"reflect"
	"testing"
	"time"

	"github.com/google/go-cmp/cmp"
	appsv1 "k8s.io/api/apps/v1"
	batchv1 "k8s.io/api/batch/v1"
	corev1 "k8s.io/api/core/v1"
	extensionsv1beta1 "k8s.io/api/extensions/v1beta1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/apimachinery/pkg/runtime/schema"
	fakeexternal "k8s.io/client-go/kubernetes/fake"
	testclient "k8s.io/client-go/testing"
	"k8s.io/kubectl/pkg/cmd/util/podcmd"
)

var (
	podsResource = schema.GroupVersionResource{Version: "v1", Resource: "pods"}
	podsKind     = schema.GroupVersionKind{Version: "v1", Kind: "Pod"}
)

func TestLogsForObject(t *testing.T) {
	tests := []struct {
		name          string
		obj           runtime.Object
		opts          *corev1.PodLogOptions
		allContainers bool
		allPods       bool
		clientsetPods []runtime.Object
		actions       []testclient.Action

		expectedErr     string
		expectedSources []corev1.ObjectReference
	}{
		{
			name: "pod logs",
			obj:  testPodWithOneContainers(),
			actions: []testclient.Action{
				getLogsAction("test", &corev1.PodLogOptions{Container: "c1"}),
			},
			expectedSources: []corev1.ObjectReference{
				{
					Kind:       testPodWithOneContainers().Kind,
					APIVersion: testPodWithOneContainers().APIVersion,
					Name:       testPodWithOneContainers().Name,
					Namespace:  testPodWithOneContainers().Namespace,
					FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithOneContainers().Spec.Containers[0].Name),
				},
			},
		},
		{
			name:          "pod logs: all containers",
			obj:           testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers(),
			opts:          &corev1.PodLogOptions{},
			allContainers: true,
			allPods:       false,
			actions: []testclient.Action{
				getLogsAction("test", &corev1.PodLogOptions{Container: "foo-2-and-2-and-1-initc1"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "foo-2-and-2-and-1-initc2"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "foo-2-and-2-and-1-c1"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "foo-2-and-2-and-1-c2"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "foo-2-and-2-and-1-e1"}),
			},
			expectedSources: []corev1.ObjectReference{
				{
					Kind:       testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Kind,
					APIVersion: testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().APIVersion,
					Name:       testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Name,
					Namespace:  testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Namespace,
					FieldPath:  fmt.Sprintf("spec.initContainers{%s}", testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Spec.InitContainers[0].Name),
				},
				{
					Kind:       testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Kind,
					APIVersion: testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().APIVersion,
					Name:       testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Name,
					Namespace:  testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Namespace,
					FieldPath:  fmt.Sprintf("spec.initContainers{%s}", testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Spec.InitContainers[1].Name),
				},
				{
					Kind:       testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Kind,
					APIVersion: testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().APIVersion,
					Name:       testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Name,
					Namespace:  testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Namespace,
					FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Spec.Containers[0].Name),
				},
				{
					Kind:       testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Kind,
					APIVersion: testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().APIVersion,
					Name:       testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Name,
					Namespace:  testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Namespace,
					FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Spec.Containers[1].Name),
				},
				{
					Kind:       testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Kind,
					APIVersion: testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().APIVersion,
					Name:       testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Name,
					Namespace:  testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Namespace,
					FieldPath:  fmt.Sprintf("spec.ephemeralContainers{%s}", testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Spec.EphemeralContainers[0].Name),
				},
			},
		},
		{
			name: "pod logs: default to first container",
			obj:  testPodWithTwoContainers(),
			actions: []testclient.Action{
				getLogsAction("test", &corev1.PodLogOptions{Container: "foo-2-c1"}),
			},
			expectedSources: []corev1.ObjectReference{
				{
					Kind:       testPodWithTwoContainers().Kind,
					APIVersion: testPodWithTwoContainers().APIVersion,
					Name:       testPodWithTwoContainers().Name,
					Namespace:  testPodWithTwoContainers().Namespace,
					FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithTwoContainers().Spec.Containers[0].Name),
				},
			},
		},
		{
			name: "pods list logs",
			obj: &corev1.PodList{
				Items: []corev1.Pod{*testPodWithOneContainers()},
			},
			actions: []testclient.Action{
				getLogsAction("test", &corev1.PodLogOptions{Container: "c1"}),
			},
			expectedSources: []corev1.ObjectReference{{
				Kind:       testPodWithOneContainers().Kind,
				APIVersion: testPodWithOneContainers().APIVersion,
				Name:       testPodWithOneContainers().Name,
				Namespace:  testPodWithOneContainers().Namespace,
				FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithOneContainers().Spec.Containers[0].Name),
			}},
		},
		{
			name: "pods list logs: default container should not leak across pods",
			obj: &corev1.PodList{
				Items: []corev1.Pod{
					{
						TypeMeta: metav1.TypeMeta{
							Kind:       "pod",
							APIVersion: "v1",
						},
						ObjectMeta: metav1.ObjectMeta{
							Name:      "foo",
							Namespace: "test",
							Labels:    map[string]string{"test": "logs"},
							Annotations: map[string]string{
								"kubectl.kubernetes.io/default-container": "c1",
							},
						},
						Spec: corev1.PodSpec{
							RestartPolicy: corev1.RestartPolicyAlways,
							DNSPolicy:     corev1.DNSClusterFirst,
							Containers: []corev1.Container{
								{Name: "c1"},
								{Name: "c2"},
							},
						},
					},
					{
						TypeMeta: metav1.TypeMeta{
							Kind:       "pod",
							APIVersion: "v1",
						},
						ObjectMeta: metav1.ObjectMeta{
							Name:      "bar",
							Namespace: "test",
							Labels:    map[string]string{"test": "logs"},
						},
						Spec: corev1.PodSpec{
							RestartPolicy: corev1.RestartPolicyAlways,
							DNSPolicy:     corev1.DNSClusterFirst,
							Containers: []corev1.Container{
								{Name: "c2"},
							},
						},
					},
				},
			},
			actions: []testclient.Action{
				getLogsAction("test", &corev1.PodLogOptions{Container: "c1"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "c2"}),
			},
			expectedSources: []corev1.ObjectReference{{
				Kind:       "pod",
				APIVersion: "v1",
				Name:       "foo",
				Namespace:  "test",
				FieldPath:  fmt.Sprintf("spec.containers{%s}", "c1"),
			}, {
				Kind:       "pod",
				APIVersion: "v1",
				Name:       "bar",
				Namespace:  "test",
				FieldPath:  fmt.Sprintf("spec.containers{%s}", "c2"),
			}},
		},
		{
			name: "pods list logs: all containers",
			obj: &corev1.PodList{
				Items: []corev1.Pod{*testPodWithTwoContainersAndTwoInitContainers()},
			},
			opts:          &corev1.PodLogOptions{},
			allContainers: true,
			allPods:       false,
			actions: []testclient.Action{
				getLogsAction("test", &corev1.PodLogOptions{Container: "foo-2-and-2-initc1"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "foo-2-and-2-initc2"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "foo-2-and-2-c1"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "foo-2-and-2-c2"}),
			},
			expectedSources: []corev1.ObjectReference{
				{
					Kind:       testPodWithTwoContainersAndTwoInitContainers().Kind,
					APIVersion: testPodWithTwoContainersAndTwoInitContainers().APIVersion,
					Name:       testPodWithTwoContainersAndTwoInitContainers().Name,
					Namespace:  testPodWithTwoContainersAndTwoInitContainers().Namespace,
					FieldPath:  fmt.Sprintf("spec.initContainers{%s}", testPodWithTwoContainersAndTwoInitContainers().Spec.InitContainers[0].Name),
				},
				{
					Kind:       testPodWithTwoContainersAndTwoInitContainers().Kind,
					APIVersion: testPodWithTwoContainersAndTwoInitContainers().APIVersion,
					Name:       testPodWithTwoContainersAndTwoInitContainers().Name,
					Namespace:  testPodWithTwoContainersAndTwoInitContainers().Namespace,
					FieldPath:  fmt.Sprintf("spec.initContainers{%s}", testPodWithTwoContainersAndTwoInitContainers().Spec.InitContainers[1].Name),
				},
				{
					Kind:       testPodWithTwoContainersAndTwoInitContainers().Kind,
					APIVersion: testPodWithTwoContainersAndTwoInitContainers().APIVersion,
					Name:       testPodWithTwoContainersAndTwoInitContainers().Name,
					Namespace:  testPodWithTwoContainersAndTwoInitContainers().Namespace,
					FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithTwoContainersAndTwoInitContainers().Spec.Containers[0].Name),
				},
				{
					Kind:       testPodWithTwoContainersAndTwoInitContainers().Kind,
					APIVersion: testPodWithTwoContainersAndTwoInitContainers().APIVersion,
					Name:       testPodWithTwoContainersAndTwoInitContainers().Name,
					Namespace:  testPodWithTwoContainersAndTwoInitContainers().Namespace,
					FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithTwoContainersAndTwoInitContainers().Spec.Containers[1].Name),
				},
			},
		},
		{
			name: "pods list logs: default to first container",
			obj: &corev1.PodList{
				Items: []corev1.Pod{*testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers()},
			},
			actions: []testclient.Action{
				getLogsAction("test", &corev1.PodLogOptions{Container: "foo-2-and-2-and-1-c1"}),
			},
			expectedSources: []corev1.ObjectReference{
				{
					Kind:       testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Kind,
					APIVersion: testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().APIVersion,
					Name:       testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Name,
					Namespace:  testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Namespace,
					FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers().Spec.Containers[0].Name),
				},
			},
		},
		{
			name: "replication controller logs",
			obj: &corev1.ReplicationController{
				ObjectMeta: metav1.ObjectMeta{Name: "hello", Namespace: "test"},
				Spec: corev1.ReplicationControllerSpec{
					Selector: map[string]string{"foo": "bar"},
				},
			},
			clientsetPods: []runtime.Object{testPodWithOneContainers()},
			actions: []testclient.Action{
				testclient.NewListAction(podsResource, podsKind, "test", metav1.ListOptions{LabelSelector: "foo=bar"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "c1"}),
			},
			expectedSources: []corev1.ObjectReference{{
				Kind:       testPodWithOneContainers().Kind,
				APIVersion: testPodWithOneContainers().APIVersion,
				Name:       testPodWithOneContainers().Name,
				Namespace:  testPodWithOneContainers().Namespace,
				FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithOneContainers().Spec.Containers[0].Name),
			}},
		},
		{
			name: "replica set logs",
			obj: &extensionsv1beta1.ReplicaSet{
				ObjectMeta: metav1.ObjectMeta{Name: "hello", Namespace: "test"},
				Spec: extensionsv1beta1.ReplicaSetSpec{
					Selector: &metav1.LabelSelector{MatchLabels: map[string]string{"foo": "bar"}},
				},
			},
			clientsetPods: []runtime.Object{testPodWithOneContainers()},
			actions: []testclient.Action{
				testclient.NewListAction(podsResource, podsKind, "test", metav1.ListOptions{LabelSelector: "foo=bar"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "c1"}),
			},
			expectedSources: []corev1.ObjectReference{{
				Kind:       testPodWithOneContainers().Kind,
				APIVersion: testPodWithOneContainers().APIVersion,
				Name:       testPodWithOneContainers().Name,
				Namespace:  testPodWithOneContainers().Namespace,
				FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithOneContainers().Spec.Containers[0].Name),
			}},
		},
		{
			name: "deployment logs",
			obj: &extensionsv1beta1.Deployment{
				ObjectMeta: metav1.ObjectMeta{Name: "hello", Namespace: "test"},
				Spec: extensionsv1beta1.DeploymentSpec{
					Selector: &metav1.LabelSelector{MatchLabels: map[string]string{"foo": "bar"}},
				},
			},
			clientsetPods: []runtime.Object{testPodWithOneContainers()},
			actions: []testclient.Action{
				testclient.NewListAction(podsResource, podsKind, "test", metav1.ListOptions{LabelSelector: "foo=bar"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "c1"}),
			},
			expectedSources: []corev1.ObjectReference{{
				Kind:       testPodWithOneContainers().Kind,
				APIVersion: testPodWithOneContainers().APIVersion,
				Name:       testPodWithOneContainers().Name,
				Namespace:  testPodWithOneContainers().Namespace,
				FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithOneContainers().Spec.Containers[0].Name),
			}},
		},
		{
			name: "job logs",
			obj: &batchv1.Job{
				ObjectMeta: metav1.ObjectMeta{Name: "hello", Namespace: "test"},
				Spec: batchv1.JobSpec{
					Selector: &metav1.LabelSelector{MatchLabels: map[string]string{"foo": "bar"}},
				},
			},
			clientsetPods: []runtime.Object{testPodWithOneContainers()},
			actions: []testclient.Action{
				testclient.NewListAction(podsResource, podsKind, "test", metav1.ListOptions{LabelSelector: "foo=bar"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "c1"}),
			},
			expectedSources: []corev1.ObjectReference{{
				Kind:       testPodWithOneContainers().Kind,
				APIVersion: testPodWithOneContainers().APIVersion,
				Name:       testPodWithOneContainers().Name,
				Namespace:  testPodWithOneContainers().Namespace,
				FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithOneContainers().Spec.Containers[0].Name),
			}},
		},
		{
			name: "stateful set logs",
			obj: &appsv1.StatefulSet{
				ObjectMeta: metav1.ObjectMeta{Name: "hello", Namespace: "test"},
				Spec: appsv1.StatefulSetSpec{
					Selector: &metav1.LabelSelector{MatchLabels: map[string]string{"foo": "bar"}},
				},
			},
			clientsetPods: []runtime.Object{testPodWithOneContainers()},
			actions: []testclient.Action{
				testclient.NewListAction(podsResource, podsKind, "test", metav1.ListOptions{LabelSelector: "foo=bar"}),
				getLogsAction("test", &corev1.PodLogOptions{Container: "c1"}),
			},
			expectedSources: []corev1.ObjectReference{{
				Kind:       testPodWithOneContainers().Kind,
				APIVersion: testPodWithOneContainers().APIVersion,
				Name:       testPodWithOneContainers().Name,
				Namespace:  testPodWithOneContainers().Namespace,
				FieldPath:  fmt.Sprintf("spec.containers{%s}", testPodWithOneContainers().Spec.Containers[0].Name),
			}},
		},
		{
			name:        "pod logs: invalid container name",
			obj:         testPodWithTwoContainers(),
			opts:        &corev1.PodLogOptions{Container: "nonexistent"},
			expectedErr: "container nonexistent is not valid for pod foo-two-containers out of: foo-2-c1, foo-2-c2",
		},
	}

	for _, test := range tests {
		fakeClientset := fakeexternal.NewSimpleClientset(test.clientsetPods...)
		responses, err := logsForObjectWithClient(fakeClientset.CoreV1(), test.obj, test.opts, 20*time.Second, test.allContainers, test.allPods)
		if test.expectedErr == "" && err != nil {
			t.Errorf("%s: unexpected error: %v", test.name, err)
			continue
		}

		if err != nil && test.expectedErr != err.Error() {
			t.Errorf("%s: expected error: %v, got: %v", test.name, test.expectedErr, err)
			continue
		}

		if len(test.expectedSources) != len(responses) {
			t.Errorf(
				"%s: the number of expected sources doesn't match the number of responses: %v, got: %v",
				test.name,
				len(test.expectedSources),
				len(responses),
			)
			continue
		}

		for _, ref := range test.expectedSources {
			if _, ok := responses[ref]; !ok {
				t.Errorf("%s: didn't find expected log source object reference: %#v", test.name, ref)
			}
		}

		var i int
		for i = range test.actions {
			if len(fakeClientset.Actions()) < i {
				t.Errorf("%s: expected action %d does not exists in actual actions: %#v",
					test.name, i, fakeClientset.Actions())
				continue
			}
			got := fakeClientset.Actions()[i]
			want := test.actions[i]
			if !reflect.DeepEqual(got, want) {
				t.Errorf("%s: unexpected diff for action: %s", test.name, cmp.Diff(got, want))
			}
		}
		for i++; i < len(fakeClientset.Actions()); i++ {
			t.Errorf("%s: actual action %d does not exist in expected: %v", test.name, i, fakeClientset.Actions()[i])
		}
	}
}

func testPodWithOneContainers() *corev1.Pod {
	return &corev1.Pod{
		TypeMeta: metav1.TypeMeta{
			Kind:       "pod",
			APIVersion: "v1",
		},
		ObjectMeta: metav1.ObjectMeta{
			Name:      "foo",
			Namespace: "test",
			Labels:    map[string]string{"foo": "bar"},
		},
		Spec: corev1.PodSpec{
			RestartPolicy: corev1.RestartPolicyAlways,
			DNSPolicy:     corev1.DNSClusterFirst,
			Containers: []corev1.Container{
				{Name: "c1"},
			},
		},
	}
}

func testPodWithTwoContainers() *corev1.Pod {
	return &corev1.Pod{
		TypeMeta: metav1.TypeMeta{
			Kind:       "pod",
			APIVersion: "v1",
		},
		ObjectMeta: metav1.ObjectMeta{
			Name:      "foo-two-containers",
			Namespace: "test",
			Labels:    map[string]string{"foo": "bar"},
		},
		Spec: corev1.PodSpec{
			RestartPolicy: corev1.RestartPolicyAlways,
			DNSPolicy:     corev1.DNSClusterFirst,
			Containers: []corev1.Container{
				{Name: "foo-2-c1"},
				{Name: "foo-2-c2"},
			},
		},
	}
}

func testPodWithTwoContainersAndTwoInitContainers() *corev1.Pod {
	return &corev1.Pod{
		TypeMeta: metav1.TypeMeta{
			Kind:       "pod",
			APIVersion: "v1",
		},
		ObjectMeta: metav1.ObjectMeta{
			Name:      "foo-two-containers-and-two-init-containers",
			Namespace: "test",
		},
		Spec: corev1.PodSpec{
			InitContainers: []corev1.Container{
				{Name: "foo-2-and-2-initc1"},
				{Name: "foo-2-and-2-initc2"},
			},
			Containers: []corev1.Container{
				{Name: "foo-2-and-2-c1"},
				{Name: "foo-2-and-2-c2"},
			},
		},
	}
}

func TestLogsForObjectWithClient(t *testing.T) {
	cases := []struct {
		name              string
		podFn             func() *corev1.Pod
		podLogOptions     *corev1.PodLogOptions
		expectedFieldPath string
		allContainers     bool
		allPods           bool
		expectedError     string
	}{
		{
			name:              "two container pod without default container selected should default to the first one",
			podFn:             testPodWithTwoContainers,
			podLogOptions:     &corev1.PodLogOptions{},
			expectedFieldPath: `spec.containers{foo-2-c1}`,
		},
		{
			name: "two container pod with default container selected",
			podFn: func() *corev1.Pod {
				pod := testPodWithTwoContainers()
				pod.Annotations = map[string]string{podcmd.DefaultContainerAnnotationName: "foo-2-c1"}
				return pod
			},
			podLogOptions:     &corev1.PodLogOptions{},
			expectedFieldPath: `spec.containers{foo-2-c1}`,
		},
		{
			name: "two container pod with default container selected but also container set explicitly",
			podFn: func() *corev1.Pod {
				pod := testPodWithTwoContainers()
				pod.Annotations = map[string]string{podcmd.DefaultContainerAnnotationName: "foo-2-c1"}
				return pod
			},
			podLogOptions: &corev1.PodLogOptions{
				Container: "foo-2-c2",
			},
			expectedFieldPath: `spec.containers{foo-2-c2}`,
		},
		{
			name: "two container pod with non-existing default container selected should default to the first container",
			podFn: func() *corev1.Pod {
				pod := testPodWithTwoContainers()
				pod.Annotations = map[string]string{podcmd.DefaultContainerAnnotationName: "non-existing"}
				return pod
			},
			podLogOptions:     &corev1.PodLogOptions{},
			expectedFieldPath: `spec.containers{foo-2-c1}`,
		},
		{
			name: "two container pod with default container set, but allContainers also set",
			podFn: func() *corev1.Pod {
				pod := testPodWithTwoContainers()
				pod.Annotations = map[string]string{podcmd.DefaultContainerAnnotationName: "foo-2-c1"}
				return pod
			},
			allContainers:     true,
			allPods:           false,
			podLogOptions:     &corev1.PodLogOptions{},
			expectedFieldPath: `spec.containers{foo-2-c2}`,
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			pod := tc.podFn()
			fakeClientset := fakeexternal.NewSimpleClientset(pod)
			responses, err := logsForObjectWithClient(fakeClientset.CoreV1(), pod, tc.podLogOptions, 20*time.Second, tc.allContainers, tc.allPods)
			if err != nil {
				if len(tc.expectedError) > 0 {
					if err.Error() == tc.expectedError {
						return
					}
				}
				t.Errorf("unexpected error: %v", err)
				return
			}
			if len(tc.expectedError) > 0 {
				t.Errorf("expected error %q, got none", tc.expectedError)
				return
			}
			if !tc.allContainers && len(responses) != 1 {
				t.Errorf("expected one response, got %d", len(responses))
				return
			}
			if tc.allContainers && len(responses) != 2 {
				t.Errorf("expected 2 responses for allContainers, got %d", len(responses))
				return
			}
			// do not check actual responses in this case as we know there are at least two, which means the preselected
			// container was not used (which is desired).
			if tc.allContainers {
				return
			}
			for r := range responses {
				if r.FieldPath != tc.expectedFieldPath {
					t.Errorf("expected %q container to be preselected, got %q", tc.expectedFieldPath, r.FieldPath)
				}
			}
		})
	}

}

func testPodWithTwoContainersAndTwoInitAndOneEphemeralContainers() *corev1.Pod {
	return &corev1.Pod{
		TypeMeta: metav1.TypeMeta{
			Kind:       "pod",
			APIVersion: "v1",
		},
		ObjectMeta: metav1.ObjectMeta{
			Name:      "foo-two-containers-and-two-init-containers",
			Namespace: "test",
		},
		Spec: corev1.PodSpec{
			InitContainers: []corev1.Container{
				{Name: "foo-2-and-2-and-1-initc1"},
				{Name: "foo-2-and-2-and-1-initc2"},
			},
			Containers: []corev1.Container{
				{Name: "foo-2-and-2-and-1-c1"},
				{Name: "foo-2-and-2-and-1-c2"},
			},
			EphemeralContainers: []corev1.EphemeralContainer{
				{
					EphemeralContainerCommon: corev1.EphemeralContainerCommon{Name: "foo-2-and-2-and-1-e1"},
				},
			},
		},
	}
}

func getLogsAction(namespace string, opts *corev1.PodLogOptions) testclient.Action {
	action := testclient.GenericActionImpl{}
	action.Verb = "get"
	action.Namespace = namespace
	action.Resource = podsResource
	action.Subresource = "log"
	action.Value = opts
	return action
}
