github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-11 21:12:07 +00:00
Code Issues Projects Releases Wiki Activity

memorymanager:unit-tests for isAffinityViolatingNUMAAllocations function

Browse Source 
Signed-off-by: Talor Itzhak <titzhak@redhat.com>
This commit is contained in:
Talor Itzhak 2024-07-11 17:05:15 +03:00
parent 37399c8d4f
commit c019114e41
1 changed files with 76 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
pkg/kubelet/cm/memorymanager/policy_static_test.go
View File

@ -3896,3 +3896,79 @@ func Test_getPodRequestedResources(t *testing.T) {
})
}
}
func Test_isAffinityViolatingNUMAAllocations(t *testing.T) {
testsCases := []struct {
description string
machineState map[int]*state.NUMANodeState
topologyHint *topologymanager.TopologyHint
isViolationExpected bool
}{
{
description: "violating NUMA allocations because given affinity asks for NUMA ID 1 which is on different cells group",
machineState: map[int]*state.NUMANodeState{
0: {
NumberOfAssignments: 1,
Cells: []int{0, 1},
},
1: {
NumberOfAssignments: 1,
Cells: []int{0, 1},
},
2: {
NumberOfAssignments: 1,
Cells: []int{2},
},
3: {
NumberOfAssignments: 0,
Cells: []int{3},
},
},
topologyHint: &topologymanager.TopologyHint{
NUMANodeAffinity: newNUMAAffinity(1, 2),
},
isViolationExpected: true,
},
{
description: "violating NUMA allocations because given affinity with multiple nodes asks for NUMA ID 1 which is used for a single NUMA node memory allocation",
machineState: map[int]*state.NUMANodeState{
0: {
NumberOfAssignments: 0,
Cells: []int{0, 1},
},
1: {
NumberOfAssignments: 1,
Cells: []int{1},
},
},
topologyHint: &topologymanager.TopologyHint{
NUMANodeAffinity: newNUMAAffinity(0, 1),
},
isViolationExpected: true,
},
{
description: "valid affinity, no prior assignments",
machineState: map[int]*state.NUMANodeState{
0: {
NumberOfAssignments: 0,
Cells: []int{0},
},
1: {
NumberOfAssignments: 0,
Cells: []int{1},
},
},
topologyHint: &topologymanager.TopologyHint{
NUMANodeAffinity: newNUMAAffinity(0, 1),
},
isViolationExpected: false,
},
}
for _, tc := range testsCases {
t.Run(tc.description, func(t *testing.T) {
if isAffinityViolatingNUMAAllocations(tc.machineState, tc.topologyHint.NUMANodeAffinity) != tc.isViolationExpected {
t.Errorf("isAffinityViolatingNUMAAllocations with affinity %v expected to return %t, got %t", tc.topologyHint.NUMANodeAffinity.GetBits(), tc.isViolationExpected, !tc.isViolationExpected)
}
})
}
}