Repository: hadoop Updated Branches: refs/heads/branch-2 6de262cdc -> 4037b22b7
YARN-5389. TestYarnClient#testReservationDelete fails. Contributed by Sean Po Project: http://git-wip-us.apache.org/repos/asf/hadoop/repo Commit: http://git-wip-us.apache.org/repos/asf/hadoop/commit/4037b22b Tree: http://git-wip-us.apache.org/repos/asf/hadoop/tree/4037b22b Diff: http://git-wip-us.apache.org/repos/asf/hadoop/diff/4037b22b Branch: refs/heads/branch-2 Commit: 4037b22b7cb2755df17a65b03f6c0b9f63a476cd Parents: 6de262c Author: Jason Lowe <jl...@apache.org> Authored: Thu Aug 25 14:42:06 2016 +0000 Committer: Jason Lowe <jl...@apache.org> Committed: Thu Aug 25 15:10:23 2016 +0000 ---------------------------------------------------------------------- .../yarn/client/api/impl/TestYarnClient.java | 45 +++++++++----------- 1 file changed, 20 insertions(+), 25 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/hadoop/blob/4037b22b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-client/src/test/java/org/apache/hadoop/yarn/client/api/impl/TestYarnClient.java ---------------------------------------------------------------------- diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-client/src/test/java/org/apache/hadoop/yarn/client/api/impl/TestYarnClient.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-client/src/test/java/org/apache/hadoop/yarn/client/api/impl/TestYarnClient.java index 360ff99..8d68acc 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-client/src/test/java/org/apache/hadoop/yarn/client/api/impl/TestYarnClient.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-client/src/test/java/org/apache/hadoop/yarn/client/api/impl/TestYarnClient.java @@ -41,6 +41,7 @@ import java.util.List; import java.util.Map; import java.util.Set; +import com.google.common.base.Supplier; import org.apache.commons.io.IOUtils; import org.apache.hadoop.conf.Configuration; import org.apache.hadoop.io.DataInputByteBuffer; @@ -52,6 +53,7 @@ import org.apache.hadoop.security.UserGroupInformation; import org.apache.hadoop.security.UserGroupInformation.AuthenticationMethod; import org.apache.hadoop.security.token.Token; import org.apache.hadoop.security.token.TokenIdentifier; +import org.apache.hadoop.test.GenericTestUtils; import org.apache.hadoop.yarn.api.ApplicationClientProtocol; import org.apache.hadoop.yarn.api.protocolrecords.GetApplicationAttemptReportRequest; import org.apache.hadoop.yarn.api.protocolrecords.GetApplicationAttemptReportResponse; @@ -1193,35 +1195,27 @@ public class TestYarnClient { } } - private MiniYARNCluster setupMiniYARNCluster() { + private MiniYARNCluster setupMiniYARNCluster() throws Exception { CapacitySchedulerConfiguration conf = new CapacitySchedulerConfiguration(); ReservationSystemTestUtil.setupQueueConfiguration(conf); conf.setClass(YarnConfiguration.RM_SCHEDULER, CapacityScheduler.class, ResourceScheduler.class); conf.setBoolean(YarnConfiguration.RM_RESERVATION_SYSTEM_ENABLE, true); - MiniYARNCluster cluster = + final MiniYARNCluster cluster = new MiniYARNCluster("testReservationAPIs", 2, 1, 1); cluster.init(conf); cluster.start(); - int attempts; - for (attempts = 10; attempts > 0; attempts--) { - if (cluster.getResourceManager().getRMContext().getReservationSystem() - .getPlan(ReservationSystemTestUtil.reservationQ).getTotalCapacity() - .getMemorySize() > 6000) { - break; - } - try { - Thread.sleep(100); - } catch (InterruptedException e) { - e.printStackTrace(); + GenericTestUtils.waitFor(new Supplier<Boolean>() { + @Override + public Boolean get() { + return cluster.getResourceManager().getRMContext() + .getReservationSystem() + .getPlan(ReservationSystemTestUtil.reservationQ) + .getTotalCapacity().getMemorySize() > 6000; } - } - if (attempts <= 0) { - Assert.fail("Exhausted attempts in checking if node capacity was " - + "added to the plan"); - } + }, 10, 10000); return cluster; } @@ -1253,7 +1247,7 @@ public class TestYarnClient { } @Test - public void testCreateReservation() { + public void testCreateReservation() throws Exception { MiniYARNCluster cluster = setupMiniYARNCluster(); YarnClient client = setupYarnClient(cluster); try { @@ -1296,7 +1290,7 @@ public class TestYarnClient { } @Test - public void testUpdateReservation() { + public void testUpdateReservation() throws Exception { MiniYARNCluster cluster = setupMiniYARNCluster(); YarnClient client = setupYarnClient(cluster); try { @@ -1338,7 +1332,7 @@ public class TestYarnClient { } @Test - public void testListReservationsByReservationId() { + public void testListReservationsByReservationId() throws Exception{ MiniYARNCluster cluster = setupMiniYARNCluster(); YarnClient client = setupYarnClient(cluster); try { @@ -1375,7 +1369,7 @@ public class TestYarnClient { } @Test - public void testListReservationsByTimeInterval() { + public void testListReservationsByTimeInterval() throws Exception { MiniYARNCluster cluster = setupMiniYARNCluster(); YarnClient client = setupYarnClient(cluster); try { @@ -1440,7 +1434,7 @@ public class TestYarnClient { } @Test - public void testListReservationsByInvalidTimeInterval() { + public void testListReservationsByInvalidTimeInterval() throws Exception { MiniYARNCluster cluster = setupMiniYARNCluster(); YarnClient client = setupYarnClient(cluster); try { @@ -1490,7 +1484,8 @@ public class TestYarnClient { } @Test - public void testListReservationsByTimeIntervalContainingNoReservations() { + public void testListReservationsByTimeIntervalContainingNoReservations() + throws Exception { MiniYARNCluster cluster = setupMiniYARNCluster(); YarnClient client = setupYarnClient(cluster); try { @@ -1580,7 +1575,7 @@ public class TestYarnClient { } @Test - public void testReservationDelete() { + public void testReservationDelete() throws Exception { MiniYARNCluster cluster = setupMiniYARNCluster(); YarnClient client = setupYarnClient(cluster); try { --------------------------------------------------------------------- To unsubscribe, e-mail: common-commits-unsubscr...@hadoop.apache.org For additional commands, e-mail: common-commits-h...@hadoop.apache.org