mirror of https://github.com/k3s-io/k3s
add tests
parent
5b8b1eb2eb
commit
c969079f49
|
@ -79,6 +79,59 @@ func TestTopNodeAllMetrics(t *testing.T) {
|
|||
}
|
||||
}
|
||||
|
||||
func TestTopNodeAllMetricsCustomDefaults(t *testing.T) {
|
||||
customBaseHeapsterServiceAddress := "/api/v1/namespaces/custom-namespace/services/https:custom-heapster-service:/proxy"
|
||||
customBaseMetricsAddress := customBaseHeapsterServiceAddress + "/apis/metrics"
|
||||
|
||||
initTestErrorHandler(t)
|
||||
metrics, nodes := testNodeMetricsData()
|
||||
expectedMetricsPath := fmt.Sprintf("%s/%s/nodes", customBaseMetricsAddress, metricsApiVersion)
|
||||
expectedNodePath := fmt.Sprintf("/%s/%s/nodes", apiPrefix, apiVersion)
|
||||
|
||||
f, tf, codec, ns := cmdtesting.NewAPIFactory()
|
||||
tf.Printer = &testPrinter{}
|
||||
tf.Client = &fake.RESTClient{
|
||||
APIRegistry: api.Registry,
|
||||
NegotiatedSerializer: ns,
|
||||
Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
|
||||
switch p, m := req.URL.Path, req.Method; {
|
||||
case p == expectedMetricsPath && m == "GET":
|
||||
body, err := marshallBody(metrics)
|
||||
if err != nil {
|
||||
t.Errorf("unexpected error: %v", err)
|
||||
}
|
||||
return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: body}, nil
|
||||
case p == expectedNodePath && m == "GET":
|
||||
return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, nodes)}, nil
|
||||
default:
|
||||
t.Fatalf("unexpected request: %#v\nGot URL: %#v\nExpected path: %#v", req, req.URL, expectedMetricsPath)
|
||||
return nil, nil
|
||||
}
|
||||
}),
|
||||
}
|
||||
tf.Namespace = "test"
|
||||
tf.ClientConfig = defaultClientConfig()
|
||||
buf := bytes.NewBuffer([]byte{})
|
||||
|
||||
opts := &TopNodeOptions{
|
||||
HeapsterOptions: HeapsterTopOptions{
|
||||
Namespace: "custom-namespace",
|
||||
Scheme: "https",
|
||||
Service: "custom-heapster-service",
|
||||
},
|
||||
}
|
||||
cmd := NewCmdTopNode(f, opts, buf)
|
||||
cmd.Run(cmd, []string{})
|
||||
|
||||
// Check the presence of node names in the output.
|
||||
result := buf.String()
|
||||
for _, m := range metrics.Items {
|
||||
if !strings.Contains(result, m.Name) {
|
||||
t.Errorf("missing metrics for %s: \n%s", m.Name, result)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestTopNodeWithNameMetrics(t *testing.T) {
|
||||
initTestErrorHandler(t)
|
||||
metrics, nodes := testNodeMetricsData()
|
||||
|
|
|
@ -170,6 +170,149 @@ func TestTopPod(t *testing.T) {
|
|||
}
|
||||
}
|
||||
|
||||
func TestTopPodCustomDefaults(t *testing.T) {
|
||||
customBaseHeapsterServiceAddress := "/api/v1/namespaces/custom-namespace/services/https:custom-heapster-service:/proxy"
|
||||
customBaseMetricsAddress := customBaseHeapsterServiceAddress + "/apis/metrics"
|
||||
customTopPathPrefix := customBaseMetricsAddress + "/" + metricsApiVersion
|
||||
|
||||
testNS := "custom-namespace"
|
||||
testCases := []struct {
|
||||
name string
|
||||
namespace string
|
||||
flags map[string]string
|
||||
args []string
|
||||
expectedPath string
|
||||
expectedQuery string
|
||||
namespaces []string
|
||||
containers bool
|
||||
listsNamespaces bool
|
||||
}{
|
||||
{
|
||||
name: "all namespaces",
|
||||
flags: map[string]string{"all-namespaces": "true"},
|
||||
expectedPath: customTopPathPrefix + "/pods",
|
||||
namespaces: []string{testNS, "secondtestns", "thirdtestns"},
|
||||
listsNamespaces: true,
|
||||
},
|
||||
{
|
||||
name: "all in namespace",
|
||||
expectedPath: customTopPathPrefix + "/namespaces/" + testNS + "/pods",
|
||||
namespaces: []string{testNS, testNS},
|
||||
},
|
||||
{
|
||||
name: "pod with name",
|
||||
args: []string{"pod1"},
|
||||
expectedPath: customTopPathPrefix + "/namespaces/" + testNS + "/pods/pod1",
|
||||
namespaces: []string{testNS},
|
||||
},
|
||||
{
|
||||
name: "pod with label selector",
|
||||
flags: map[string]string{"selector": "key=value"},
|
||||
expectedPath: customTopPathPrefix + "/namespaces/" + testNS + "/pods",
|
||||
expectedQuery: "labelSelector=" + url.QueryEscape("key=value"),
|
||||
namespaces: []string{testNS, testNS},
|
||||
},
|
||||
{
|
||||
name: "pod with container metrics",
|
||||
flags: map[string]string{"containers": "true"},
|
||||
args: []string{"pod1"},
|
||||
expectedPath: customTopPathPrefix + "/namespaces/" + testNS + "/pods/pod1",
|
||||
namespaces: []string{testNS},
|
||||
containers: true,
|
||||
},
|
||||
}
|
||||
initTestErrorHandler(t)
|
||||
for _, testCase := range testCases {
|
||||
t.Logf("Running test case: %s", testCase.name)
|
||||
metricsList := testPodMetricsData()
|
||||
var expectedMetrics []metricsapi.PodMetrics
|
||||
var expectedContainerNames, nonExpectedMetricsNames []string
|
||||
for n, m := range metricsList {
|
||||
if n < len(testCase.namespaces) {
|
||||
m.Namespace = testCase.namespaces[n]
|
||||
expectedMetrics = append(expectedMetrics, m)
|
||||
for _, c := range m.Containers {
|
||||
expectedContainerNames = append(expectedContainerNames, c.Name)
|
||||
}
|
||||
} else {
|
||||
nonExpectedMetricsNames = append(nonExpectedMetricsNames, m.Name)
|
||||
}
|
||||
}
|
||||
|
||||
var response interface{}
|
||||
if len(expectedMetrics) == 1 {
|
||||
response = expectedMetrics[0]
|
||||
} else {
|
||||
response = metricsapi.PodMetricsList{
|
||||
ListMeta: metav1.ListMeta{
|
||||
ResourceVersion: "2",
|
||||
},
|
||||
Items: expectedMetrics,
|
||||
}
|
||||
}
|
||||
|
||||
f, tf, _, ns := cmdtesting.NewAPIFactory()
|
||||
tf.Printer = &testPrinter{}
|
||||
tf.Client = &fake.RESTClient{
|
||||
APIRegistry: api.Registry,
|
||||
NegotiatedSerializer: ns,
|
||||
Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
|
||||
switch p, m, q := req.URL.Path, req.Method, req.URL.RawQuery; {
|
||||
case p == testCase.expectedPath && m == "GET" && (testCase.expectedQuery == "" || q == testCase.expectedQuery):
|
||||
body, err := marshallBody(response)
|
||||
if err != nil {
|
||||
t.Errorf("%s: unexpected error: %v", testCase.name, err)
|
||||
}
|
||||
return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: body}, nil
|
||||
default:
|
||||
t.Fatalf("%s: unexpected request: %#v\nGot URL: %#v\nExpected path: %#v\nExpected query: %#v",
|
||||
testCase.name, req, req.URL, testCase.expectedPath, testCase.expectedQuery)
|
||||
return nil, nil
|
||||
}
|
||||
}),
|
||||
}
|
||||
tf.Namespace = testNS
|
||||
tf.ClientConfig = defaultClientConfig()
|
||||
buf := bytes.NewBuffer([]byte{})
|
||||
|
||||
opts := &TopPodOptions{
|
||||
HeapsterOptions: HeapsterTopOptions{
|
||||
Namespace: "custom-namespace",
|
||||
Scheme: "https",
|
||||
Service: "custom-heapster-service",
|
||||
},
|
||||
}
|
||||
cmd := NewCmdTopPod(f, opts, buf)
|
||||
for name, value := range testCase.flags {
|
||||
cmd.Flags().Set(name, value)
|
||||
}
|
||||
cmd.Run(cmd, testCase.args)
|
||||
|
||||
// Check the presence of pod names&namespaces/container names in the output.
|
||||
result := buf.String()
|
||||
if testCase.containers {
|
||||
for _, containerName := range expectedContainerNames {
|
||||
if !strings.Contains(result, containerName) {
|
||||
t.Errorf("%s: missing metrics for container %s: \n%s", testCase.name, containerName, result)
|
||||
}
|
||||
}
|
||||
}
|
||||
for _, m := range expectedMetrics {
|
||||
if !strings.Contains(result, m.Name) {
|
||||
t.Errorf("%s: missing metrics for %s: \n%s", testCase.name, m.Name, result)
|
||||
}
|
||||
if testCase.listsNamespaces && !strings.Contains(result, m.Namespace) {
|
||||
t.Errorf("%s: missing metrics for %s/%s: \n%s", testCase.name, m.Namespace, m.Name, result)
|
||||
}
|
||||
}
|
||||
for _, name := range nonExpectedMetricsNames {
|
||||
if strings.Contains(result, name) {
|
||||
t.Errorf("%s: unexpected metrics for %s: \n%s", testCase.name, name, result)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func testPodMetricsData() []metricsapi.PodMetrics {
|
||||
return []metricsapi.PodMetrics{
|
||||
{
|
||||
|
|
Loading…
Reference in New Issue