117 lines
3.6 KiB
Go
117 lines
3.6 KiB
Go
/*
|
|
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 cmd
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io"
|
|
"io/ioutil"
|
|
"time"
|
|
|
|
"testing"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
metricsapi "k8s.io/heapster/metrics/apis/metrics/v1alpha1"
|
|
"k8s.io/kubernetes/pkg/api"
|
|
"k8s.io/kubernetes/pkg/api/resource"
|
|
"k8s.io/kubernetes/pkg/api/unversioned"
|
|
v1 "k8s.io/kubernetes/pkg/api/v1"
|
|
cmdtesting "k8s.io/kubernetes/pkg/kubectl/cmd/testing"
|
|
)
|
|
|
|
const (
|
|
baseHeapsterServiceAddress = "/api/v1/proxy/namespaces/kube-system/services/http:heapster:"
|
|
baseMetricsAddress = baseHeapsterServiceAddress + "/apis/metrics"
|
|
metricsApiVersion = "v1alpha1"
|
|
)
|
|
|
|
func TestTopSubcommandsExist(t *testing.T) {
|
|
initTestErrorHandler(t)
|
|
|
|
f, _, _, _ := cmdtesting.NewAPIFactory()
|
|
buf := bytes.NewBuffer([]byte{})
|
|
|
|
cmd := NewCmdTop(f, buf, buf)
|
|
if !cmd.HasSubCommands() {
|
|
t.Error("top command should have subcommands")
|
|
}
|
|
}
|
|
|
|
func marshallBody(metrics interface{}) (io.ReadCloser, error) {
|
|
result, err := json.Marshal(metrics)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return ioutil.NopCloser(bytes.NewReader(result)), nil
|
|
}
|
|
|
|
func testNodeMetricsData() (*metricsapi.NodeMetricsList, *api.NodeList) {
|
|
metrics := &metricsapi.NodeMetricsList{
|
|
ListMeta: unversioned.ListMeta{
|
|
ResourceVersion: "1",
|
|
},
|
|
Items: []metricsapi.NodeMetrics{
|
|
{
|
|
ObjectMeta: v1.ObjectMeta{Name: "node1", ResourceVersion: "10"},
|
|
Window: unversioned.Duration{Duration: time.Minute},
|
|
Usage: v1.ResourceList{
|
|
v1.ResourceCPU: *resource.NewMilliQuantity(1, resource.DecimalSI),
|
|
v1.ResourceMemory: *resource.NewQuantity(2*(1024*1024), resource.DecimalSI),
|
|
v1.ResourceStorage: *resource.NewQuantity(3*(1024*1024), resource.DecimalSI),
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: v1.ObjectMeta{Name: "node2", ResourceVersion: "11"},
|
|
Window: unversioned.Duration{Duration: time.Minute},
|
|
Usage: v1.ResourceList{
|
|
v1.ResourceCPU: *resource.NewMilliQuantity(5, resource.DecimalSI),
|
|
v1.ResourceMemory: *resource.NewQuantity(6*(1024*1024), resource.DecimalSI),
|
|
v1.ResourceStorage: *resource.NewQuantity(7*(1024*1024), resource.DecimalSI),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
nodes := &api.NodeList{
|
|
ListMeta: metav1.ListMeta{
|
|
ResourceVersion: "15",
|
|
},
|
|
Items: []api.Node{
|
|
{
|
|
ObjectMeta: api.ObjectMeta{Name: "node1", ResourceVersion: "10"},
|
|
Status: api.NodeStatus{
|
|
Allocatable: api.ResourceList{
|
|
api.ResourceCPU: *resource.NewMilliQuantity(10, resource.DecimalSI),
|
|
api.ResourceMemory: *resource.NewQuantity(20*(1024*1024), resource.DecimalSI),
|
|
api.ResourceStorage: *resource.NewQuantity(30*(1024*1024), resource.DecimalSI),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: api.ObjectMeta{Name: "node2", ResourceVersion: "11"},
|
|
Status: api.NodeStatus{
|
|
Allocatable: api.ResourceList{
|
|
api.ResourceCPU: *resource.NewMilliQuantity(50, resource.DecimalSI),
|
|
api.ResourceMemory: *resource.NewQuantity(60*(1024*1024), resource.DecimalSI),
|
|
api.ResourceStorage: *resource.NewQuantity(70*(1024*1024), resource.DecimalSI),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
return metrics, nodes
|
|
}
|