@@ -30,18 +30,16 @@ import (
3030 "sigs.k8s.io/kueue/pkg/controller/constants"
3131 "sigs.k8s.io/kueue/pkg/controller/jobframework"
3232 podconstants "sigs.k8s.io/kueue/pkg/controller/jobs/pod/constants"
33- "sigs.k8s.io/kueue/pkg/features"
3433 utiltesting "sigs.k8s.io/kueue/pkg/util/testing"
3534 utiltestingapi "sigs.k8s.io/kueue/pkg/util/testing/v1beta2"
3635 testingdeployment "sigs.k8s.io/kueue/pkg/util/testingjobs/deployment"
3736)
3837
3938func TestDefault (t * testing.T ) {
4039 testCases := map [string ]struct {
41- deployment * appsv1.Deployment
42- localQueueDefaulting bool
43- defaultLqExist bool
44- want * appsv1.Deployment
40+ deployment * appsv1.Deployment
41+ defaultLqExist bool
42+ want * appsv1.Deployment
4543 }{
4644 "deployment without queue" : {
4745 deployment : testingdeployment .MakeDeployment ("test-pod" , "" ).Obj (),
@@ -74,32 +72,29 @@ func TestDefault(t *testing.T) {
7472 deployment : testingdeployment .MakeDeployment ("test-pod" , "" ).PodTemplateSpecQueue ("test-queue" ).Obj (),
7573 want : testingdeployment .MakeDeployment ("test-pod" , "" ).PodTemplateSpecQueue ("test-queue" ).Obj (),
7674 },
77- "LocalQueueDefaulting enabled, default lq is created, job doesn't have queue label" : {
78- localQueueDefaulting : true ,
79- defaultLqExist : true ,
80- deployment : testingdeployment .MakeDeployment ("test-pod" , "default" ).Obj (),
75+ "default lq is created, job doesn't have queue label" : {
76+ defaultLqExist : true ,
77+ deployment : testingdeployment .MakeDeployment ("test-pod" , "default" ).Obj (),
8178 want : testingdeployment .MakeDeployment ("test-pod" , "default" ).
8279 PodTemplateSpecManagedByKueue ().
8380 Queue ("default" ).
8481 PodTemplateSpecQueue ("default" ).
8582 PodTemplateAnnotation (podconstants .SuspendedByParentAnnotation , FrameworkName ).
8683 Obj (),
8784 },
88- "LocalQueueDefaulting enabled, default lq is created, job has queue label" : {
89- localQueueDefaulting : true ,
90- defaultLqExist : true ,
91- deployment : testingdeployment .MakeDeployment ("test-pod" , "" ).Queue ("test-queue" ).Obj (),
85+ "default lq is created, job has queue label" : {
86+ defaultLqExist : true ,
87+ deployment : testingdeployment .MakeDeployment ("test-pod" , "" ).Queue ("test-queue" ).Obj (),
9288 want : testingdeployment .MakeDeployment ("test-pod" , "" ).
9389 PodTemplateSpecManagedByKueue ().
9490 Queue ("test-queue" ).
9591 PodTemplateSpecQueue ("test-queue" ).
9692 PodTemplateAnnotation (podconstants .SuspendedByParentAnnotation , FrameworkName ).
9793 Obj (),
9894 },
99- "LocalQueueDefaulting enabled, default lq isn't created, job doesn't have queue label" : {
100- localQueueDefaulting : true ,
101- defaultLqExist : false ,
102- deployment : testingdeployment .MakeDeployment ("test-pod" , "" ).Obj (),
95+ "default lq isn't created, job doesn't have queue label" : {
96+ defaultLqExist : false ,
97+ deployment : testingdeployment .MakeDeployment ("test-pod" , "" ).Obj (),
10398 want : testingdeployment .MakeDeployment ("test-pod" , "" ).
10499 Obj (),
105100 },
@@ -148,7 +143,6 @@ func TestDefault(t *testing.T) {
148143 for name , tc := range testCases {
149144 t .Run (name , func (t * testing.T ) {
150145 ctx , _ := utiltesting .ContextWithLog (t )
151- features .SetFeatureGateDuringTest (t , features .LocalQueueDefaulting , tc .localQueueDefaulting )
152146 t .Cleanup (jobframework .EnableIntegrationsForTest (t , "pod" ))
153147 builder := utiltesting .NewClientBuilder ()
154148 client := builder .Build ()
0 commit comments