@@ -103,10 +103,9 @@ func TestUpdatePodLabelForAdmissionController(t *testing.T) {
103103 // verify when appId/queue are not given,
104104 pod := createTestingPodWithMeta ()
105105
106- if result := updatePodLabel (pod , "default" , false , "root.default" ); result != nil {
107- assert .Equal (t , len (result ), 4 )
106+ if result := updatePodLabel (pod , "default" , false ); result != nil {
107+ assert .Equal (t , len (result ), 3 )
108108 assert .Equal (t , result ["random" ], "random" )
109- assert .Equal (t , result ["queue" ], "root.default" )
110109 assert .Equal (t , result ["disableStateAware" ], "true" )
111110 assert .Equal (t , strings .HasPrefix (result ["applicationId" ], constants .AutoGenAppPrefix ), true )
112111 } else {
@@ -117,10 +116,9 @@ func TestUpdatePodLabelForAdmissionController(t *testing.T) {
117116 // we won't modify it
118117 pod = createTestingPodWithAppId ()
119118
120- if result := updatePodLabel (pod , "default" , false , "root.default" ); result != nil {
121- assert .Equal (t , len (result ), 3 )
119+ if result := updatePodLabel (pod , "default" , false ); result != nil {
120+ assert .Equal (t , len (result ), 2 )
122121 assert .Equal (t , result ["random" ], "random" )
123- assert .Equal (t , result ["queue" ], "root.default" )
124122 assert .Equal (t , result ["applicationId" ], "app-0001" )
125123 } else {
126124 t .Fatal ("UpdatePodLabelForAdmissionController is not as expected" )
@@ -129,23 +127,22 @@ func TestUpdatePodLabelForAdmissionController(t *testing.T) {
129127 // verify if queue is given in the labels,
130128 // we won't modify it
131129 pod = createTestingPodWithQueue ()
132- if result := updatePodLabel (pod , "default" , false , "root.default" ); result != nil {
130+ if result := updatePodLabel (pod , "default" , false ); result != nil {
133131 assert .Equal (t , len (result ), 4 )
134132 assert .Equal (t , result ["random" ], "random" )
135- assert .Equal (t , result ["queue" ], "root.abc" )
136133 assert .Equal (t , result ["disableStateAware" ], "true" )
134+ assert .Equal (t , result ["queue" ], "root.abc" )
137135 assert .Equal (t , strings .HasPrefix (result ["applicationId" ], constants .AutoGenAppPrefix ), true )
138136 } else {
139- t .Fatal ("UpdatePodLabelForAdmissionControllert is not as expected" )
137+ t .Fatal ("UpdatePodLabelForAdmissionController is not as expected" )
140138 }
141139
142140 // namespace might be empty
143141 // labels might be empty
144142 pod = createTestingPodNoNamespaceAndLabels ()
145143
146- if result := updatePodLabel (pod , "default" , false , "root.default" ); result != nil {
147- assert .Equal (t , len (result ), 3 )
148- assert .Equal (t , result ["queue" ], "root.default" )
144+ if result := updatePodLabel (pod , "default" , false ); result != nil {
145+ assert .Equal (t , len (result ), 2 )
149146 assert .Equal (t , result ["disableStateAware" ], "true" )
150147 assert .Equal (t , strings .HasPrefix (result ["applicationId" ], constants .AutoGenAppPrefix ), true )
151148 } else {
@@ -154,76 +151,20 @@ func TestUpdatePodLabelForAdmissionController(t *testing.T) {
154151
155152 // pod name might be empty, it can comes from generatedName
156153 pod = createTestingPodWithGenerateName ()
157- if result := updatePodLabel (pod , "default" , false , "root.default" ); result != nil {
158- assert .Equal (t , len (result ), 3 )
159- assert .Equal (t , result ["queue" ], "root.default" )
154+ if result := updatePodLabel (pod , "default" , false ); result != nil {
155+ assert .Equal (t , len (result ), 2 )
160156 assert .Equal (t , result ["disableStateAware" ], "true" )
161157 assert .Equal (t , strings .HasPrefix (result ["applicationId" ], constants .AutoGenAppPrefix ), true )
162158 } else {
163159 t .Fatal ("UpdatePodLabelForAdmissionController is not as expected" )
164160 }
165161
166162 pod = createMinimalTestingPod ()
167- if result := updatePodLabel (pod , "default" , false , "root.default" ); result != nil {
168- assert .Equal (t , len (result ), 3 )
169- assert .Equal (t , result ["queue" ], "root.default" )
163+ if result := updatePodLabel (pod , "default" , false ); result != nil {
164+ assert .Equal (t , len (result ), 2 )
170165 assert .Equal (t , result ["disableStateAware" ], "true" )
171166 assert .Equal (t , strings .HasPrefix (result ["applicationId" ], constants .AutoGenAppPrefix ), true )
172167 } else {
173168 t .Fatal ("UpdatePodLabelForAdmissionController is not as expected" )
174169 }
175170}
176-
177- func TestDefaultQueueName (t * testing.T ) {
178- defaultConf := createConfig ()
179- pod := createTestingPodWithMeta ()
180- if result := updatePodLabel (pod , defaultConf .GetNamespace (), defaultConf .GetGenerateUniqueAppIds (), defaultConf .GetDefaultQueueName ()); result != nil {
181- assert .Equal (t , len (result ), 4 )
182- assert .Equal (t , result ["random" ], "random" )
183- assert .Equal (t , result ["applicationId" ], "yunikorn-default-autogen" )
184- assert .Equal (t , result ["disableStateAware" ], "true" )
185- assert .Equal (t , result ["queue" ], "root.default" )
186- } else {
187- t .Fatal ("UpdatePodLabelForAdmissionController is not as expected" )
188- }
189-
190- queueNameEmptyConf := createConfigWithOverrides (map [string ]string {
191- conf .AMFilteringDefaultQueueName : "" ,
192- })
193- if result := updatePodLabel (pod , queueNameEmptyConf .GetNamespace (), queueNameEmptyConf .GetGenerateUniqueAppIds (), queueNameEmptyConf .GetDefaultQueueName ()); result != nil {
194- assert .Equal (t , len (result ), 3 )
195- assert .Equal (t , result ["random" ], "random" )
196- assert .Equal (t , result ["applicationId" ], "yunikorn-default-autogen" )
197- assert .Equal (t , result ["disableStateAware" ], "true" )
198- assert .Equal (t , result ["queue" ], "" )
199- } else {
200- t .Fatal ("UpdatePodLabelForAdmissionController is not as expected" )
201- }
202-
203- customQueueNameConf := createConfigWithOverrides (map [string ]string {
204- conf .AMFilteringDefaultQueueName : "yunikorn" ,
205- })
206- if result := updatePodLabel (pod , customQueueNameConf .GetNamespace (), customQueueNameConf .GetGenerateUniqueAppIds (), customQueueNameConf .GetDefaultQueueName ()); result != nil {
207- assert .Equal (t , len (result ), 4 )
208- assert .Equal (t , result ["random" ], "random" )
209- assert .Equal (t , result ["applicationId" ], "yunikorn-default-autogen" )
210- assert .Equal (t , result ["disableStateAware" ], "true" )
211- assert .Assert (t , result ["queue" ] != "yunikorn" )
212- } else {
213- t .Fatal ("UpdatePodLabelForAdmissionController is not as expected" )
214- }
215-
216- customValidQueueNameConf := createConfigWithOverrides (map [string ]string {
217- conf .AMFilteringDefaultQueueName : "root.yunikorn" ,
218- })
219- if result := updatePodLabel (pod , customValidQueueNameConf .GetNamespace (),
220- customValidQueueNameConf .GetGenerateUniqueAppIds (), customValidQueueNameConf .GetDefaultQueueName ()); result != nil {
221- assert .Equal (t , len (result ), 4 )
222- assert .Equal (t , result ["random" ], "random" )
223- assert .Equal (t , result ["applicationId" ], "yunikorn-default-autogen" )
224- assert .Equal (t , result ["disableStateAware" ], "true" )
225- assert .Equal (t , result ["queue" ], "root.yunikorn" )
226- } else {
227- t .Fatal ("UpdatePodLabelForAdmissionController is not as expected" )
228- }
229- }
0 commit comments