Add requests to nodes (#2078)

* add requests and limits to nodes

* add requests and limits to nodes

pkg/models/resources/v1alpha3/node/nodes_test.go

@@ -0,0 +1,130 @@
+package node
+
+import (
+	"github.com/google/go-cmp/cmp"
+	corev1 "k8s.io/api/core/v1"
+	"k8s.io/apimachinery/pkg/api/resource"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/informers"
+	"k8s.io/client-go/kubernetes/fake"
+	"testing"
+)
+
+// mergeResourceLists will merge resoure lists. When two lists have the same resourece, the value from
+// the last list will be present in the result
+func mergeResourceLists(resourceLists ...corev1.ResourceList) corev1.ResourceList {
+	result := corev1.ResourceList{}
+	for _, rl := range resourceLists {
+		for resource, quantity := range rl {
+			result[resource] = quantity
+		}
+	}
+	return result
+}
+
+func getResourceList(cpu, memory string) corev1.ResourceList {
+	res := corev1.ResourceList{}
+	if cpu != "" {
+		res[corev1.ResourceCPU] = resource.MustParse(cpu)
+	}
+	if memory != "" {
+		res[corev1.ResourceMemory] = resource.MustParse(memory)
+	}
+	return res
+}
+
+var nodeAllocatable = mergeResourceLists(getResourceList("4", "12Gi"))
+var node = &corev1.Node{
+	ObjectMeta: metav1.ObjectMeta{
+		Name: "foo",
+	},
+	Status: corev1.NodeStatus{
+		Allocatable: nodeAllocatable,
+	},
+}
+
+var pods = []*corev1.Pod{
+	&corev1.Pod{
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: "foo",
+			Name:      "pod-with-resources",
+		},
+		TypeMeta: metav1.TypeMeta{
+			Kind: "Pod",
+		},
+		Spec: corev1.PodSpec{
+			NodeName: node.Name,
+			Containers: []corev1.Container{
+				{
+					Name:  "cpu-mem",
+					Image: "image:latest",
+					Resources: corev1.ResourceRequirements{
+						Requests: getResourceList("1", "1Gi"),
+						Limits:   getResourceList("2", "2Gi"),
+					},
+				},
+			},
+		},
+		Status: corev1.PodStatus{
+			Phase: corev1.PodRunning,
+		},
+	},
+	&corev1.Pod{
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: "foo2",
+			Name:      "pod-with-resources",
+		},
+		TypeMeta: metav1.TypeMeta{
+			Kind: "Pod",
+		},
+		Spec: corev1.PodSpec{
+			NodeName: node.Name,
+			Containers: []corev1.Container{
+				{
+					Name:  "cpu-mem",
+					Image: "image:latest",
+					Resources: corev1.ResourceRequirements{
+						Requests: getResourceList("1", "1Gi"),
+						Limits:   getResourceList("2", "2Gi"),
+					},
+				},
+			},
+		},
+		Status: corev1.PodStatus{
+			Phase: corev1.PodRunning,
+		},
+	},
+}
+
+var expectedAnnotations = map[string]string{
+	nodeCPURequests:            "2",
+	nodeCPULimits:              "4",
+	nodeCPURequestsFraction:    "50%",
+	nodeCPULimitsFraction:      "100%",
+	nodeMemoryRequests:         "2Gi",
+	nodeMemoryLimits:           "4Gi",
+	nodeMemoryRequestsFraction: "16%",
+	nodeMemoryLimitsFraction:   "33%",
+}
+
+func TestNodesGetterGet(t *testing.T) {
+	fake := fake.NewSimpleClientset(node, pods[0], pods[1])
+
+	informer := informers.NewSharedInformerFactory(fake, 0)
+	informer.Core().V1().Nodes().Informer().GetIndexer().Add(node)
+	for _, pod := range pods {
+		informer.Core().V1().Pods().Informer().GetIndexer().Add(pod)
+	}
+
+	nodeGetter := New(informer)
+	got, err := nodeGetter.Get("", node.Name)
+	if err != nil {
+		t.Fatal(err)
+	}
+	nodeGot := got.(*corev1.Node)
+
+	if diff := cmp.Diff(nodeGot.Annotations, expectedAnnotations); len(diff) != 0 {
+		t.Errorf("%T, diff(-got, +expected), %v", expectedAnnotations, nodeGot.Annotations)
+	}
+
+}