343 lines
7.5 KiB
Go
343 lines
7.5 KiB
Go
/*
|
|
Copyright 2015 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 k8s
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
meta_v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
testclient "k8s.io/client-go/kubernetes/fake"
|
|
api "k8s.io/client-go/pkg/api/v1"
|
|
)
|
|
|
|
func TestParseNameNS(t *testing.T) {
|
|
tests := []struct {
|
|
title string
|
|
input string
|
|
ns string
|
|
name string
|
|
expErr bool
|
|
}{
|
|
{"empty string", "", "", "", true},
|
|
{"demo", "demo", "", "", true},
|
|
{"kube-system", "kube-system", "", "", true},
|
|
{"default/kube-system", "default/kube-system", "default", "kube-system", false},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
ns, name, err := ParseNameNS(test.input)
|
|
if test.expErr {
|
|
if err == nil {
|
|
t.Errorf("%v: expected error but returned nil", test.title)
|
|
}
|
|
continue
|
|
}
|
|
if test.ns != ns {
|
|
t.Errorf("%v: expected %v but returned %v", test.title, test.ns, ns)
|
|
}
|
|
if test.name != name {
|
|
t.Errorf("%v: expected %v but returned %v", test.title, test.name, name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsValidService(t *testing.T) {
|
|
fk := testclient.NewSimpleClientset(&api.Service{
|
|
ObjectMeta: meta_v1.ObjectMeta{
|
|
Namespace: api.NamespaceDefault,
|
|
Name: "demo",
|
|
},
|
|
})
|
|
|
|
_, err := IsValidService(fk, "")
|
|
if err == nil {
|
|
t.Errorf("expected error but returned nil")
|
|
}
|
|
s, err := IsValidService(fk, "default/demo")
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if s == nil {
|
|
t.Errorf("expected a Service but returned nil")
|
|
}
|
|
|
|
fk = testclient.NewSimpleClientset()
|
|
s, err = IsValidService(fk, "default/demo")
|
|
if err == nil {
|
|
t.Errorf("expected an error but returned nil")
|
|
}
|
|
if s != nil {
|
|
t.Errorf("unexpected Service returned: %v", s)
|
|
}
|
|
}
|
|
|
|
func TestIsValidNamespace(t *testing.T) {
|
|
|
|
fk := testclient.NewSimpleClientset(&api.Namespace{
|
|
ObjectMeta: meta_v1.ObjectMeta{
|
|
Name: "default",
|
|
},
|
|
})
|
|
|
|
_, err := IsValidNamespace(fk, "empty")
|
|
if err == nil {
|
|
t.Errorf("expected error but return nill")
|
|
}
|
|
|
|
ns, err := IsValidNamespace(fk, "default")
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
|
|
if ns == nil {
|
|
t.Errorf("expected a configmap but returned nil")
|
|
}
|
|
|
|
}
|
|
|
|
func TestIsValidConfigMap(t *testing.T) {
|
|
|
|
fk := testclient.NewSimpleClientset(&api.ConfigMap{
|
|
ObjectMeta: meta_v1.ObjectMeta{
|
|
Namespace: api.NamespaceDefault,
|
|
Name: "demo",
|
|
},
|
|
})
|
|
|
|
_, err := IsValidConfigMap(fk, "")
|
|
if err == nil {
|
|
t.Errorf("expected error but return nill")
|
|
}
|
|
|
|
s, err := IsValidConfigMap(fk, "default/demo")
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
|
|
if s == nil {
|
|
t.Errorf("expected a configmap but returned nil")
|
|
}
|
|
|
|
fk = testclient.NewSimpleClientset()
|
|
s, err = IsValidConfigMap(fk, "default/demo")
|
|
if err == nil {
|
|
t.Errorf("expected an error but returned nil")
|
|
}
|
|
if s != nil {
|
|
t.Errorf("unexpected Configmap returned: %v", s)
|
|
}
|
|
|
|
}
|
|
|
|
func TestIsValidSecret(t *testing.T) {
|
|
fk := testclient.NewSimpleClientset(&api.Secret{
|
|
ObjectMeta: meta_v1.ObjectMeta{
|
|
Namespace: api.NamespaceDefault,
|
|
Name: "demo",
|
|
},
|
|
})
|
|
|
|
_, err := IsValidSecret(fk, "")
|
|
if err == nil {
|
|
t.Errorf("expected error but returned nil")
|
|
}
|
|
s, err := IsValidSecret(fk, "default/demo")
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if s == nil {
|
|
t.Errorf("expected a Secret but returned nil")
|
|
}
|
|
|
|
fk = testclient.NewSimpleClientset()
|
|
s, err = IsValidSecret(fk, "default/demo")
|
|
if err == nil {
|
|
t.Errorf("expected an error but returned nil")
|
|
}
|
|
if s != nil {
|
|
t.Errorf("unexpected Secret returned: %v", s)
|
|
}
|
|
}
|
|
|
|
func TestGetNodeIP(t *testing.T) {
|
|
fKNodes := []struct {
|
|
cs *testclient.Clientset
|
|
n string
|
|
ea string
|
|
}{
|
|
// empty node list
|
|
{testclient.NewSimpleClientset(), "demo", ""},
|
|
|
|
// node not exist
|
|
{testclient.NewSimpleClientset(&api.NodeList{Items: []api.Node{{
|
|
ObjectMeta: meta_v1.ObjectMeta{
|
|
Name: "demo",
|
|
},
|
|
Status: api.NodeStatus{
|
|
Addresses: []api.NodeAddress{
|
|
{
|
|
Type: api.NodeLegacyHostIP,
|
|
Address: "10.0.0.1",
|
|
},
|
|
},
|
|
},
|
|
}}}), "notexistnode", ""},
|
|
|
|
// node exist
|
|
{testclient.NewSimpleClientset(&api.NodeList{Items: []api.Node{{
|
|
ObjectMeta: meta_v1.ObjectMeta{
|
|
Name: "demo",
|
|
},
|
|
Status: api.NodeStatus{
|
|
Addresses: []api.NodeAddress{
|
|
{
|
|
Type: api.NodeLegacyHostIP,
|
|
Address: "10.0.0.1",
|
|
},
|
|
},
|
|
},
|
|
}}}), "demo", "10.0.0.1"},
|
|
|
|
// search the correct node
|
|
{testclient.NewSimpleClientset(&api.NodeList{Items: []api.Node{
|
|
{
|
|
ObjectMeta: meta_v1.ObjectMeta{
|
|
Name: "demo1",
|
|
},
|
|
Status: api.NodeStatus{
|
|
Addresses: []api.NodeAddress{
|
|
{
|
|
Type: api.NodeLegacyHostIP,
|
|
Address: "10.0.0.1",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: meta_v1.ObjectMeta{
|
|
Name: "demo2",
|
|
},
|
|
Status: api.NodeStatus{
|
|
Addresses: []api.NodeAddress{
|
|
{
|
|
Type: api.NodeLegacyHostIP,
|
|
Address: "10.0.0.2",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}}), "demo2", "10.0.0.2"},
|
|
|
|
// get NodeExternalIP
|
|
{testclient.NewSimpleClientset(&api.NodeList{Items: []api.Node{{
|
|
ObjectMeta: meta_v1.ObjectMeta{
|
|
Name: "demo",
|
|
},
|
|
Status: api.NodeStatus{
|
|
Addresses: []api.NodeAddress{
|
|
{
|
|
Type: api.NodeLegacyHostIP,
|
|
Address: "10.0.0.1",
|
|
}, {
|
|
Type: api.NodeExternalIP,
|
|
Address: "10.0.0.2",
|
|
},
|
|
},
|
|
},
|
|
}}}), "demo", "10.0.0.2"},
|
|
|
|
// get NodeLegacyHostIP
|
|
{testclient.NewSimpleClientset(&api.NodeList{Items: []api.Node{{
|
|
ObjectMeta: meta_v1.ObjectMeta{
|
|
Name: "demo",
|
|
},
|
|
Status: api.NodeStatus{
|
|
Addresses: []api.NodeAddress{
|
|
{
|
|
Type: api.NodeExternalIP,
|
|
Address: "",
|
|
}, {
|
|
Type: api.NodeLegacyHostIP,
|
|
Address: "10.0.0.2",
|
|
},
|
|
},
|
|
},
|
|
}}}), "demo", "10.0.0.2"},
|
|
}
|
|
|
|
for _, fk := range fKNodes {
|
|
address := GetNodeIP(fk.cs, fk.n)
|
|
if address != fk.ea {
|
|
t.Errorf("expected %s, but returned %s", fk.ea, address)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGetPodDetails(t *testing.T) {
|
|
// POD_NAME & POD_NAMESPACE not exist
|
|
os.Setenv("POD_NAME", "")
|
|
os.Setenv("POD_NAMESPACE", "")
|
|
_, err1 := GetPodDetails(testclient.NewSimpleClientset())
|
|
if err1 == nil {
|
|
t.Errorf("expected an error but returned nil")
|
|
}
|
|
|
|
// POD not exist
|
|
os.Setenv("POD_NAME", "testpod")
|
|
os.Setenv("POD_NAMESPACE", api.NamespaceDefault)
|
|
_, err2 := GetPodDetails(testclient.NewSimpleClientset())
|
|
if err2 == nil {
|
|
t.Errorf("expected an error but returned nil")
|
|
}
|
|
|
|
// success to get PodInfo
|
|
fkClient := testclient.NewSimpleClientset(
|
|
&api.PodList{Items: []api.Pod{{
|
|
ObjectMeta: meta_v1.ObjectMeta{
|
|
Name: "testpod",
|
|
Namespace: api.NamespaceDefault,
|
|
Labels: map[string]string{
|
|
"first": "first_label",
|
|
"second": "second_label",
|
|
},
|
|
},
|
|
}}},
|
|
&api.NodeList{Items: []api.Node{{
|
|
ObjectMeta: meta_v1.ObjectMeta{
|
|
Name: "demo",
|
|
},
|
|
Status: api.NodeStatus{
|
|
Addresses: []api.NodeAddress{
|
|
{
|
|
Type: api.NodeLegacyHostIP,
|
|
Address: "10.0.0.1",
|
|
},
|
|
},
|
|
},
|
|
}}})
|
|
|
|
epi, err3 := GetPodDetails(fkClient)
|
|
if err3 != nil {
|
|
t.Errorf("expected a PodInfo but returned error")
|
|
return
|
|
}
|
|
|
|
if epi == nil {
|
|
t.Errorf("expected a PodInfo but returned nil")
|
|
}
|
|
}
|