in pkg/util/testing/wrappers.go [218:246]
func MakeAdmission(cq string, podSetNames ...string) *AdmissionWrapper {
wrap := &AdmissionWrapper{kueue.Admission{
ClusterQueue: kueue.ClusterQueueReference(cq),
}}
if len(podSetNames) == 0 {
wrap.PodSetAssignments = []kueue.PodSetAssignment{
{
Name: kueue.DefaultPodSetName,
Flavors: make(map[corev1.ResourceName]kueue.ResourceFlavorReference),
ResourceUsage: make(corev1.ResourceList),
Count: 1,
},
}
return wrap
}
var psFlavors []kueue.PodSetAssignment
for _, name := range podSetNames {
psFlavors = append(psFlavors, kueue.PodSetAssignment{
Name: name,
Flavors: make(map[corev1.ResourceName]kueue.ResourceFlavorReference),
ResourceUsage: make(corev1.ResourceList),
Count: 1,
})
}
wrap.PodSetAssignments = psFlavors
return wrap
}