Skip to content

Commit

Permalink
nit
Browse files Browse the repository at this point in the history
  • Loading branch information
vicentefb committed Jun 17, 2024
1 parent be8dcd0 commit 698cd3b
Showing 1 changed file with 31 additions and 31 deletions.
62 changes: 31 additions & 31 deletions pkg/controller/core/workload_controller_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ import (
"github.com/google/go-cmp/cmp/cmpopts"
batchv1 "k8s.io/api/batch/v1"
corev1 "k8s.io/api/core/v1"
v1 "k8s.io/api/core/v1"

"k8s.io/apimachinery/pkg/api/errors"
"k8s.io/apimachinery/pkg/api/resource"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
Expand Down Expand Up @@ -446,15 +446,15 @@ func TestReconcile(t *testing.T) {
kueue.PodSet{
Name: "head",
Count: int32(1),
Template: v1.PodTemplateSpec{
Spec: v1.PodSpec{
Template: corev1.PodTemplateSpec{
Spec: corev1.PodSpec{

RestartPolicy: v1.RestartPolicyNever,
Containers: []v1.Container{
RestartPolicy: corev1.RestartPolicyNever,
Containers: []corev1.Container{
{
Name: "head-container",
Resources: v1.ResourceRequirements{
Requests: make(v1.ResourceList),
Resources: corev1.ResourceRequirements{
Requests: make(corev1.ResourceList),
},
},
},
Expand All @@ -464,25 +464,25 @@ func TestReconcile(t *testing.T) {
kueue.PodSet{
Name: "workers-group-0",
Count: int32(3),
Template: v1.PodTemplateSpec{
Spec: v1.PodSpec{
RestartPolicy: v1.RestartPolicyNever,
Template: corev1.PodTemplateSpec{
Spec: corev1.PodSpec{
RestartPolicy: corev1.RestartPolicyNever,

Containers: []v1.Container{
Containers: []corev1.Container{
{
Name: "worker-container",
Resources: v1.ResourceRequirements{
Requests: v1.ResourceList{v1.ResourceCPU: resource.MustParse("1")},
Resources: corev1.ResourceRequirements{
Requests: corev1.ResourceList{corev1.ResourceCPU: resource.MustParse("1")},
},
},
},
},
},
}).
Request(v1.ResourceCPU, "1").
Request(corev1.ResourceCPU, "1").
ReserveQuota(
utiltesting.MakeAdmissionScaleDown(v1.ResourceCPU, "10", int32(4), "cq", "head", "workers-group-0").
Assignment(v1.ResourceCPU, "unit-test-flavor", "1").
utiltesting.MakeAdmissionScaleDown(corev1.ResourceCPU, "10", int32(4), "cq", "head", "workers-group-0").
Assignment(corev1.ResourceCPU, "unit-test-flavor", "1").
AssignmentPodCount(1).
Obj(),
).
Expand All @@ -506,14 +506,14 @@ func TestReconcile(t *testing.T) {
kueue.PodSet{
Name: "head",
Count: int32(1),
Template: v1.PodTemplateSpec{
Spec: v1.PodSpec{
RestartPolicy: v1.RestartPolicyNever,
Containers: []v1.Container{
Template: corev1.PodTemplateSpec{
Spec: corev1.PodSpec{
RestartPolicy: corev1.RestartPolicyNever,
Containers: []corev1.Container{
{
Name: "head-container",
Resources: v1.ResourceRequirements{
Requests: make(v1.ResourceList),
Resources: corev1.ResourceRequirements{
Requests: make(corev1.ResourceList),
},
},
},
Expand All @@ -523,24 +523,24 @@ func TestReconcile(t *testing.T) {
kueue.PodSet{
Name: "workers-group-0",
Count: int32(3),
Template: v1.PodTemplateSpec{
Spec: v1.PodSpec{
RestartPolicy: v1.RestartPolicyNever,
Containers: []v1.Container{
Template: corev1.PodTemplateSpec{
Spec: corev1.PodSpec{
RestartPolicy: corev1.RestartPolicyNever,
Containers: []corev1.Container{
{
Name: "worker-container",
Resources: v1.ResourceRequirements{
Requests: v1.ResourceList{v1.ResourceCPU: resource.MustParse("1")},
Resources: corev1.ResourceRequirements{
Requests: corev1.ResourceList{corev1.ResourceCPU: resource.MustParse("1")},
},
},
},
},
},
}).
Request(v1.ResourceCPU, "1").
Request(corev1.ResourceCPU, "1").
ReserveQuota(
utiltesting.MakeAdmissionScaleDown(v1.ResourceCPU, "9", int32(3), "cq", "head", "workers-group-0").
Assignment(v1.ResourceCPU, "unit-test-flavor", "1").
utiltesting.MakeAdmissionScaleDown(corev1.ResourceCPU, "9", int32(3), "cq", "head", "workers-group-0").
Assignment(corev1.ResourceCPU, "unit-test-flavor", "1").
AssignmentPodCount(1).
Obj(),
).
Expand Down

0 comments on commit 698cd3b

Please sign in to comment.