diff --git a/pkg/kubelet/apis/podresources/server_v1_test.go b/pkg/kubelet/apis/podresources/server_v1_test.go index 3913879b7f4..4cb2f909f68 100644 --- a/pkg/kubelet/apis/podresources/server_v1_test.go +++ b/pkg/kubelet/apis/podresources/server_v1_test.go @@ -181,6 +181,31 @@ func TestAllocatableResources(t *testing.T) { ResourceName: "resource-nt", DeviceIds: []string{"devA"}, }, + { + ResourceName: "resource-mm", + DeviceIds: []string{"devM0"}, + Topology: &podresourcesapi.TopologyInfo{ + Nodes: []*podresourcesapi.NUMANode{ + { + ID: 0, + }, + }, + }, + }, + { + ResourceName: "resource-mm", + DeviceIds: []string{"devMM"}, + Topology: &podresourcesapi.TopologyInfo{ + Nodes: []*podresourcesapi.NUMANode{ + { + ID: 0, + }, + { + ID: 1, + }, + }, + }, + }, } allCPUs := []int64{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16} @@ -238,6 +263,31 @@ func TestAllocatableResources(t *testing.T) { ResourceName: "resource-nt", DeviceIds: []string{"devA"}, }, + { + ResourceName: "resource-mm", + DeviceIds: []string{"devM0"}, + Topology: &podresourcesapi.TopologyInfo{ + Nodes: []*podresourcesapi.NUMANode{ + { + ID: 0, + }, + }, + }, + }, + { + ResourceName: "resource-mm", + DeviceIds: []string{"devMM"}, + Topology: &podresourcesapi.TopologyInfo{ + Nodes: []*podresourcesapi.NUMANode{ + { + ID: 0, + }, + { + ID: 1, + }, + }, + }, + }, }, }, }, @@ -273,6 +323,31 @@ func TestAllocatableResources(t *testing.T) { ResourceName: "resource-nt", DeviceIds: []string{"devA"}, }, + { + ResourceName: "resource-mm", + DeviceIds: []string{"devM0"}, + Topology: &podresourcesapi.TopologyInfo{ + Nodes: []*podresourcesapi.NUMANode{ + { + ID: 0, + }, + }, + }, + }, + { + ResourceName: "resource-mm", + DeviceIds: []string{"devMM"}, + Topology: &podresourcesapi.TopologyInfo{ + Nodes: []*podresourcesapi.NUMANode{ + { + ID: 0, + }, + { + ID: 1, + }, + }, + }, + }, }, }, },