123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497 |
- /*
- Copyright 2016 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 top
- import (
- "bytes"
- "fmt"
- "io/ioutil"
- "net/http"
- "strings"
- "testing"
- "net/url"
- "k8s.io/api/core/v1"
- "k8s.io/apimachinery/pkg/runtime"
- "k8s.io/cli-runtime/pkg/genericclioptions"
- "k8s.io/client-go/rest/fake"
- core "k8s.io/client-go/testing"
- cmdtesting "k8s.io/kubernetes/pkg/kubectl/cmd/testing"
- "k8s.io/kubernetes/pkg/kubectl/scheme"
- metricsv1alpha1api "k8s.io/metrics/pkg/apis/metrics/v1alpha1"
- metricsv1beta1api "k8s.io/metrics/pkg/apis/metrics/v1beta1"
- metricsfake "k8s.io/metrics/pkg/client/clientset/versioned/fake"
- )
- const (
- apiPrefix = "api"
- apiVersion = "v1"
- )
- func TestTopNodeAllMetrics(t *testing.T) {
- cmdtesting.InitTestErrorHandler(t)
- metrics, nodes := testNodeV1alpha1MetricsData()
- expectedMetricsPath := fmt.Sprintf("%s/%s/nodes", baseMetricsAddress, metricsAPIVersion)
- expectedNodePath := fmt.Sprintf("/%s/%s/nodes", apiPrefix, apiVersion)
- tf := cmdtesting.NewTestFactory().WithNamespace("test")
- defer tf.Cleanup()
- codec := scheme.Codecs.LegacyCodec(scheme.Scheme.PrioritizedVersionsAllGroups()...)
- ns := scheme.Codecs
- tf.Client = &fake.RESTClient{
- NegotiatedSerializer: ns,
- Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
- switch p, m := req.URL.Path, req.Method; {
- case p == "/api":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apibody)))}, nil
- case p == "/apis":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apisbody)))}, nil
- case p == expectedMetricsPath && m == "GET":
- body, err := marshallBody(metrics)
- if err != nil {
- t.Errorf("unexpected error: %v", err)
- }
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: body}, nil
- case p == expectedNodePath && m == "GET":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: cmdtesting.ObjBody(codec, nodes)}, nil
- default:
- t.Fatalf("unexpected request: %#v\nGot URL: %#v\nExpected path: %#v", req, req.URL, expectedMetricsPath)
- return nil, nil
- }
- }),
- }
- tf.ClientConfigVal = cmdtesting.DefaultClientConfig()
- streams, _, buf, _ := genericclioptions.NewTestIOStreams()
- cmd := NewCmdTopNode(tf, nil, streams)
- cmd.Flags().Set("no-headers", "true")
- 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)
- }
- }
- if strings.Contains(result, "MEMORY") {
- t.Errorf("should not print headers with --no-headers option set:\n%s\n", result)
- }
- }
- func TestTopNodeAllMetricsCustomDefaults(t *testing.T) {
- customBaseHeapsterServiceAddress := "/api/v1/namespaces/custom-namespace/services/https:custom-heapster-service:/proxy"
- customBaseMetricsAddress := customBaseHeapsterServiceAddress + "/apis/metrics"
- cmdtesting.InitTestErrorHandler(t)
- metrics, nodes := testNodeV1alpha1MetricsData()
- expectedMetricsPath := fmt.Sprintf("%s/%s/nodes", customBaseMetricsAddress, metricsAPIVersion)
- expectedNodePath := fmt.Sprintf("/%s/%s/nodes", apiPrefix, apiVersion)
- tf := cmdtesting.NewTestFactory().WithNamespace("test")
- defer tf.Cleanup()
- codec := scheme.Codecs.LegacyCodec(scheme.Scheme.PrioritizedVersionsAllGroups()...)
- ns := scheme.Codecs
- tf.Client = &fake.RESTClient{
- NegotiatedSerializer: ns,
- Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
- switch p, m := req.URL.Path, req.Method; {
- case p == "/api":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apibody)))}, nil
- case p == "/apis":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apisbody)))}, nil
- case p == expectedMetricsPath && m == "GET":
- body, err := marshallBody(metrics)
- if err != nil {
- t.Errorf("unexpected error: %v", err)
- }
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: body}, nil
- case p == expectedNodePath && m == "GET":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: cmdtesting.ObjBody(codec, nodes)}, nil
- default:
- t.Fatalf("unexpected request: %#v\nGot URL: %#v\nExpected path: %#v", req, req.URL, expectedMetricsPath)
- return nil, nil
- }
- }),
- }
- tf.ClientConfigVal = cmdtesting.DefaultClientConfig()
- streams, _, buf, _ := genericclioptions.NewTestIOStreams()
- opts := &TopNodeOptions{
- HeapsterOptions: HeapsterTopOptions{
- Namespace: "custom-namespace",
- Scheme: "https",
- Service: "custom-heapster-service",
- },
- IOStreams: streams,
- }
- cmd := NewCmdTopNode(tf, opts, streams)
- 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) {
- cmdtesting.InitTestErrorHandler(t)
- metrics, nodes := testNodeV1alpha1MetricsData()
- expectedMetrics := metrics.Items[0]
- expectedNode := nodes.Items[0]
- nonExpectedMetrics := metricsv1alpha1api.NodeMetricsList{
- ListMeta: metrics.ListMeta,
- Items: metrics.Items[1:],
- }
- expectedPath := fmt.Sprintf("%s/%s/nodes/%s", baseMetricsAddress, metricsAPIVersion, expectedMetrics.Name)
- expectedNodePath := fmt.Sprintf("/%s/%s/nodes/%s", apiPrefix, apiVersion, expectedMetrics.Name)
- tf := cmdtesting.NewTestFactory().WithNamespace("test")
- defer tf.Cleanup()
- codec := scheme.Codecs.LegacyCodec(scheme.Scheme.PrioritizedVersionsAllGroups()...)
- ns := scheme.Codecs
- tf.Client = &fake.RESTClient{
- NegotiatedSerializer: ns,
- Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
- switch p, m := req.URL.Path, req.Method; {
- case p == "/api":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apibody)))}, nil
- case p == "/apis":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apisbody)))}, nil
- case p == expectedPath && m == "GET":
- body, err := marshallBody(expectedMetrics)
- if err != nil {
- t.Errorf("unexpected error: %v", err)
- }
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: body}, nil
- case p == expectedNodePath && m == "GET":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: cmdtesting.ObjBody(codec, &expectedNode)}, nil
- default:
- t.Fatalf("unexpected request: %#v\nGot URL: %#v\nExpected path: %#v", req, req.URL, expectedPath)
- return nil, nil
- }
- }),
- }
- tf.ClientConfigVal = cmdtesting.DefaultClientConfig()
- streams, _, buf, _ := genericclioptions.NewTestIOStreams()
- cmd := NewCmdTopNode(tf, nil, streams)
- cmd.Run(cmd, []string{expectedMetrics.Name})
- // Check the presence of node names in the output.
- result := buf.String()
- if !strings.Contains(result, expectedMetrics.Name) {
- t.Errorf("missing metrics for %s: \n%s", expectedMetrics.Name, result)
- }
- for _, m := range nonExpectedMetrics.Items {
- if strings.Contains(result, m.Name) {
- t.Errorf("unexpected metrics for %s: \n%s", m.Name, result)
- }
- }
- }
- func TestTopNodeWithLabelSelectorMetrics(t *testing.T) {
- cmdtesting.InitTestErrorHandler(t)
- metrics, nodes := testNodeV1alpha1MetricsData()
- expectedMetrics := metricsv1alpha1api.NodeMetricsList{
- ListMeta: metrics.ListMeta,
- Items: metrics.Items[0:1],
- }
- expectedNodes := v1.NodeList{
- ListMeta: nodes.ListMeta,
- Items: nodes.Items[0:1],
- }
- nonExpectedMetrics := metricsv1alpha1api.NodeMetricsList{
- ListMeta: metrics.ListMeta,
- Items: metrics.Items[1:],
- }
- label := "key=value"
- expectedPath := fmt.Sprintf("%s/%s/nodes", baseMetricsAddress, metricsAPIVersion)
- expectedQuery := fmt.Sprintf("labelSelector=%s", url.QueryEscape(label))
- expectedNodePath := fmt.Sprintf("/%s/%s/nodes", apiPrefix, apiVersion)
- tf := cmdtesting.NewTestFactory().WithNamespace("test")
- defer tf.Cleanup()
- codec := scheme.Codecs.LegacyCodec(scheme.Scheme.PrioritizedVersionsAllGroups()...)
- ns := scheme.Codecs
- tf.Client = &fake.RESTClient{
- 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 == "/api":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apibody)))}, nil
- case p == "/apis":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apisbody)))}, nil
- case p == expectedPath && m == "GET" && q == expectedQuery:
- body, err := marshallBody(expectedMetrics)
- if err != nil {
- t.Errorf("unexpected error: %v", err)
- }
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: body}, nil
- case p == expectedNodePath && m == "GET":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: cmdtesting.ObjBody(codec, &expectedNodes)}, nil
- default:
- t.Fatalf("unexpected request: %#v\nGot URL: %#v\nExpected path: %#v", req, req.URL, expectedPath)
- return nil, nil
- }
- }),
- }
- tf.ClientConfigVal = cmdtesting.DefaultClientConfig()
- streams, _, buf, _ := genericclioptions.NewTestIOStreams()
- cmd := NewCmdTopNode(tf, nil, streams)
- cmd.Flags().Set("selector", label)
- cmd.Run(cmd, []string{})
- // Check the presence of node names in the output.
- result := buf.String()
- for _, m := range expectedMetrics.Items {
- if !strings.Contains(result, m.Name) {
- t.Errorf("missing metrics for %s: \n%s", m.Name, result)
- }
- }
- for _, m := range nonExpectedMetrics.Items {
- if strings.Contains(result, m.Name) {
- t.Errorf("unexpected metrics for %s: \n%s", m.Name, result)
- }
- }
- }
- func TestTopNodeAllMetricsFromMetricsServer(t *testing.T) {
- cmdtesting.InitTestErrorHandler(t)
- expectedMetrics, nodes := testNodeV1beta1MetricsData()
- expectedNodePath := fmt.Sprintf("/%s/%s/nodes", apiPrefix, apiVersion)
- tf := cmdtesting.NewTestFactory().WithNamespace("test")
- defer tf.Cleanup()
- codec := scheme.Codecs.LegacyCodec(scheme.Scheme.PrioritizedVersionsAllGroups()...)
- ns := scheme.Codecs
- tf.Client = &fake.RESTClient{
- NegotiatedSerializer: ns,
- Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
- switch p, m := req.URL.Path, req.Method; {
- case p == "/api":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apibody)))}, nil
- case p == "/apis":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apisbodyWithMetrics)))}, nil
- case p == expectedNodePath && m == "GET":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: cmdtesting.ObjBody(codec, nodes)}, nil
- default:
- t.Fatalf("unexpected request: %#v\nGot URL: %#v\n", req, req.URL)
- return nil, nil
- }
- }),
- }
- fakemetricsClientset := &metricsfake.Clientset{}
- fakemetricsClientset.AddReactor("list", "nodes", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- return true, expectedMetrics, nil
- })
- tf.ClientConfigVal = cmdtesting.DefaultClientConfig()
- streams, _, buf, _ := genericclioptions.NewTestIOStreams()
- cmd := NewCmdTopNode(tf, nil, streams)
- // TODO in the long run, we want to test most of our commands like this. Wire the options struct with specific mocks
- // TODO then check the particular Run functionality and harvest results from fake clients
- cmdOptions := &TopNodeOptions{
- IOStreams: streams,
- }
- if err := cmdOptions.Complete(tf, cmd, []string{}); err != nil {
- t.Fatal(err)
- }
- cmdOptions.MetricsClient = fakemetricsClientset
- if err := cmdOptions.Validate(); err != nil {
- t.Fatal(err)
- }
- if err := cmdOptions.RunTopNode(); err != nil {
- t.Fatal(err)
- }
- // Check the presence of node names in the output.
- result := buf.String()
- for _, m := range expectedMetrics.Items {
- if !strings.Contains(result, m.Name) {
- t.Errorf("missing metrics for %s: \n%s", m.Name, result)
- }
- }
- }
- func TestTopNodeWithNameMetricsFromMetricsServer(t *testing.T) {
- cmdtesting.InitTestErrorHandler(t)
- metrics, nodes := testNodeV1beta1MetricsData()
- expectedMetrics := metrics.Items[0]
- expectedNode := nodes.Items[0]
- nonExpectedMetrics := metricsv1beta1api.NodeMetricsList{
- ListMeta: metrics.ListMeta,
- Items: metrics.Items[1:],
- }
- expectedNodePath := fmt.Sprintf("/%s/%s/nodes/%s", apiPrefix, apiVersion, expectedMetrics.Name)
- tf := cmdtesting.NewTestFactory().WithNamespace("test")
- defer tf.Cleanup()
- codec := scheme.Codecs.LegacyCodec(scheme.Scheme.PrioritizedVersionsAllGroups()...)
- ns := scheme.Codecs
- tf.Client = &fake.RESTClient{
- NegotiatedSerializer: ns,
- Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
- switch p, m := req.URL.Path, req.Method; {
- case p == "/api":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apibody)))}, nil
- case p == "/apis":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apisbodyWithMetrics)))}, nil
- case p == expectedNodePath && m == "GET":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: cmdtesting.ObjBody(codec, &expectedNode)}, nil
- default:
- t.Fatalf("unexpected request: %#v\nGot URL: %#v\n", req, req.URL)
- return nil, nil
- }
- }),
- }
- fakemetricsClientset := &metricsfake.Clientset{}
- fakemetricsClientset.AddReactor("get", "nodes", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- return true, &expectedMetrics, nil
- })
- tf.ClientConfigVal = cmdtesting.DefaultClientConfig()
- streams, _, buf, _ := genericclioptions.NewTestIOStreams()
- cmd := NewCmdTopNode(tf, nil, streams)
- // TODO in the long run, we want to test most of our commands like this. Wire the options struct with specific mocks
- // TODO then check the particular Run functionality and harvest results from fake clients
- cmdOptions := &TopNodeOptions{
- IOStreams: streams,
- }
- if err := cmdOptions.Complete(tf, cmd, []string{expectedMetrics.Name}); err != nil {
- t.Fatal(err)
- }
- cmdOptions.MetricsClient = fakemetricsClientset
- if err := cmdOptions.Validate(); err != nil {
- t.Fatal(err)
- }
- if err := cmdOptions.RunTopNode(); err != nil {
- t.Fatal(err)
- }
- // Check the presence of node names in the output.
- result := buf.String()
- if !strings.Contains(result, expectedMetrics.Name) {
- t.Errorf("missing metrics for %s: \n%s", expectedMetrics.Name, result)
- }
- for _, m := range nonExpectedMetrics.Items {
- if strings.Contains(result, m.Name) {
- t.Errorf("unexpected metrics for %s: \n%s", m.Name, result)
- }
- }
- }
- func TestTopNodeWithLabelSelectorMetricsFromMetricsServer(t *testing.T) {
- cmdtesting.InitTestErrorHandler(t)
- metrics, nodes := testNodeV1beta1MetricsData()
- expectedMetrics := &metricsv1beta1api.NodeMetricsList{
- ListMeta: metrics.ListMeta,
- Items: metrics.Items[0:1],
- }
- expectedNodes := v1.NodeList{
- ListMeta: nodes.ListMeta,
- Items: nodes.Items[0:1],
- }
- nonExpectedMetrics := &metricsv1beta1api.NodeMetricsList{
- ListMeta: metrics.ListMeta,
- Items: metrics.Items[1:],
- }
- label := "key=value"
- expectedNodePath := fmt.Sprintf("/%s/%s/nodes", apiPrefix, apiVersion)
- tf := cmdtesting.NewTestFactory().WithNamespace("test")
- defer tf.Cleanup()
- codec := scheme.Codecs.LegacyCodec(scheme.Scheme.PrioritizedVersionsAllGroups()...)
- ns := scheme.Codecs
- tf.Client = &fake.RESTClient{
- NegotiatedSerializer: ns,
- Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
- switch p, m, _ := req.URL.Path, req.Method, req.URL.RawQuery; {
- case p == "/api":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apibody)))}, nil
- case p == "/apis":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: ioutil.NopCloser(bytes.NewReader([]byte(apisbodyWithMetrics)))}, nil
- case p == expectedNodePath && m == "GET":
- return &http.Response{StatusCode: 200, Header: cmdtesting.DefaultHeader(), Body: cmdtesting.ObjBody(codec, &expectedNodes)}, nil
- default:
- t.Fatalf("unexpected request: %#v\nGot URL: %#v\n", req, req.URL)
- return nil, nil
- }
- }),
- }
- fakemetricsClientset := &metricsfake.Clientset{}
- fakemetricsClientset.AddReactor("list", "nodes", func(action core.Action) (handled bool, ret runtime.Object, err error) {
- return true, expectedMetrics, nil
- })
- tf.ClientConfigVal = cmdtesting.DefaultClientConfig()
- streams, _, buf, _ := genericclioptions.NewTestIOStreams()
- cmd := NewCmdTopNode(tf, nil, streams)
- cmd.Flags().Set("selector", label)
- // TODO in the long run, we want to test most of our commands like this. Wire the options struct with specific mocks
- // TODO then check the particular Run functionality and harvest results from fake clients
- cmdOptions := &TopNodeOptions{
- IOStreams: streams,
- }
- if err := cmdOptions.Complete(tf, cmd, []string{}); err != nil {
- t.Fatal(err)
- }
- cmdOptions.MetricsClient = fakemetricsClientset
- if err := cmdOptions.Validate(); err != nil {
- t.Fatal(err)
- }
- if err := cmdOptions.RunTopNode(); err != nil {
- t.Fatal(err)
- }
- // Check the presence of node names in the output.
- result := buf.String()
- for _, m := range expectedMetrics.Items {
- if !strings.Contains(result, m.Name) {
- t.Errorf("missing metrics for %s: \n%s", m.Name, result)
- }
- }
- for _, m := range nonExpectedMetrics.Items {
- if strings.Contains(result, m.Name) {
- t.Errorf("unexpected metrics for %s: \n%s", m.Name, result)
- }
- }
- }
|