|
@@ -20,11 +20,7 @@ package org.apache.hadoop.yarn.server.resourcemanager.scheduler.fair;
|
|
|
import static org.apache.hadoop.yarn.server.resourcemanager.scheduler.fair.FairSchedulerConfiguration.parseResourceConfigValue;
|
|
|
import static org.junit.Assert.assertEquals;
|
|
|
|
|
|
-import java.io.File;
|
|
|
-
|
|
|
-import org.junit.Assert;
|
|
|
-
|
|
|
-import org.apache.hadoop.conf.Configuration;
|
|
|
+import org.apache.hadoop.yarn.api.records.Resource;
|
|
|
import org.apache.hadoop.yarn.server.utils.BuilderUtils;
|
|
|
import org.junit.Test;
|
|
|
|
|
@@ -32,23 +28,59 @@ public class TestFairSchedulerConfiguration {
|
|
|
@Test
|
|
|
public void testParseResourceConfigValue() throws Exception {
|
|
|
assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
- parseResourceConfigValue("2 vcores, 1024 mb"));
|
|
|
+ parseResourceConfigValue("2 vcores, 1024 mb").getResource());
|
|
|
+ assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
+ parseResourceConfigValue("1024 mb, 2 vcores").getResource());
|
|
|
+ assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
+ parseResourceConfigValue("2vcores,1024mb").getResource());
|
|
|
+ assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
+ parseResourceConfigValue("1024mb,2vcores").getResource());
|
|
|
+ assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
+ parseResourceConfigValue("1024 mb, 2 vcores").getResource());
|
|
|
assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
- parseResourceConfigValue("1024 mb, 2 vcores"));
|
|
|
+ parseResourceConfigValue("1024 Mb, 2 vCores").getResource());
|
|
|
assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
- parseResourceConfigValue("2vcores,1024mb"));
|
|
|
+ parseResourceConfigValue(" 1024 mb, 2 vcores ").getResource());
|
|
|
assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
- parseResourceConfigValue("1024mb,2vcores"));
|
|
|
+ parseResourceConfigValue(" 1024.3 mb, 2.35 vcores ").getResource());
|
|
|
assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
- parseResourceConfigValue("1024 mb, 2 vcores"));
|
|
|
+ parseResourceConfigValue(" 1024. mb, 2. vcores ").getResource());
|
|
|
+
|
|
|
+ Resource clusterResource = BuilderUtils.newResource(2048, 4);
|
|
|
+ assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
+ parseResourceConfigValue("50% memory, 50% cpu").
|
|
|
+ getResource(clusterResource));
|
|
|
+ assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
+ parseResourceConfigValue("50% Memory, 50% CpU").
|
|
|
+ getResource(clusterResource));
|
|
|
assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
- parseResourceConfigValue("1024 Mb, 2 vCores"));
|
|
|
+ parseResourceConfigValue("50%").getResource(clusterResource));
|
|
|
+ assertEquals(BuilderUtils.newResource(1024, 4),
|
|
|
+ parseResourceConfigValue("50% memory, 100% cpu").
|
|
|
+ getResource(clusterResource));
|
|
|
+ assertEquals(BuilderUtils.newResource(1024, 4),
|
|
|
+ parseResourceConfigValue(" 100% cpu, 50% memory").
|
|
|
+ getResource(clusterResource));
|
|
|
+ assertEquals(BuilderUtils.newResource(1024, 0),
|
|
|
+ parseResourceConfigValue("50% memory, 0% cpu").
|
|
|
+ getResource(clusterResource));
|
|
|
assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
- parseResourceConfigValue(" 1024 mb, 2 vcores "));
|
|
|
+ parseResourceConfigValue("50 % memory, 50 % cpu").
|
|
|
+ getResource(clusterResource));
|
|
|
assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
- parseResourceConfigValue(" 1024.3 mb, 2.35 vcores "));
|
|
|
+ parseResourceConfigValue("50%memory,50%cpu").
|
|
|
+ getResource(clusterResource));
|
|
|
assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
- parseResourceConfigValue(" 1024. mb, 2. vcores "));
|
|
|
+ parseResourceConfigValue(" 50 % memory, 50 % cpu ").
|
|
|
+ getResource(clusterResource));
|
|
|
+ assertEquals(BuilderUtils.newResource(1024, 2),
|
|
|
+ parseResourceConfigValue("50.% memory, 50.% cpu").
|
|
|
+ getResource(clusterResource));
|
|
|
+
|
|
|
+ clusterResource = BuilderUtils.newResource(1024 * 10, 4);
|
|
|
+ assertEquals(BuilderUtils.newResource((int)(1024 * 10 * 0.109), 2),
|
|
|
+ parseResourceConfigValue("10.9% memory, 50.6% cpu").
|
|
|
+ getResource(clusterResource));
|
|
|
}
|
|
|
|
|
|
@Test(expected = AllocationConfigurationException.class)
|
|
@@ -70,5 +102,24 @@ public class TestFairSchedulerConfiguration {
|
|
|
public void testGibberish() throws Exception {
|
|
|
parseResourceConfigValue("1o24vc0res");
|
|
|
}
|
|
|
-
|
|
|
+
|
|
|
+ @Test(expected = AllocationConfigurationException.class)
|
|
|
+ public void testNoUnitsPercentage() throws Exception {
|
|
|
+ parseResourceConfigValue("95%, 50% memory");
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = AllocationConfigurationException.class)
|
|
|
+ public void testInvalidNumPercentage() throws Exception {
|
|
|
+ parseResourceConfigValue("95A% cpu, 50% memory");
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = AllocationConfigurationException.class)
|
|
|
+ public void testCpuPercentageMemoryAbsolute() throws Exception {
|
|
|
+ parseResourceConfigValue("50% cpu, 1024 mb");
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = AllocationConfigurationException.class)
|
|
|
+ public void testMemoryPercentageCpuAbsolute() throws Exception {
|
|
|
+ parseResourceConfigValue("50% memory, 2 vcores");
|
|
|
+ }
|
|
|
}
|