1717package com .cloud .kubernetes .cluster ;
1818
1919
20+ import java .util .Arrays ;
2021import java .util .HashMap ;
22+ import java .util .List ;
2123import java .util .Map ;
2224import java .util .UUID ;
2325
@@ -218,115 +220,129 @@ public void testCheckNodeTypeAffinityGroupEntryNodeTypeValid() {
218220 }
219221
220222 @ Test (expected = InvalidParameterValueException .class )
221- public void testValidateAffinityGroupUuidBlank () {
222- kubernetesServiceHelper .validateAffinityGroupUuid ("" );
223+ public void testValidateAffinityGroupUuidAndGetIdBlank () {
224+ kubernetesServiceHelper .validateAffinityGroupUuidAndGetId ("" );
223225 }
224226
225227 @ Test (expected = InvalidParameterValueException .class )
226- public void testValidateAffinityGroupUuidNotFound () {
228+ public void testValidateAffinityGroupUuidAndGetIdNotFound () {
227229 Mockito .when (affinityGroupDao .findByUuid ("non-existent-uuid" )).thenReturn (null );
228- kubernetesServiceHelper .validateAffinityGroupUuid ("non-existent-uuid" );
230+ kubernetesServiceHelper .validateAffinityGroupUuidAndGetId ("non-existent-uuid" );
229231 }
230232
231233 @ Test
232- public void testValidateAffinityGroupUuidValid () {
234+ public void testValidateAffinityGroupUuidAndGetIdValid () {
233235 AffinityGroupVO affinityGroup = Mockito .mock (AffinityGroupVO .class );
236+ Mockito .when (affinityGroup .getId ()).thenReturn (100L );
234237 Mockito .when (affinityGroupDao .findByUuid ("valid-uuid" )).thenReturn (affinityGroup );
235- kubernetesServiceHelper .validateAffinityGroupUuid ("valid-uuid" );
238+ Long result = kubernetesServiceHelper .validateAffinityGroupUuidAndGetId ("valid-uuid" );
239+ Assert .assertEquals (Long .valueOf (100L ), result );
236240 }
237241
238242 @ Test
239- public void testValidateAndNormalizeAffinityGroupUuidsSingleUuid () {
243+ public void testValidateAndGetAffinityGroupIdsSingleUuid () {
240244 AffinityGroupVO affinityGroup = Mockito .mock (AffinityGroupVO .class );
245+ Mockito .when (affinityGroup .getId ()).thenReturn (1L );
241246 Mockito .when (affinityGroupDao .findByUuid ("uuid1" )).thenReturn (affinityGroup );
242247
243- String result = kubernetesServiceHelper .validateAndNormalizeAffinityGroupUuids ("uuid1" );
244- Assert .assertEquals ("uuid1" , result );
248+ List <Long > result = kubernetesServiceHelper .validateAndGetAffinityGroupIds ("uuid1" );
249+ Assert .assertEquals (1 , result .size ());
250+ Assert .assertEquals (Long .valueOf (1L ), result .get (0 ));
245251 }
246252
247253 @ Test
248- public void testValidateAndNormalizeAffinityGroupUuidsMultipleUuids () {
254+ public void testValidateAndGetAffinityGroupIdsMultipleUuids () {
249255 AffinityGroupVO affinityGroup1 = Mockito .mock (AffinityGroupVO .class );
250256 AffinityGroupVO affinityGroup2 = Mockito .mock (AffinityGroupVO .class );
251257 AffinityGroupVO affinityGroup3 = Mockito .mock (AffinityGroupVO .class );
258+ Mockito .when (affinityGroup1 .getId ()).thenReturn (1L );
259+ Mockito .when (affinityGroup2 .getId ()).thenReturn (2L );
260+ Mockito .when (affinityGroup3 .getId ()).thenReturn (3L );
252261 Mockito .when (affinityGroupDao .findByUuid ("uuid1" )).thenReturn (affinityGroup1 );
253262 Mockito .when (affinityGroupDao .findByUuid ("uuid2" )).thenReturn (affinityGroup2 );
254263 Mockito .when (affinityGroupDao .findByUuid ("uuid3" )).thenReturn (affinityGroup3 );
255264
256- String result = kubernetesServiceHelper .validateAndNormalizeAffinityGroupUuids ("uuid1,uuid2,uuid3" );
257- Assert .assertEquals ("uuid1,uuid2,uuid3" , result );
265+ List <Long > result = kubernetesServiceHelper .validateAndGetAffinityGroupIds ("uuid1,uuid2,uuid3" );
266+ Assert .assertEquals (3 , result .size ());
267+ Assert .assertEquals (Arrays .asList (1L , 2L , 3L ), result );
258268 }
259269
260270 @ Test
261- public void testValidateAndNormalizeAffinityGroupUuidsWithSpaces () {
271+ public void testValidateAndGetAffinityGroupIdsWithSpaces () {
262272 AffinityGroupVO affinityGroup1 = Mockito .mock (AffinityGroupVO .class );
263273 AffinityGroupVO affinityGroup2 = Mockito .mock (AffinityGroupVO .class );
274+ Mockito .when (affinityGroup1 .getId ()).thenReturn (1L );
275+ Mockito .when (affinityGroup2 .getId ()).thenReturn (2L );
264276 Mockito .when (affinityGroupDao .findByUuid ("uuid1" )).thenReturn (affinityGroup1 );
265277 Mockito .when (affinityGroupDao .findByUuid ("uuid2" )).thenReturn (affinityGroup2 );
266278
267- String result = kubernetesServiceHelper .validateAndNormalizeAffinityGroupUuids (" uuid1 , uuid2 " );
268- Assert .assertEquals ("uuid1,uuid2" , result );
279+ List <Long > result = kubernetesServiceHelper .validateAndGetAffinityGroupIds (" uuid1 , uuid2 " );
280+ Assert .assertEquals (2 , result .size ());
281+ Assert .assertEquals (Arrays .asList (1L , 2L ), result );
269282 }
270283
271284 @ Test (expected = InvalidParameterValueException .class )
272- public void testValidateAndNormalizeAffinityGroupUuidsOneInvalid () {
285+ public void testValidateAndGetAffinityGroupIdsOneInvalid () {
273286 AffinityGroupVO affinityGroup1 = Mockito .mock (AffinityGroupVO .class );
274287 Mockito .when (affinityGroupDao .findByUuid ("uuid1" )).thenReturn (affinityGroup1 );
275288 Mockito .when (affinityGroupDao .findByUuid ("invalid-uuid" )).thenReturn (null );
276289
277- kubernetesServiceHelper .validateAndNormalizeAffinityGroupUuids ("uuid1,invalid-uuid" );
290+ kubernetesServiceHelper .validateAndGetAffinityGroupIds ("uuid1,invalid-uuid" );
278291 }
279292
280293 @ Test
281294 public void testAddNodeTypeAffinityGroupEntry () {
282- Map <String , String > mapping = new HashMap <>();
283- kubernetesServiceHelper .addNodeTypeAffinityGroupEntry ("control" , "uuid1,uuid2" , mapping );
295+ Map <String , List < Long > > mapping = new HashMap <>();
296+ kubernetesServiceHelper .addNodeTypeAffinityGroupEntry ("control" , Arrays . asList ( 1L , 2L ) , mapping );
284297 Assert .assertEquals (1 , mapping .size ());
285- Assert .assertEquals ("uuid1,uuid2" , mapping .get ("CONTROL" ));
298+ Assert .assertEquals (Arrays . asList ( 1L , 2L ) , mapping .get ("CONTROL" ));
286299 }
287300
288301 @ Test
289302 public void testProcessNodeTypeAffinityGroupEntryAndAddToMappingIfValidEmptyEntry () {
290- Map <String , String > mapping = new HashMap <>();
303+ Map <String , List < Long > > mapping = new HashMap <>();
291304 kubernetesServiceHelper .processNodeTypeAffinityGroupEntryAndAddToMappingIfValid (new HashMap <>(), mapping );
292305 Assert .assertTrue (mapping .isEmpty ());
293306 }
294307
295308 @ Test
296309 public void testProcessNodeTypeAffinityGroupEntryAndAddToMappingIfValidValidEntry () {
297310 AffinityGroupVO affinityGroup = Mockito .mock (AffinityGroupVO .class );
311+ Mockito .when (affinityGroup .getId ()).thenReturn (100L );
298312 Mockito .when (affinityGroupDao .findByUuid ("affinity-group-uuid" )).thenReturn (affinityGroup );
299313
300314 Map <String , String > entry = new HashMap <>();
301315 entry .put (VmDetailConstants .CKS_NODE_TYPE , "control" );
302316 entry .put (VmDetailConstants .AFFINITY_GROUP , "affinity-group-uuid" );
303317
304- Map <String , String > mapping = new HashMap <>();
318+ Map <String , List < Long > > mapping = new HashMap <>();
305319 kubernetesServiceHelper .processNodeTypeAffinityGroupEntryAndAddToMappingIfValid (entry , mapping );
306320 Assert .assertEquals (1 , mapping .size ());
307- Assert .assertEquals ("affinity-group-uuid" , mapping .get ("CONTROL" ));
321+ Assert .assertEquals (Arrays . asList ( 100L ) , mapping .get ("CONTROL" ));
308322 }
309323
310324 @ Test
311325 public void testProcessNodeTypeAffinityGroupEntryAndAddToMappingIfValidMultipleUuids () {
312326 AffinityGroupVO affinityGroup1 = Mockito .mock (AffinityGroupVO .class );
313327 AffinityGroupVO affinityGroup2 = Mockito .mock (AffinityGroupVO .class );
328+ Mockito .when (affinityGroup1 .getId ()).thenReturn (1L );
329+ Mockito .when (affinityGroup2 .getId ()).thenReturn (2L );
314330 Mockito .when (affinityGroupDao .findByUuid ("uuid1" )).thenReturn (affinityGroup1 );
315331 Mockito .when (affinityGroupDao .findByUuid ("uuid2" )).thenReturn (affinityGroup2 );
316332
317333 Map <String , String > entry = new HashMap <>();
318334 entry .put (VmDetailConstants .CKS_NODE_TYPE , "worker" );
319335 entry .put (VmDetailConstants .AFFINITY_GROUP , "uuid1,uuid2" );
320336
321- Map <String , String > mapping = new HashMap <>();
337+ Map <String , List < Long > > mapping = new HashMap <>();
322338 kubernetesServiceHelper .processNodeTypeAffinityGroupEntryAndAddToMappingIfValid (entry , mapping );
323339 Assert .assertEquals (1 , mapping .size ());
324- Assert .assertEquals ("uuid1,uuid2" , mapping .get ("WORKER" ));
340+ Assert .assertEquals (Arrays . asList ( 1L , 2L ) , mapping .get ("WORKER" ));
325341 }
326342
327343 @ Test
328344 public void testGetAffinityGroupNodeTypeMapEmptyMap () {
329- Map <String , String > result = kubernetesServiceHelper .getAffinityGroupNodeTypeMap (null );
345+ Map <String , List < Long > > result = kubernetesServiceHelper .getAffinityGroupNodeTypeMap (null );
330346 Assert .assertTrue (result .isEmpty ());
331347
332348 result = kubernetesServiceHelper .getAffinityGroupNodeTypeMap (new HashMap <>());
@@ -336,9 +352,11 @@ public void testGetAffinityGroupNodeTypeMapEmptyMap() {
336352 @ Test
337353 public void testGetAffinityGroupNodeTypeMapValidEntries () {
338354 AffinityGroupVO controlAffinityGroup = Mockito .mock (AffinityGroupVO .class );
355+ Mockito .when (controlAffinityGroup .getId ()).thenReturn (100L );
339356 Mockito .when (affinityGroupDao .findByUuid ("control-affinity-uuid" )).thenReturn (controlAffinityGroup );
340357
341358 AffinityGroupVO workerAffinityGroup = Mockito .mock (AffinityGroupVO .class );
359+ Mockito .when (workerAffinityGroup .getId ()).thenReturn (200L );
342360 Mockito .when (affinityGroupDao .findByUuid ("worker-affinity-uuid" )).thenReturn (workerAffinityGroup );
343361
344362 Map <String , Map <String , String >> affinityGroupNodeTypeMap = new HashMap <>();
@@ -353,17 +371,20 @@ public void testGetAffinityGroupNodeTypeMapValidEntries() {
353371 workerEntry .put (VmDetailConstants .AFFINITY_GROUP , "worker-affinity-uuid" );
354372 affinityGroupNodeTypeMap .put ("1" , workerEntry );
355373
356- Map <String , String > result = kubernetesServiceHelper .getAffinityGroupNodeTypeMap (affinityGroupNodeTypeMap );
374+ Map <String , List < Long > > result = kubernetesServiceHelper .getAffinityGroupNodeTypeMap (affinityGroupNodeTypeMap );
357375 Assert .assertEquals (2 , result .size ());
358- Assert .assertEquals ("control-affinity-uuid" , result .get ("CONTROL" ));
359- Assert .assertEquals ("worker-affinity-uuid" , result .get ("WORKER" ));
376+ Assert .assertEquals (Arrays . asList ( 100L ) , result .get ("CONTROL" ));
377+ Assert .assertEquals (Arrays . asList ( 200L ) , result .get ("WORKER" ));
360378 }
361379
362380 @ Test
363- public void testGetAffinityGroupNodeTypeMapMultipleUuidsPerNodeType () {
381+ public void testGetAffinityGroupNodeTypeMapMultipleIdsPerNodeType () {
364382 AffinityGroupVO ag1 = Mockito .mock (AffinityGroupVO .class );
365383 AffinityGroupVO ag2 = Mockito .mock (AffinityGroupVO .class );
366384 AffinityGroupVO ag3 = Mockito .mock (AffinityGroupVO .class );
385+ Mockito .when (ag1 .getId ()).thenReturn (1L );
386+ Mockito .when (ag2 .getId ()).thenReturn (2L );
387+ Mockito .when (ag3 .getId ()).thenReturn (3L );
367388 Mockito .when (affinityGroupDao .findByUuid ("ag1" )).thenReturn (ag1 );
368389 Mockito .when (affinityGroupDao .findByUuid ("ag2" )).thenReturn (ag2 );
369390 Mockito .when (affinityGroupDao .findByUuid ("ag3" )).thenReturn (ag3 );
@@ -375,8 +396,8 @@ public void testGetAffinityGroupNodeTypeMapMultipleUuidsPerNodeType() {
375396 controlEntry .put (VmDetailConstants .AFFINITY_GROUP , "ag1,ag2,ag3" );
376397 affinityGroupNodeTypeMap .put ("0" , controlEntry );
377398
378- Map <String , String > result = kubernetesServiceHelper .getAffinityGroupNodeTypeMap (affinityGroupNodeTypeMap );
399+ Map <String , List < Long > > result = kubernetesServiceHelper .getAffinityGroupNodeTypeMap (affinityGroupNodeTypeMap );
379400 Assert .assertEquals (1 , result .size ());
380- Assert .assertEquals ("ag1,ag2,ag3" , result .get ("CONTROL" ));
401+ Assert .assertEquals (Arrays . asList ( 1L , 2L , 3L ) , result .get ("CONTROL" ));
381402 }
382403}
0 commit comments