3535import org .apache .hadoop .yarn .server .resourcemanager .scheduler .capacity .QueuePrefixes ;
3636import org .apache .hadoop .yarn .server .resourcemanager .webapp .dao .NodeLabelInfo ;
3737import org .apache .hadoop .yarn .server .resourcemanager .webapp .dao .NodeLabelsInfo ;
38+ import org .apache .hadoop .yarn .server .resourcemanager .webapp .jsonprovider .ExcludeRootJSONProvider ;
39+ import org .apache .hadoop .yarn .server .resourcemanager .webapp .jsonprovider .IncludeRootJSONProvider ;
3840import org .apache .hadoop .yarn .server .resourcemanager .webapp .jsonprovider .JsonProviderFeature ;
3941import org .apache .hadoop .yarn .webapp .GenericExceptionHandler ;
4042import org .apache .hadoop .yarn .webapp .JerseyTestBase ;
@@ -245,7 +247,9 @@ public void testFormatSchedulerConf() throws Exception {
245247 updateInfo .getUpdateQueueInfo ().add (stoppedInfo );
246248
247249 // Add a queue root.formattest to the existing three queues
248- WebTarget r = target ();
250+ WebTarget r = target ()
251+ .register (new IncludeRootJSONProvider ())
252+ .register (new ExcludeRootJSONProvider ());
249253 Response response = r .path ("ws" ).path ("v1" ).path ("cluster" )
250254 .path ("scheduler-conf" ).queryParam ("user.name" , userName )
251255 .request (MediaType .APPLICATION_JSON )
@@ -290,7 +294,9 @@ public void testAddNestedQueue() throws Exception {
290294 assertNotNull (orgConf );
291295 assertEquals (4 , orgConf .getQueues (ROOT ).size ());
292296
293- WebTarget r = target ();
297+ WebTarget r = target ()
298+ .register (new IncludeRootJSONProvider ())
299+ .register (new ExcludeRootJSONProvider ());
294300
295301 Response response ;
296302
@@ -335,7 +341,9 @@ public void testAddNestedQueue() throws Exception {
335341
336342 @ Test
337343 public void testAddWithUpdate () throws Exception {
338- WebTarget r = target ();
344+ WebTarget r = target ()
345+ .register (new IncludeRootJSONProvider ())
346+ .register (new ExcludeRootJSONProvider ());
339347
340348 Response response ;
341349
@@ -365,7 +373,9 @@ public void testAddWithUpdate() throws Exception {
365373
366374 @ Test
367375 public void testUnsetParentQueueOrderingPolicy () throws Exception {
368- WebTarget r = target ();
376+ WebTarget r = target ()
377+ .register (new IncludeRootJSONProvider ())
378+ .register (new ExcludeRootJSONProvider ());
369379 Response response ;
370380
371381 // Update ordering policy of Leaf Queue root.b to fair
@@ -412,7 +422,9 @@ public void testUnsetParentQueueOrderingPolicy() throws Exception {
412422
413423 @ Test
414424 public void testUnsetLeafQueueOrderingPolicy () throws Exception {
415- WebTarget r = target ();
425+ WebTarget r = target ()
426+ .register (new IncludeRootJSONProvider ())
427+ .register (new ExcludeRootJSONProvider ());
416428 Response response ;
417429
418430 // Update ordering policy of Parent Queue root.c to priority-utilization
@@ -456,7 +468,9 @@ public void testUnsetLeafQueueOrderingPolicy() throws Exception {
456468
457469 @ Test
458470 public void testRemoveQueue () throws Exception {
459- WebTarget r = target ();
471+ WebTarget r = target ()
472+ .register (new IncludeRootJSONProvider ())
473+ .register (new ExcludeRootJSONProvider ());
460474
461475 Response response ;
462476
@@ -482,7 +496,9 @@ public void testRemoveQueue() throws Exception {
482496
483497 @ Test
484498 public void testStopWithRemoveQueue () throws Exception {
485- WebTarget r = target ();
499+ WebTarget r = target ()
500+ .register (new IncludeRootJSONProvider ())
501+ .register (new ExcludeRootJSONProvider ());
486502
487503 Response response ;
488504
@@ -511,7 +527,9 @@ public void testStopWithRemoveQueue() throws Exception {
511527
512528 @ Test
513529 public void testRemoveQueueWhichHasQueueMapping () throws Exception {
514- WebTarget r = target ();
530+ WebTarget r = target ()
531+ .register (new IncludeRootJSONProvider ())
532+ .register (new ExcludeRootJSONProvider ());
515533
516534 Response response ;
517535 CapacityScheduler cs = (CapacityScheduler ) rm .getResourceScheduler ();
@@ -550,7 +568,9 @@ public void testRemoveQueueWhichHasQueueMapping() throws Exception {
550568
551569 @ Test
552570 public void testStopWithConvertLeafToParentQueue () throws Exception {
553- WebTarget r = target ();
571+ WebTarget r = target ()
572+ .register (new IncludeRootJSONProvider ())
573+ .register (new ExcludeRootJSONProvider ());
554574 Response response ;
555575
556576 // Set state of queues to STOPPED.
@@ -581,7 +601,9 @@ public void testStopWithConvertLeafToParentQueue() throws Exception {
581601
582602 @ Test
583603 public void testRemoveParentQueue () throws Exception {
584- WebTarget r = target ();
604+ WebTarget r = target ()
605+ .register (new IncludeRootJSONProvider ())
606+ .register (new ExcludeRootJSONProvider ());
585607
586608 Response response ;
587609
@@ -604,7 +626,9 @@ public void testRemoveParentQueue() throws Exception {
604626
605627 @ Test
606628 public void testRemoveParentQueueWithCapacity () throws Exception {
607- WebTarget r = target ();
629+ WebTarget r = target ()
630+ .register (new IncludeRootJSONProvider ())
631+ .register (new ExcludeRootJSONProvider ());
608632
609633 Response response ;
610634
@@ -634,7 +658,9 @@ public void testRemoveParentQueueWithCapacity() throws Exception {
634658
635659 @ Test
636660 public void testRemoveMultipleQueues () throws Exception {
637- WebTarget r = target ();
661+ WebTarget r = target ()
662+ .register (new IncludeRootJSONProvider ())
663+ .register (new ExcludeRootJSONProvider ());
638664
639665 Response response ;
640666
@@ -661,7 +687,9 @@ public void testRemoveMultipleQueues() throws Exception {
661687 }
662688
663689 private void stopQueue (QueuePath ... queuePaths ) throws Exception {
664- WebTarget r = target ();
690+ WebTarget r = target ()
691+ .register (new IncludeRootJSONProvider ())
692+ .register (new ExcludeRootJSONProvider ());
665693
666694 Response response ;
667695
@@ -689,7 +717,9 @@ private void stopQueue(QueuePath... queuePaths) throws Exception {
689717
690718 @ Test
691719 public void testUpdateQueue () throws Exception {
692- WebTarget r = target ();
720+ WebTarget r = target ()
721+ .register (new IncludeRootJSONProvider ())
722+ .register (new ExcludeRootJSONProvider ());
693723
694724 Response response ;
695725
@@ -740,7 +770,9 @@ public void testUpdateQueue() throws Exception {
740770
741771 @ Test
742772 public void testUpdateQueueCapacity () throws Exception {
743- WebTarget r = target ();
773+ WebTarget r = target ()
774+ .register (new IncludeRootJSONProvider ())
775+ .register (new ExcludeRootJSONProvider ());
744776
745777 Response response ;
746778
@@ -767,7 +799,9 @@ public void testUpdateQueueCapacity() throws Exception {
767799
768800 @ Test
769801 public void testGlobalConfChange () throws Exception {
770- WebTarget r = target ();
802+ WebTarget r = target ()
803+ .register (new IncludeRootJSONProvider ())
804+ .register (new ExcludeRootJSONProvider ());
771805
772806 Response response ;
773807
@@ -804,7 +838,9 @@ public void testGlobalConfChange() throws Exception {
804838
805839 @ Test
806840 public void testNodeLabelRemovalResidualConfigsAreCleared () throws Exception {
807- WebTarget r = target ();
841+ WebTarget r = target ()
842+ .register (new IncludeRootJSONProvider ())
843+ .register (new ExcludeRootJSONProvider ());
808844 Response response ;
809845
810846 // 1. Create Node Label: label1
@@ -1007,7 +1043,9 @@ private String getAccessibleNodeLabelsMaxCapacityPropertyName(String label) {
10071043
10081044 @ Test
10091045 public void testValidateWithClusterMaxAllocation () throws Exception {
1010- WebTarget r = target ();
1046+ WebTarget r = target ()
1047+ .register (new IncludeRootJSONProvider ())
1048+ .register (new ExcludeRootJSONProvider ());
10111049 int clusterMax = YarnConfiguration .
10121050 DEFAULT_RM_SCHEDULER_MAXIMUM_ALLOCATION_MB * 2 ;
10131051 conf .setInt (YarnConfiguration .RM_SCHEDULER_MAXIMUM_ALLOCATION_MB ,
0 commit comments