mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-25 04:33:26 +00:00
Add tests for GetPodTopologyHints() for cpumanager
* Add tests for getPodRequestedCPU() Signed-off-by: Krzysztof Wiatrzyk <k.wiatrzyk@samsung.com>
This commit is contained in:
sw.han
Krzysztof Wiatrzyk
parent
35b1f28d0f
commit
7ad65bf22d
@ -62,6 +62,95 @@ func returnMachineInfo() cadvisorapi.MachineInfo {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestPodGuaranteedCPUs(t *testing.T) {
|
||||||
|
CPUs := [][]struct {
|
||||||
|
request string
|
||||||
|
limit string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
{request: "0", limit: "0"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
{request: "2", limit: "2"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
{request: "5", limit: "5"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
{request: "2", limit: "2"},
|
||||||
|
{request: "4", limit: "4"},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
// tc for not guaranteed Pod
|
||||||
|
testPod1 := makeMultiContainerPod(CPUs[0], CPUs[0])
|
||||||
|
testPod2 := makeMultiContainerPod(CPUs[0], CPUs[1])
|
||||||
|
testPod3 := makeMultiContainerPod(CPUs[1], CPUs[0])
|
||||||
|
// tc for guaranteed Pod
|
||||||
|
testPod4 := makeMultiContainerPod(CPUs[1], CPUs[1])
|
||||||
|
testPod5 := makeMultiContainerPod(CPUs[2], CPUs[2])
|
||||||
|
// tc for comparing init containers and user containers
|
||||||
|
testPod6 := makeMultiContainerPod(CPUs[1], CPUs[2])
|
||||||
|
testPod7 := makeMultiContainerPod(CPUs[2], CPUs[1])
|
||||||
|
// tc for multi containers
|
||||||
|
testPod8 := makeMultiContainerPod(CPUs[3], CPUs[3])
|
||||||
|
|
||||||
|
p := staticPolicy{}
|
||||||
|
|
||||||
|
tcases := []struct {
|
||||||
|
name string
|
||||||
|
pod *v1.Pod
|
||||||
|
expectedCPU int
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "TestCase01: if requestedCPU == 0, Pod is not Guaranteed Qos",
|
||||||
|
pod: testPod1,
|
||||||
|
expectedCPU: 0,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "TestCase02: if requestedCPU == 0, Pod is not Guaranteed Qos",
|
||||||
|
pod: testPod2,
|
||||||
|
expectedCPU: 0,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "TestCase03: if requestedCPU == 0, Pod is not Guaranteed Qos",
|
||||||
|
pod: testPod3,
|
||||||
|
expectedCPU: 0,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "TestCase04: Guaranteed Pod requests 2 CPUs",
|
||||||
|
pod: testPod4,
|
||||||
|
expectedCPU: 2,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "TestCase05: Guaranteed Pod requests 5 CPUs",
|
||||||
|
pod: testPod5,
|
||||||
|
expectedCPU: 5,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "TestCase06: The number of CPUs requested By app is bigger than the number of CPUs requested by init",
|
||||||
|
pod: testPod6,
|
||||||
|
expectedCPU: 5,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "TestCase07: The number of CPUs requested By init is bigger than the number of CPUs requested by app",
|
||||||
|
pod: testPod7,
|
||||||
|
expectedCPU: 5,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "TestCase08: Sum of CPUs requested by multiple containers",
|
||||||
|
pod: testPod8,
|
||||||
|
expectedCPU: 6,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tc := range tcases {
|
||||||
|
requestedCPU := p.podGuaranteedCPUs(tc.pod)
|
||||||
|
|
||||||
|
if requestedCPU != tc.expectedCPU {
|
||||||
|
t.Errorf("Expected in result to be %v , got %v", tc.expectedCPU, requestedCPU)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestGetTopologyHints(t *testing.T) {
|
func TestGetTopologyHints(t *testing.T) {
|
||||||
machineInfo := returnMachineInfo()
|
machineInfo := returnMachineInfo()
|
||||||
tcases := returnTestCases()
|
tcases := returnTestCases()
|
||||||
@ -111,6 +200,54 @@ func TestGetTopologyHints(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestGetPodTopologyHints(t *testing.T) {
|
||||||
|
machineInfo := returnMachineInfo()
|
||||||
|
|
||||||
|
for _, tc := range returnTestCases() {
|
||||||
|
topology, _ := topology.Discover(&machineInfo)
|
||||||
|
|
||||||
|
var activePods []*v1.Pod
|
||||||
|
for p := range tc.assignments {
|
||||||
|
pod := v1.Pod{}
|
||||||
|
pod.UID = types.UID(p)
|
||||||
|
for c := range tc.assignments[p] {
|
||||||
|
container := v1.Container{}
|
||||||
|
container.Name = c
|
||||||
|
pod.Spec.Containers = append(pod.Spec.Containers, container)
|
||||||
|
}
|
||||||
|
activePods = append(activePods, &pod)
|
||||||
|
}
|
||||||
|
|
||||||
|
m := manager{
|
||||||
|
policy: &staticPolicy{
|
||||||
|
topology: topology,
|
||||||
|
},
|
||||||
|
state: &mockState{
|
||||||
|
assignments: tc.assignments,
|
||||||
|
defaultCPUSet: tc.defaultCPUSet,
|
||||||
|
},
|
||||||
|
topology: topology,
|
||||||
|
activePods: func() []*v1.Pod { return activePods },
|
||||||
|
podStatusProvider: mockPodStatusProvider{},
|
||||||
|
sourcesReady: &sourcesReadyStub{},
|
||||||
|
}
|
||||||
|
|
||||||
|
podHints := m.GetPodTopologyHints(&tc.pod)[string(v1.ResourceCPU)]
|
||||||
|
if len(tc.expectedHints) == 0 && len(podHints) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
sort.SliceStable(podHints, func(i, j int) bool {
|
||||||
|
return podHints[i].LessThan(podHints[j])
|
||||||
|
})
|
||||||
|
sort.SliceStable(tc.expectedHints, func(i, j int) bool {
|
||||||
|
return tc.expectedHints[i].LessThan(tc.expectedHints[j])
|
||||||
|
})
|
||||||
|
if !reflect.DeepEqual(tc.expectedHints, podHints) {
|
||||||
|
t.Errorf("Expected in result to be %v , got %v", tc.expectedHints, podHints)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func returnTestCases() []testCase {
|
func returnTestCases() []testCase {
|
||||||
testPod1 := makePod("fakePod", "fakeContainer", "2", "2")
|
testPod1 := makePod("fakePod", "fakeContainer", "2", "2")
|
||||||
testContainer1 := &testPod1.Spec.Containers[0]
|
testContainer1 := &testPod1.Spec.Containers[0]
|
||||||
|
|||||||
@ -52,324 +52,7 @@ func makeSocketMask(sockets ...int) bitmask.BitMask {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestGetTopologyHints(t *testing.T) {
|
func TestGetTopologyHints(t *testing.T) {
|
||||||
tcases := []struct {
|
tcases := getCommonTestCases()
|
||||||
description string
|
|
||||||
podUID string
|
|
||||||
containerName string
|
|
||||||
request map[string]string
|
|
||||||
devices map[string][]pluginapi.Device
|
|
||||||
allocatedDevices map[string]map[string]map[string][]string
|
|
||||||
expectedHints map[string][]topologymanager.TopologyHint
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
description: "Single Request, no alignment",
|
|
||||||
podUID: "fakePod",
|
|
||||||
containerName: "fakeContainer",
|
|
||||||
request: map[string]string{
|
|
||||||
"testdevice": "1",
|
|
||||||
},
|
|
||||||
devices: map[string][]pluginapi.Device{
|
|
||||||
"testdevice": {
|
|
||||||
{ID: "Dev1"},
|
|
||||||
{ID: "Dev2"},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedHints: map[string][]topologymanager.TopologyHint{
|
|
||||||
"testdevice": nil,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
description: "Single Request, only one with alignment",
|
|
||||||
podUID: "fakePod",
|
|
||||||
containerName: "fakeContainer",
|
|
||||||
request: map[string]string{
|
|
||||||
"testdevice": "1",
|
|
||||||
},
|
|
||||||
devices: map[string][]pluginapi.Device{
|
|
||||||
"testdevice": {
|
|
||||||
{ID: "Dev1"},
|
|
||||||
makeNUMADevice("Dev2", 1),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedHints: map[string][]topologymanager.TopologyHint{
|
|
||||||
"testdevice": {
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(1),
|
|
||||||
Preferred: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0, 1),
|
|
||||||
Preferred: false,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
description: "Single Request, one device per socket",
|
|
||||||
podUID: "fakePod",
|
|
||||||
containerName: "fakeContainer",
|
|
||||||
request: map[string]string{
|
|
||||||
"testdevice": "1",
|
|
||||||
},
|
|
||||||
devices: map[string][]pluginapi.Device{
|
|
||||||
"testdevice": {
|
|
||||||
makeNUMADevice("Dev1", 0),
|
|
||||||
makeNUMADevice("Dev2", 1),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedHints: map[string][]topologymanager.TopologyHint{
|
|
||||||
"testdevice": {
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0),
|
|
||||||
Preferred: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(1),
|
|
||||||
Preferred: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0, 1),
|
|
||||||
Preferred: false,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
description: "Request for 2, one device per socket",
|
|
||||||
podUID: "fakePod",
|
|
||||||
containerName: "fakeContainer",
|
|
||||||
request: map[string]string{
|
|
||||||
"testdevice": "2",
|
|
||||||
},
|
|
||||||
devices: map[string][]pluginapi.Device{
|
|
||||||
"testdevice": {
|
|
||||||
makeNUMADevice("Dev1", 0),
|
|
||||||
makeNUMADevice("Dev2", 1),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedHints: map[string][]topologymanager.TopologyHint{
|
|
||||||
"testdevice": {
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0, 1),
|
|
||||||
Preferred: true,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
description: "Request for 2, 2 devices per socket",
|
|
||||||
podUID: "fakePod",
|
|
||||||
containerName: "fakeContainer",
|
|
||||||
request: map[string]string{
|
|
||||||
"testdevice": "2",
|
|
||||||
},
|
|
||||||
devices: map[string][]pluginapi.Device{
|
|
||||||
"testdevice": {
|
|
||||||
makeNUMADevice("Dev1", 0),
|
|
||||||
makeNUMADevice("Dev2", 1),
|
|
||||||
makeNUMADevice("Dev3", 0),
|
|
||||||
makeNUMADevice("Dev4", 1),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedHints: map[string][]topologymanager.TopologyHint{
|
|
||||||
"testdevice": {
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0),
|
|
||||||
Preferred: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(1),
|
|
||||||
Preferred: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0, 1),
|
|
||||||
Preferred: false,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
description: "Request for 2, optimal on 1 NUMA node, forced cross-NUMA",
|
|
||||||
podUID: "fakePod",
|
|
||||||
containerName: "fakeContainer",
|
|
||||||
request: map[string]string{
|
|
||||||
"testdevice": "2",
|
|
||||||
},
|
|
||||||
devices: map[string][]pluginapi.Device{
|
|
||||||
"testdevice": {
|
|
||||||
makeNUMADevice("Dev1", 0),
|
|
||||||
makeNUMADevice("Dev2", 1),
|
|
||||||
makeNUMADevice("Dev3", 0),
|
|
||||||
makeNUMADevice("Dev4", 1),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
allocatedDevices: map[string]map[string]map[string][]string{
|
|
||||||
"fakePod": {
|
|
||||||
"fakeOtherContainer": {
|
|
||||||
"testdevice": {"Dev1", "Dev2"},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedHints: map[string][]topologymanager.TopologyHint{
|
|
||||||
"testdevice": {
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0, 1),
|
|
||||||
Preferred: false,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
description: "2 device types, mixed configuration",
|
|
||||||
podUID: "fakePod",
|
|
||||||
containerName: "fakeContainer",
|
|
||||||
request: map[string]string{
|
|
||||||
"testdevice1": "2",
|
|
||||||
"testdevice2": "1",
|
|
||||||
},
|
|
||||||
devices: map[string][]pluginapi.Device{
|
|
||||||
"testdevice1": {
|
|
||||||
makeNUMADevice("Dev1", 0),
|
|
||||||
makeNUMADevice("Dev2", 1),
|
|
||||||
makeNUMADevice("Dev3", 0),
|
|
||||||
makeNUMADevice("Dev4", 1),
|
|
||||||
},
|
|
||||||
"testdevice2": {
|
|
||||||
makeNUMADevice("Dev1", 0),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedHints: map[string][]topologymanager.TopologyHint{
|
|
||||||
"testdevice1": {
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0),
|
|
||||||
Preferred: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(1),
|
|
||||||
Preferred: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0, 1),
|
|
||||||
Preferred: false,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
"testdevice2": {
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0),
|
|
||||||
Preferred: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0, 1),
|
|
||||||
Preferred: false,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
description: "Single device type, more requested than available",
|
|
||||||
podUID: "fakePod",
|
|
||||||
containerName: "fakeContainer",
|
|
||||||
request: map[string]string{
|
|
||||||
"testdevice": "6",
|
|
||||||
},
|
|
||||||
devices: map[string][]pluginapi.Device{
|
|
||||||
"testdevice": {
|
|
||||||
makeNUMADevice("Dev1", 0),
|
|
||||||
makeNUMADevice("Dev2", 0),
|
|
||||||
makeNUMADevice("Dev3", 1),
|
|
||||||
makeNUMADevice("Dev4", 1),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedHints: map[string][]topologymanager.TopologyHint{
|
|
||||||
"testdevice": {},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
description: "Single device type, all already allocated to container",
|
|
||||||
podUID: "fakePod",
|
|
||||||
containerName: "fakeContainer",
|
|
||||||
request: map[string]string{
|
|
||||||
"testdevice": "2",
|
|
||||||
},
|
|
||||||
devices: map[string][]pluginapi.Device{
|
|
||||||
"testdevice": {
|
|
||||||
makeNUMADevice("Dev1", 0),
|
|
||||||
makeNUMADevice("Dev2", 0),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
allocatedDevices: map[string]map[string]map[string][]string{
|
|
||||||
"fakePod": {
|
|
||||||
"fakeContainer": {
|
|
||||||
"testdevice": {"Dev1", "Dev2"},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedHints: map[string][]topologymanager.TopologyHint{
|
|
||||||
"testdevice": {
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0),
|
|
||||||
Preferred: true,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
NUMANodeAffinity: makeSocketMask(0, 1),
|
|
||||||
Preferred: false,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
description: "Single device type, less already allocated to container than requested",
|
|
||||||
podUID: "fakePod",
|
|
||||||
containerName: "fakeContainer",
|
|
||||||
request: map[string]string{
|
|
||||||
"testdevice": "4",
|
|
||||||
},
|
|
||||||
devices: map[string][]pluginapi.Device{
|
|
||||||
"testdevice": {
|
|
||||||
makeNUMADevice("Dev1", 0),
|
|
||||||
makeNUMADevice("Dev2", 0),
|
|
||||||
makeNUMADevice("Dev3", 1),
|
|
||||||
makeNUMADevice("Dev4", 1),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
allocatedDevices: map[string]map[string]map[string][]string{
|
|
||||||
"fakePod": {
|
|
||||||
"fakeContainer": {
|
|
||||||
"testdevice": {"Dev1", "Dev2"},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedHints: map[string][]topologymanager.TopologyHint{
|
|
||||||
"testdevice": {},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
description: "Single device type, more already allocated to container than requested",
|
|
||||||
podUID: "fakePod",
|
|
||||||
containerName: "fakeContainer",
|
|
||||||
request: map[string]string{
|
|
||||||
"testdevice": "2",
|
|
||||||
},
|
|
||||||
devices: map[string][]pluginapi.Device{
|
|
||||||
"testdevice": {
|
|
||||||
makeNUMADevice("Dev1", 0),
|
|
||||||
makeNUMADevice("Dev2", 0),
|
|
||||||
makeNUMADevice("Dev3", 1),
|
|
||||||
makeNUMADevice("Dev4", 1),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
allocatedDevices: map[string]map[string]map[string][]string{
|
|
||||||
"fakePod": {
|
|
||||||
"fakeContainer": {
|
|
||||||
"testdevice": {"Dev1", "Dev2", "Dev3", "Dev4"},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedHints: map[string][]topologymanager.TopologyHint{
|
|
||||||
"testdevice": {},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tc := range tcases {
|
for _, tc := range tcases {
|
||||||
resourceList := v1.ResourceList{}
|
resourceList := v1.ResourceList{}
|
||||||
@ -980,3 +663,326 @@ func TestGetPreferredAllocationParameters(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type topologyHintTestCase struct {
|
||||||
|
description string
|
||||||
|
podUID string
|
||||||
|
containerName string
|
||||||
|
request map[string]string
|
||||||
|
devices map[string][]pluginapi.Device
|
||||||
|
allocatedDevices map[string]map[string]map[string][]string
|
||||||
|
expectedHints map[string][]topologymanager.TopologyHint
|
||||||
|
}
|
||||||
|
|
||||||
|
func getCommonTestCases() []topologyHintTestCase {
|
||||||
|
return []topologyHintTestCase{
|
||||||
|
{
|
||||||
|
description: "Single Request, no alignment",
|
||||||
|
podUID: "fakePod",
|
||||||
|
containerName: "fakeContainer",
|
||||||
|
request: map[string]string{
|
||||||
|
"testdevice": "1",
|
||||||
|
},
|
||||||
|
devices: map[string][]pluginapi.Device{
|
||||||
|
"testdevice": {
|
||||||
|
{ID: "Dev1"},
|
||||||
|
{ID: "Dev2"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedHints: map[string][]topologymanager.TopologyHint{
|
||||||
|
"testdevice": nil,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "Single Request, only one with alignment",
|
||||||
|
podUID: "fakePod",
|
||||||
|
containerName: "fakeContainer",
|
||||||
|
request: map[string]string{
|
||||||
|
"testdevice": "1",
|
||||||
|
},
|
||||||
|
devices: map[string][]pluginapi.Device{
|
||||||
|
"testdevice": {
|
||||||
|
{ID: "Dev1"},
|
||||||
|
makeNUMADevice("Dev2", 1),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedHints: map[string][]topologymanager.TopologyHint{
|
||||||
|
"testdevice": {
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(1),
|
||||||
|
Preferred: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0, 1),
|
||||||
|
Preferred: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "Single Request, one device per socket",
|
||||||
|
podUID: "fakePod",
|
||||||
|
containerName: "fakeContainer",
|
||||||
|
request: map[string]string{
|
||||||
|
"testdevice": "1",
|
||||||
|
},
|
||||||
|
devices: map[string][]pluginapi.Device{
|
||||||
|
"testdevice": {
|
||||||
|
makeNUMADevice("Dev1", 0),
|
||||||
|
makeNUMADevice("Dev2", 1),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedHints: map[string][]topologymanager.TopologyHint{
|
||||||
|
"testdevice": {
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0),
|
||||||
|
Preferred: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(1),
|
||||||
|
Preferred: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0, 1),
|
||||||
|
Preferred: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "Request for 2, one device per socket",
|
||||||
|
podUID: "fakePod",
|
||||||
|
containerName: "fakeContainer",
|
||||||
|
request: map[string]string{
|
||||||
|
"testdevice": "2",
|
||||||
|
},
|
||||||
|
devices: map[string][]pluginapi.Device{
|
||||||
|
"testdevice": {
|
||||||
|
makeNUMADevice("Dev1", 0),
|
||||||
|
makeNUMADevice("Dev2", 1),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedHints: map[string][]topologymanager.TopologyHint{
|
||||||
|
"testdevice": {
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0, 1),
|
||||||
|
Preferred: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "Request for 2, 2 devices per socket",
|
||||||
|
podUID: "fakePod",
|
||||||
|
containerName: "fakeContainer",
|
||||||
|
request: map[string]string{
|
||||||
|
"testdevice": "2",
|
||||||
|
},
|
||||||
|
devices: map[string][]pluginapi.Device{
|
||||||
|
"testdevice": {
|
||||||
|
makeNUMADevice("Dev1", 0),
|
||||||
|
makeNUMADevice("Dev2", 1),
|
||||||
|
makeNUMADevice("Dev3", 0),
|
||||||
|
makeNUMADevice("Dev4", 1),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedHints: map[string][]topologymanager.TopologyHint{
|
||||||
|
"testdevice": {
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0),
|
||||||
|
Preferred: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(1),
|
||||||
|
Preferred: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0, 1),
|
||||||
|
Preferred: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "Request for 2, optimal on 1 NUMA node, forced cross-NUMA",
|
||||||
|
podUID: "fakePod",
|
||||||
|
containerName: "fakeContainer",
|
||||||
|
request: map[string]string{
|
||||||
|
"testdevice": "2",
|
||||||
|
},
|
||||||
|
devices: map[string][]pluginapi.Device{
|
||||||
|
"testdevice": {
|
||||||
|
makeNUMADevice("Dev1", 0),
|
||||||
|
makeNUMADevice("Dev2", 1),
|
||||||
|
makeNUMADevice("Dev3", 0),
|
||||||
|
makeNUMADevice("Dev4", 1),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
allocatedDevices: map[string]map[string]map[string][]string{
|
||||||
|
"fakePod": {
|
||||||
|
"fakeOtherContainer": {
|
||||||
|
"testdevice": {"Dev1", "Dev2"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedHints: map[string][]topologymanager.TopologyHint{
|
||||||
|
"testdevice": {
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0, 1),
|
||||||
|
Preferred: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "2 device types, mixed configuration",
|
||||||
|
podUID: "fakePod",
|
||||||
|
containerName: "fakeContainer",
|
||||||
|
request: map[string]string{
|
||||||
|
"testdevice1": "2",
|
||||||
|
"testdevice2": "1",
|
||||||
|
},
|
||||||
|
devices: map[string][]pluginapi.Device{
|
||||||
|
"testdevice1": {
|
||||||
|
makeNUMADevice("Dev1", 0),
|
||||||
|
makeNUMADevice("Dev2", 1),
|
||||||
|
makeNUMADevice("Dev3", 0),
|
||||||
|
makeNUMADevice("Dev4", 1),
|
||||||
|
},
|
||||||
|
"testdevice2": {
|
||||||
|
makeNUMADevice("Dev1", 0),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedHints: map[string][]topologymanager.TopologyHint{
|
||||||
|
"testdevice1": {
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0),
|
||||||
|
Preferred: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(1),
|
||||||
|
Preferred: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0, 1),
|
||||||
|
Preferred: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"testdevice2": {
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0),
|
||||||
|
Preferred: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0, 1),
|
||||||
|
Preferred: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "Single device type, more requested than available",
|
||||||
|
podUID: "fakePod",
|
||||||
|
containerName: "fakeContainer",
|
||||||
|
request: map[string]string{
|
||||||
|
"testdevice": "6",
|
||||||
|
},
|
||||||
|
devices: map[string][]pluginapi.Device{
|
||||||
|
"testdevice": {
|
||||||
|
makeNUMADevice("Dev1", 0),
|
||||||
|
makeNUMADevice("Dev2", 0),
|
||||||
|
makeNUMADevice("Dev3", 1),
|
||||||
|
makeNUMADevice("Dev4", 1),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedHints: map[string][]topologymanager.TopologyHint{
|
||||||
|
"testdevice": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "Single device type, all already allocated to container",
|
||||||
|
podUID: "fakePod",
|
||||||
|
containerName: "fakeContainer",
|
||||||
|
request: map[string]string{
|
||||||
|
"testdevice": "2",
|
||||||
|
},
|
||||||
|
devices: map[string][]pluginapi.Device{
|
||||||
|
"testdevice": {
|
||||||
|
makeNUMADevice("Dev1", 0),
|
||||||
|
makeNUMADevice("Dev2", 0),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
allocatedDevices: map[string]map[string]map[string][]string{
|
||||||
|
"fakePod": {
|
||||||
|
"fakeContainer": {
|
||||||
|
"testdevice": {"Dev1", "Dev2"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedHints: map[string][]topologymanager.TopologyHint{
|
||||||
|
"testdevice": {
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0),
|
||||||
|
Preferred: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
NUMANodeAffinity: makeSocketMask(0, 1),
|
||||||
|
Preferred: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "Single device type, less already allocated to container than requested",
|
||||||
|
podUID: "fakePod",
|
||||||
|
containerName: "fakeContainer",
|
||||||
|
request: map[string]string{
|
||||||
|
"testdevice": "4",
|
||||||
|
},
|
||||||
|
devices: map[string][]pluginapi.Device{
|
||||||
|
"testdevice": {
|
||||||
|
makeNUMADevice("Dev1", 0),
|
||||||
|
makeNUMADevice("Dev2", 0),
|
||||||
|
makeNUMADevice("Dev3", 1),
|
||||||
|
makeNUMADevice("Dev4", 1),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
allocatedDevices: map[string]map[string]map[string][]string{
|
||||||
|
"fakePod": {
|
||||||
|
"fakeContainer": {
|
||||||
|
"testdevice": {"Dev1", "Dev2"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedHints: map[string][]topologymanager.TopologyHint{
|
||||||
|
"testdevice": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
description: "Single device type, more already allocated to container than requested",
|
||||||
|
podUID: "fakePod",
|
||||||
|
containerName: "fakeContainer",
|
||||||
|
request: map[string]string{
|
||||||
|
"testdevice": "2",
|
||||||
|
},
|
||||||
|
devices: map[string][]pluginapi.Device{
|
||||||
|
"testdevice": {
|
||||||
|
makeNUMADevice("Dev1", 0),
|
||||||
|
makeNUMADevice("Dev2", 0),
|
||||||
|
makeNUMADevice("Dev3", 1),
|
||||||
|
makeNUMADevice("Dev4", 1),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
allocatedDevices: map[string]map[string]map[string][]string{
|
||||||
|
"fakePod": {
|
||||||
|
"fakeContainer": {
|
||||||
|
"testdevice": {"Dev1", "Dev2", "Dev3", "Dev4"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedHints: map[string][]topologymanager.TopologyHint{
|
||||||
|
"testdevice": {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user