@@ -23,6 +23,7 @@ import (
23
23
24
24
cadvisorapi "github.com/google/cadvisor/info/v1"
25
25
v1 "k8s.io/api/core/v1"
26
+ "k8s.io/apimachinery/pkg/types"
26
27
"k8s.io/kubernetes/pkg/kubelet/cm/cpumanager/state"
27
28
"k8s.io/kubernetes/pkg/kubelet/cm/cpumanager/topology"
28
29
"k8s.io/kubernetes/pkg/kubelet/cm/cpuset"
@@ -238,6 +239,18 @@ func TestGetTopologyHints(t *testing.T) {
238
239
for _ , tc := range tcases {
239
240
topology , _ := topology .Discover (& machineInfo , numaNodeInfo )
240
241
242
+ var activePods []* v1.Pod
243
+ for p := range tc .assignments {
244
+ pod := v1.Pod {}
245
+ pod .UID = types .UID (p )
246
+ for c := range tc .assignments [p ] {
247
+ container := v1.Container {}
248
+ container .Name = c
249
+ pod .Spec .Containers = append (pod .Spec .Containers , container )
250
+ }
251
+ activePods = append (activePods , & pod )
252
+ }
253
+
241
254
m := manager {
242
255
policy : & staticPolicy {
243
256
topology : topology ,
@@ -247,7 +260,7 @@ func TestGetTopologyHints(t *testing.T) {
247
260
defaultCPUSet : tc .defaultCPUSet ,
248
261
},
249
262
topology : topology ,
250
- activePods : func () []* v1.Pod { return nil },
263
+ activePods : func () []* v1.Pod { return activePods },
251
264
podStatusProvider : mockPodStatusProvider {},
252
265
sourcesReady : & sourcesReadyStub {},
253
266
}
0 commit comments