mirror of https://github.com/k3s-io/k3s
Added CPU manager unit tests (none policy)
Balaji Subramaniam
Connor Doyle
parent
ff471913f9
commit
7567f1765f
|
|
@ -1,9 +1,10 @@
|
||||||
load("@io_bazel_rules_go//go:def.bzl", "go_library")
|
load("@io_bazel_rules_go//go:def.bzl", "go_library", "go_test")
|
||||||
|
|
||||||
go_library(
|
go_library(
|
||||||
name = "go_default_library",
|
name = "go_default_library",
|
||||||
srcs = [
|
srcs = [
|
||||||
"cpu_manager.go",
|
"cpu_manager.go",
|
||||||
|
"fake_cpu_manager.go",
|
||||||
"policy.go",
|
"policy.go",
|
||||||
"policy_none.go",
|
"policy_none.go",
|
||||||
],
|
],
|
||||||
|
|
@ -38,3 +39,22 @@ filegroup(
|
||||||
tags = ["automanaged"],
|
tags = ["automanaged"],
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
go_test(
|
||||||
|
name = "go_default_test",
|
||||||
|
srcs = [
|
||||||
|
"cpu_manager_test.go",
|
||||||
|
"policy_none_test.go",
|
||||||
|
],
|
||||||
|
library = ":go_default_library",
|
||||||
|
deps = [
|
||||||
|
"//pkg/kubelet/apis/cri/v1alpha1/runtime:go_default_library",
|
||||||
|
"//pkg/kubelet/cm/cpumanager/state:go_default_library",
|
||||||
|
"//pkg/kubelet/cm/cpuset:go_default_library",
|
||||||
|
"//vendor/k8s.io/api/core/v1:go_default_library",
|
||||||
|
"//vendor/k8s.io/apimachinery/pkg/api/resource:go_default_library",
|
||||||
|
"//vendor/k8s.io/apimachinery/pkg/apis/meta/v1:go_default_library",
|
||||||
|
"//vendor/k8s.io/apimachinery/pkg/runtime:go_default_library",
|
||||||
|
"//vendor/k8s.io/apimachinery/pkg/types:go_default_library",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,452 @@
|
||||||
|
/*
|
||||||
|
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 (
|
||||||
|
"fmt"
|
||||||
|
"reflect"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"k8s.io/api/core/v1"
|
||||||
|
"k8s.io/apimachinery/pkg/api/resource"
|
||||||
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||||
|
"k8s.io/apimachinery/pkg/runtime"
|
||||||
|
"k8s.io/apimachinery/pkg/types"
|
||||||
|
runtimeapi "k8s.io/kubernetes/pkg/kubelet/apis/cri/v1alpha1/runtime"
|
||||||
|
"k8s.io/kubernetes/pkg/kubelet/cm/cpumanager/state"
|
||||||
|
"k8s.io/kubernetes/pkg/kubelet/cm/cpuset"
|
||||||
|
)
|
||||||
|
|
||||||
|
type mockState struct {
|
||||||
|
assignments map[string]cpuset.CPUSet
|
||||||
|
defaultCPUSet cpuset.CPUSet
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *mockState) GetCPUSet(containerID string) (cpuset.CPUSet, bool) {
|
||||||
|
res, ok := s.assignments[containerID]
|
||||||
|
return res.Clone(), ok
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *mockState) GetDefaultCPUSet() cpuset.CPUSet {
|
||||||
|
return s.defaultCPUSet.Clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *mockState) GetCPUSetOrDefault(containerID string) cpuset.CPUSet {
|
||||||
|
if res, ok := s.GetCPUSet(containerID); ok {
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
return s.GetDefaultCPUSet()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *mockState) SetCPUSet(containerID string, cset cpuset.CPUSet) {
|
||||||
|
s.assignments[containerID] = cset
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *mockState) SetDefaultCPUSet(cset cpuset.CPUSet) {
|
||||||
|
s.defaultCPUSet = cset
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *mockState) Delete(containerID string) {
|
||||||
|
delete(s.assignments, containerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
type mockPolicy struct {
|
||||||
|
err error
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *mockPolicy) Name() string {
|
||||||
|
return "mock"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *mockPolicy) Start(s state.State) {
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *mockPolicy) AddContainer(s state.State, pod *v1.Pod, container *v1.Container, containerID string) error {
|
||||||
|
return p.err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *mockPolicy) RemoveContainer(s state.State, containerID string) error {
|
||||||
|
return p.err
|
||||||
|
}
|
||||||
|
|
||||||
|
type mockRuntimeService struct {
|
||||||
|
err error
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rt mockRuntimeService) UpdateContainerResources(id string, resources *runtimeapi.LinuxContainerResources) error {
|
||||||
|
return rt.err
|
||||||
|
}
|
||||||
|
|
||||||
|
type mockPodStatusProvider struct {
|
||||||
|
podStatus v1.PodStatus
|
||||||
|
found bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (psp mockPodStatusProvider) GetPodStatus(uid types.UID) (v1.PodStatus, bool) {
|
||||||
|
return psp.podStatus, psp.found
|
||||||
|
}
|
||||||
|
|
||||||
|
type mockPodKiller struct {
|
||||||
|
killedPods []*v1.Pod
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *mockPodKiller) killPodNow(pod *v1.Pod, status v1.PodStatus, gracePeriodOverride *int64) error {
|
||||||
|
f.killedPods = append(f.killedPods, pod)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type mockPodProvider struct {
|
||||||
|
pods []*v1.Pod
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *mockPodProvider) getPods() []*v1.Pod {
|
||||||
|
return f.pods
|
||||||
|
}
|
||||||
|
|
||||||
|
type mockRecorder struct{}
|
||||||
|
|
||||||
|
func (r *mockRecorder) Eventf(object runtime.Object, eventtype, reason, messageFmt string, args ...interface{}) {
|
||||||
|
}
|
||||||
|
|
||||||
|
func makePod(cpuRequest, cpuLimit string) *v1.Pod {
|
||||||
|
return &v1.Pod{
|
||||||
|
Spec: v1.PodSpec{
|
||||||
|
Containers: []v1.Container{
|
||||||
|
{
|
||||||
|
Resources: v1.ResourceRequirements{
|
||||||
|
Requests: v1.ResourceList{
|
||||||
|
v1.ResourceName(v1.ResourceCPU): resource.MustParse(cpuRequest),
|
||||||
|
v1.ResourceName(v1.ResourceMemory): resource.MustParse("1G"),
|
||||||
|
},
|
||||||
|
Limits: v1.ResourceList{
|
||||||
|
v1.ResourceName(v1.ResourceCPU): resource.MustParse(cpuLimit),
|
||||||
|
v1.ResourceName(v1.ResourceMemory): resource.MustParse("1G"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// CpuAllocatable must be <= CpuCapacity
|
||||||
|
func prepareCPUNodeStatus(CPUCapacity, CPUAllocatable string) v1.NodeStatus {
|
||||||
|
nodestatus := v1.NodeStatus{
|
||||||
|
Capacity: make(v1.ResourceList, 1),
|
||||||
|
Allocatable: make(v1.ResourceList, 1),
|
||||||
|
}
|
||||||
|
cpucap, _ := resource.ParseQuantity(CPUCapacity)
|
||||||
|
cpuall, _ := resource.ParseQuantity(CPUAllocatable)
|
||||||
|
|
||||||
|
nodestatus.Capacity[v1.ResourceCPU] = cpucap
|
||||||
|
nodestatus.Allocatable[v1.ResourceCPU] = cpuall
|
||||||
|
return nodestatus
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCPUManagerAdd(t *testing.T) {
|
||||||
|
testCases := []struct {
|
||||||
|
description string
|
||||||
|
regErr error
|
||||||
|
updateErr error
|
||||||
|
expErr error
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
description: "cpu manager add - no error",
|
||||||
|
regErr: nil,
|
||||||
|
updateErr: nil,
|
||||||
|
expErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "cpu manager add - policy add container error",
|
||||||
|
regErr: fmt.Errorf("fake reg error"),
|
||||||
|
updateErr: nil,
|
||||||
|
expErr: fmt.Errorf("fake reg error"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "cpu manager add - container update error",
|
||||||
|
regErr: nil,
|
||||||
|
updateErr: fmt.Errorf("fake update error"),
|
||||||
|
expErr: fmt.Errorf("fake update error"),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, testCase := range testCases {
|
||||||
|
mgr := &manager{
|
||||||
|
policy: &mockPolicy{
|
||||||
|
err: testCase.regErr,
|
||||||
|
},
|
||||||
|
state: &mockState{
|
||||||
|
assignments: map[string]cpuset.CPUSet{},
|
||||||
|
defaultCPUSet: cpuset.NewCPUSet(),
|
||||||
|
},
|
||||||
|
containerRuntime: mockRuntimeService{
|
||||||
|
err: testCase.updateErr,
|
||||||
|
},
|
||||||
|
activePods: func() []*v1.Pod { return nil },
|
||||||
|
podStatusProvider: mockPodStatusProvider{},
|
||||||
|
}
|
||||||
|
|
||||||
|
pod := makePod("1000", "1000")
|
||||||
|
container := &pod.Spec.Containers[0]
|
||||||
|
err := mgr.AddContainer(pod, container, "fakeID")
|
||||||
|
if !reflect.DeepEqual(err, testCase.expErr) {
|
||||||
|
t.Errorf("CPU Manager AddContainer() error (%v). expected error: %v but got: %v",
|
||||||
|
testCase.description, testCase.expErr, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCPUManagerRemove(t *testing.T) {
|
||||||
|
mgr := &manager{
|
||||||
|
policy: &mockPolicy{
|
||||||
|
err: nil,
|
||||||
|
},
|
||||||
|
state: &mockState{
|
||||||
|
assignments: map[string]cpuset.CPUSet{},
|
||||||
|
defaultCPUSet: cpuset.NewCPUSet(),
|
||||||
|
},
|
||||||
|
containerRuntime: mockRuntimeService{},
|
||||||
|
activePods: func() []*v1.Pod { return nil },
|
||||||
|
podStatusProvider: mockPodStatusProvider{},
|
||||||
|
}
|
||||||
|
|
||||||
|
err := mgr.RemoveContainer("fakeID")
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("CPU Manager RemoveContainer() error. expected error to be nil but got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
mgr = &manager{
|
||||||
|
policy: &mockPolicy{
|
||||||
|
err: fmt.Errorf("fake error"),
|
||||||
|
},
|
||||||
|
state: state.NewMemoryState(),
|
||||||
|
containerRuntime: mockRuntimeService{},
|
||||||
|
activePods: func() []*v1.Pod { return nil },
|
||||||
|
podStatusProvider: mockPodStatusProvider{},
|
||||||
|
}
|
||||||
|
|
||||||
|
err = mgr.RemoveContainer("fakeID")
|
||||||
|
if !reflect.DeepEqual(err, fmt.Errorf("fake error")) {
|
||||||
|
t.Errorf("CPU Manager RemoveContainer() error. expected error: fake error but got: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestReconcileState(t *testing.T) {
|
||||||
|
testCases := []struct {
|
||||||
|
description string
|
||||||
|
activePods []*v1.Pod
|
||||||
|
pspPS v1.PodStatus
|
||||||
|
pspFound bool
|
||||||
|
stAssignments map[string]cpuset.CPUSet
|
||||||
|
stDefaultCPUSet cpuset.CPUSet
|
||||||
|
updateErr error
|
||||||
|
expectFailedContainerName string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
description: "cpu manager reconclie - no error",
|
||||||
|
activePods: []*v1.Pod{
|
||||||
|
{
|
||||||
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
Name: "fakePodName",
|
||||||
|
UID: "fakeUID",
|
||||||
|
},
|
||||||
|
Spec: v1.PodSpec{
|
||||||
|
Containers: []v1.Container{
|
||||||
|
{
|
||||||
|
Name: "fakeName",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
pspPS: v1.PodStatus{
|
||||||
|
ContainerStatuses: []v1.ContainerStatus{
|
||||||
|
{
|
||||||
|
Name: "fakeName",
|
||||||
|
ContainerID: "docker://fakeID",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
pspFound: true,
|
||||||
|
stAssignments: map[string]cpuset.CPUSet{
|
||||||
|
"fakeID": cpuset.NewCPUSet(1, 2),
|
||||||
|
},
|
||||||
|
stDefaultCPUSet: cpuset.NewCPUSet(3, 4, 5, 6, 7),
|
||||||
|
updateErr: nil,
|
||||||
|
expectFailedContainerName: "",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "cpu manager reconclie - pod status not found",
|
||||||
|
activePods: []*v1.Pod{
|
||||||
|
{
|
||||||
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
Name: "fakePodName",
|
||||||
|
UID: "fakeUID",
|
||||||
|
},
|
||||||
|
Spec: v1.PodSpec{
|
||||||
|
Containers: []v1.Container{
|
||||||
|
{
|
||||||
|
Name: "fakeName",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
pspPS: v1.PodStatus{},
|
||||||
|
pspFound: false,
|
||||||
|
stAssignments: map[string]cpuset.CPUSet{},
|
||||||
|
stDefaultCPUSet: cpuset.NewCPUSet(),
|
||||||
|
updateErr: nil,
|
||||||
|
expectFailedContainerName: "fakeName",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "cpu manager reconclie - container id not found",
|
||||||
|
activePods: []*v1.Pod{
|
||||||
|
{
|
||||||
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
Name: "fakePodName",
|
||||||
|
UID: "fakeUID",
|
||||||
|
},
|
||||||
|
Spec: v1.PodSpec{
|
||||||
|
Containers: []v1.Container{
|
||||||
|
{
|
||||||
|
Name: "fakeName",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
pspPS: v1.PodStatus{
|
||||||
|
ContainerStatuses: []v1.ContainerStatus{
|
||||||
|
{
|
||||||
|
Name: "fakeName1",
|
||||||
|
ContainerID: "docker://fakeID",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
pspFound: true,
|
||||||
|
stAssignments: map[string]cpuset.CPUSet{},
|
||||||
|
stDefaultCPUSet: cpuset.NewCPUSet(),
|
||||||
|
updateErr: nil,
|
||||||
|
expectFailedContainerName: "fakeName",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "cpu manager reconclie - cpuset is empty",
|
||||||
|
activePods: []*v1.Pod{
|
||||||
|
{
|
||||||
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
Name: "fakePodName",
|
||||||
|
UID: "fakeUID",
|
||||||
|
},
|
||||||
|
Spec: v1.PodSpec{
|
||||||
|
Containers: []v1.Container{
|
||||||
|
{
|
||||||
|
Name: "fakeName",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
pspPS: v1.PodStatus{
|
||||||
|
ContainerStatuses: []v1.ContainerStatus{
|
||||||
|
{
|
||||||
|
Name: "fakeName",
|
||||||
|
ContainerID: "docker://fakeID",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
pspFound: true,
|
||||||
|
stAssignments: map[string]cpuset.CPUSet{
|
||||||
|
"fakeID": cpuset.NewCPUSet(),
|
||||||
|
},
|
||||||
|
stDefaultCPUSet: cpuset.NewCPUSet(1, 2, 3, 4, 5, 6, 7),
|
||||||
|
updateErr: nil,
|
||||||
|
expectFailedContainerName: "fakeName",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "cpu manager reconclie - container update error",
|
||||||
|
activePods: []*v1.Pod{
|
||||||
|
{
|
||||||
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
Name: "fakePodName",
|
||||||
|
UID: "fakeUID",
|
||||||
|
},
|
||||||
|
Spec: v1.PodSpec{
|
||||||
|
Containers: []v1.Container{
|
||||||
|
{
|
||||||
|
Name: "fakeName",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
pspPS: v1.PodStatus{
|
||||||
|
ContainerStatuses: []v1.ContainerStatus{
|
||||||
|
{
|
||||||
|
Name: "fakeName",
|
||||||
|
ContainerID: "docker://fakeID",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
pspFound: true,
|
||||||
|
stAssignments: map[string]cpuset.CPUSet{
|
||||||
|
"fakeID": cpuset.NewCPUSet(1, 2),
|
||||||
|
},
|
||||||
|
stDefaultCPUSet: cpuset.NewCPUSet(3, 4, 5, 6, 7),
|
||||||
|
updateErr: fmt.Errorf("fake container update error"),
|
||||||
|
expectFailedContainerName: "fakeName",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, testCase := range testCases {
|
||||||
|
mgr := &manager{
|
||||||
|
policy: &mockPolicy{
|
||||||
|
err: nil,
|
||||||
|
},
|
||||||
|
state: &mockState{
|
||||||
|
assignments: testCase.stAssignments,
|
||||||
|
defaultCPUSet: testCase.stDefaultCPUSet,
|
||||||
|
},
|
||||||
|
containerRuntime: mockRuntimeService{
|
||||||
|
err: testCase.updateErr,
|
||||||
|
},
|
||||||
|
activePods: func() []*v1.Pod {
|
||||||
|
return testCase.activePods
|
||||||
|
},
|
||||||
|
podStatusProvider: mockPodStatusProvider{
|
||||||
|
podStatus: testCase.pspPS,
|
||||||
|
found: testCase.pspFound,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
_, failure := mgr.reconcileState()
|
||||||
|
|
||||||
|
if testCase.expectFailedContainerName != "" {
|
||||||
|
// Search failed reconciled containers for the supplied name.
|
||||||
|
foundFailedContainer := false
|
||||||
|
for _, reconciled := range failure {
|
||||||
|
if reconciled.containerName == testCase.expectFailedContainerName {
|
||||||
|
foundFailedContainer = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !foundFailedContainer {
|
||||||
|
t.Errorf("Expected reconciliation failure for container: %s", testCase.expectFailedContainerName)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,58 @@
|
||||||
|
/*
|
||||||
|
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 (
|
||||||
|
"github.com/golang/glog"
|
||||||
|
"k8s.io/api/core/v1"
|
||||||
|
"k8s.io/kubernetes/pkg/kubelet/cm/cpumanager/state"
|
||||||
|
"k8s.io/kubernetes/pkg/kubelet/status"
|
||||||
|
)
|
||||||
|
|
||||||
|
type fakeManager struct {
|
||||||
|
state state.State
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *fakeManager) Start(activePods ActivePodsFunc, podStatusProvider status.PodStatusProvider, containerRuntime runtimeService) {
|
||||||
|
glog.Info("[fake cpumanager] Start()")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *fakeManager) Policy() Policy {
|
||||||
|
glog.Info("[fake cpumanager] Policy()")
|
||||||
|
return NewNonePolicy()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *fakeManager) AddContainer(pod *v1.Pod, container *v1.Container, containerID string) error {
|
||||||
|
glog.Infof("[fake cpumanager] AddContainer (pod: %s, container: %s, container id: %s)", pod.Name, container.Name, containerID)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *fakeManager) RemoveContainer(containerID string) error {
|
||||||
|
glog.Infof("[fake cpumanager] RemoveContainer (container id: %s)", containerID)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *fakeManager) State() state.Reader {
|
||||||
|
return m.state
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewFakeManager creates empty/fake cpu manager
|
||||||
|
func NewFakeManager() Manager {
|
||||||
|
return &fakeManager{
|
||||||
|
state: state.NewMemoryState(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,64 @@
|
||||||
|
/*
|
||||||
|
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/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 TestNonePolicyAdd(t *testing.T) {
|
||||||
|
policy := &nonePolicy{}
|
||||||
|
|
||||||
|
st := &mockState{
|
||||||
|
assignments: map[string]cpuset.CPUSet{},
|
||||||
|
defaultCPUSet: cpuset.NewCPUSet(1, 2, 3, 4, 5, 6, 7),
|
||||||
|
}
|
||||||
|
|
||||||
|
testPod := makePod("1000m", "1000m")
|
||||||
|
|
||||||
|
container := &testPod.Spec.Containers[0]
|
||||||
|
err := policy.AddContainer(st, testPod, container, "fakeID")
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("NonePolicy AddContainer() error. expected no error but got: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNonePolicyRemove(t *testing.T) {
|
||||||
|
policy := &nonePolicy{}
|
||||||
|
|
||||||
|
st := &mockState{
|
||||||
|
assignments: map[string]cpuset.CPUSet{},
|
||||||
|
defaultCPUSet: cpuset.NewCPUSet(1, 2, 3, 4, 5, 6, 7),
|
||||||
|
}
|
||||||
|
|
||||||
|
err := policy.RemoveContainer(st, "fakeID")
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("NonePolicy RemoveContainer() error. expected no error but got %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue