6d33354e4c
Extend the podresources API implementing the GetAllocatableResources endpoint, as specified in the KEPs: https://github.com/kubernetes/enhancements/tree/master/keps/sig-node/2043-pod-resource-concrete-assigments https://github.com/kubernetes/enhancements/pull/2404 Signed-off-by: Francesco Romani <fromani@redhat.com>
89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
/*
|
|
Copyright 2017 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 cpumanager
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"k8s.io/kubernetes/pkg/kubelet/cm/cpumanager/state"
|
|
"k8s.io/kubernetes/pkg/kubelet/cm/cpuset"
|
|
)
|
|
|
|
func TestNonePolicyName(t *testing.T) {
|
|
policy := &nonePolicy{}
|
|
|
|
policyName := policy.Name()
|
|
if policyName != "none" {
|
|
t.Errorf("NonePolicy Name() error. expected: none, returned: %v", policyName)
|
|
}
|
|
}
|
|
|
|
func TestNonePolicyAllocate(t *testing.T) {
|
|
policy := &nonePolicy{}
|
|
|
|
st := &mockState{
|
|
assignments: state.ContainerCPUAssignments{},
|
|
defaultCPUSet: cpuset.NewCPUSet(1, 2, 3, 4, 5, 6, 7),
|
|
}
|
|
|
|
testPod := makePod("fakePod", "fakeContainer", "1000m", "1000m")
|
|
|
|
container := &testPod.Spec.Containers[0]
|
|
err := policy.Allocate(st, testPod, container)
|
|
if err != nil {
|
|
t.Errorf("NonePolicy Allocate() error. expected no error but got: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestNonePolicyRemove(t *testing.T) {
|
|
policy := &nonePolicy{}
|
|
|
|
st := &mockState{
|
|
assignments: state.ContainerCPUAssignments{},
|
|
defaultCPUSet: cpuset.NewCPUSet(1, 2, 3, 4, 5, 6, 7),
|
|
}
|
|
|
|
testPod := makePod("fakePod", "fakeContainer", "1000m", "1000m")
|
|
|
|
container := &testPod.Spec.Containers[0]
|
|
err := policy.RemoveContainer(st, string(testPod.UID), container.Name)
|
|
if err != nil {
|
|
t.Errorf("NonePolicy RemoveContainer() error. expected no error but got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestNonePolicyGetAllocatableCPUs(t *testing.T) {
|
|
// any random topology is fine
|
|
|
|
var cpuIDs []int
|
|
for cpuID := range topoSingleSocketHT.CPUDetails {
|
|
cpuIDs = append(cpuIDs, cpuID)
|
|
}
|
|
|
|
policy := &nonePolicy{}
|
|
|
|
st := &mockState{
|
|
assignments: state.ContainerCPUAssignments{},
|
|
defaultCPUSet: cpuset.NewCPUSet(cpuIDs...),
|
|
}
|
|
|
|
cpus := policy.GetAllocatableCPUs(st)
|
|
if cpus.Size() != 0 {
|
|
t.Errorf("NonePolicy GetAllocatableCPUs() error. expected empty set, returned: %v", cpus)
|
|
}
|
|
}
|