643 lines
23 KiB
Go
643 lines
23 KiB
Go
/*
|
|
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),
|
|
}},
|
|
},
|
|
}
|
|
|
|
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
|
|
}
|