internal/rds/kubernetes/pods_test.go

// Copyright 2019 The Cloudprober 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 kubernetes

import (
	"os"
	"reflect"
	"testing"

	pb "github.com/cloudprober/cloudprober/internal/rds/proto"
	"github.com/stretchr/testify/assert"
	"google.golang.org/protobuf/proto"
)

func testPodInfo(name, ns, ip string, labels map[string]string) *podInfo {
	labels["namespace"] = ns
	pi := &podInfo{Metadata: kMetadata{Name: name, Namespace: ns, Labels: labels}}
	pi.Status.PodIP = ip
	return pi
}

func TestListResources(t *testing.T) {
	pl := &podsLister{
		cache: make(map[resourceKey]*podInfo),
	}
	for _, pi := range []*podInfo{
		testPodInfo("podA", "nsAB", "10.1.1.1", map[string]string{"app": "appA"}),
		testPodInfo("podB", "nsAB", "10.1.1.2", map[string]string{"app": "appB"}),
		testPodInfo("podC", "nsC", "10.1.1.3", map[string]string{"app": "appC", "func": "web"}),
		testPodInfo("podC", "devC", "10.2.1.3", map[string]string{"app": "appC", "func": "web"}),
	} {
		key := resourceKey{pi.Metadata.Namespace, pi.Metadata.Name}
		pl.keys = append(pl.keys, key)
		pl.cache[key] = pi
	}

	tests := []struct {
		desc         string
		nameFilter   string
		filters      map[string]string
		labelsFilter map[string]string
		wantPods     []resourceKey
		wantErr      bool
	}{
		{
			desc:    "bad filter key, expect error",
			filters: map[string]string{"names": "pod(B|C)"},
			wantErr: true,
		},
		{
			desc:     "only name filter for podB and podC",
			filters:  map[string]string{"name": "pod(B|C)"},
			wantPods: []resourceKey{{"nsAB", "podB"}, {"nsC", "podC"}, {"devC", "podC"}},
		},
		{
			desc:     "name filter for podB and podC, and namespace filter",
			filters:  map[string]string{"name": "pod(B|C)", "namespace": "ns.*"},
			wantPods: []resourceKey{{"nsAB", "podB"}, {"nsC", "podC"}},
		},
		{
			desc:     "name and namespace filter for podB",
			filters:  map[string]string{"name": "pod(B|C)", "namespace": "nsAB"},
			wantPods: []resourceKey{{"nsAB", "podB"}},
		},
		{
			desc:     "only namespace filter for podA and podB",
			filters:  map[string]string{"namespace": "nsAB"},
			wantPods: []resourceKey{{"nsAB", "podA"}, {"nsAB", "podB"}},
		},
	}

	for _, test := range tests {
		t.Run(test.desc, func(t *testing.T) {
			var filtersPB []*pb.Filter
			for k, v := range test.filters {
				filtersPB = append(filtersPB, &pb.Filter{Key: proto.String(k), Value: proto.String(v)})
			}

			results, err := pl.listResources(&pb.ListResourcesRequest{Filter: filtersPB})
			if err != nil {
				if !test.wantErr {
					t.Errorf("got unexpected error: %v", err)
				}
				return
			}

			var gotPods []resourceKey
			for _, res := range results {
				gotPods = append(gotPods, resourceKey{res.GetLabels()["namespace"], res.GetName()})
			}

			if !reflect.DeepEqual(gotPods, test.wantPods) {
				t.Errorf("pods.listResources: got=%v, expected=%v", gotPods, test.wantPods)
			}
		})
	}
}

func TestParseResourceList(t *testing.T) {
	podsListFile := "./testdata/pods.json"
	data, err := os.ReadFile(podsListFile)

	if err != nil {
		t.Fatalf("error reading test data file: %s", podsListFile)
	}

	keys, cache, err := parsePodsJSON(data)
	if err != nil {
		t.Fatalf("Error while parsing pods JSON data: %v", err)
	}
	pl := &podsLister{
		keys:  keys,
		cache: cache,
	}

	var tests = []struct {
		ns         string
		wantName   string
		wantIP     string
		wantLabels [][2]string
	}{
		{
			ns:       "prod",
			wantName: "cloudprober-54778d95f5-7hqtd",
			wantIP:   "10.28.0.3",
		},
		{
			ns:       "dev",
			wantName: "cloudprober-54778d95f5-7hqtd-dev",
			wantIP:   "10.22.0.3",
		},
		{
			ns:       "default",
			wantName: "test-running", // Only running pod shows up.
			wantIP:   "10.28.1.3",
		},
	}

	for _, test := range tests {
		t.Run(test.ns, func(t *testing.T) {
			filtersPB := []*pb.Filter{{Key: proto.String("namespace"), Value: proto.String(test.ns)}}
			results, _ := pl.listResources(&pb.ListResourcesRequest{Filter: filtersPB})

			if test.wantName == "" {
				assert.Len(t, results, 0)
				return
			}

			assert.Len(t, results, 1)
			result := results[0]
			assert.Equal(t, result.GetName(), test.wantName)
			assert.Equal(t, result.GetIp(), test.wantIP)

		})
	}
}