YARN-7387: org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.TestIncreaseAllocationExpirer fails intermittently. Contributed by Jim Brennan (Jim_Brennan)

hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/capacity/TestIncreaseAllocationExpirer.java

@@ -227,6 +227,7 @@ public class TestIncreaseAllocationExpirer {
     Thread.sleep(10000);
     // Verify container size is 1G
     am1.allocate(null, null);
+    rm1.drainEvents();
     Assert.assertEquals(
         1 * GB, rm1.getResourceScheduler().getRMContainer(containerId2)
             .getAllocatedResource().getMemorySize());
@@ -467,6 +468,7 @@ public class TestIncreaseAllocationExpirer {
 
     am1.allocate(null, null);
 
+    rm1.drainEvents();
     Assert.assertEquals(
         2 * GB, rm1.getResourceScheduler().getRMContainer(containerId2)
             .getAllocatedResource().getMemorySize());