|
@@ -112,37 +112,37 @@ public class TestSchedulerUtils {
|
|
|
|
|
|
// case negative memory
|
|
// case negative memory
|
|
ask.setCapability(Resources.createResource(-1024));
|
|
ask.setCapability(Resources.createResource(-1024));
|
|
- SchedulerUtils.normalizeRequest(ask, resourceCalculator, null, minResource,
|
|
|
|
|
|
+ SchedulerUtils.normalizeRequest(ask, resourceCalculator, minResource,
|
|
maxResource);
|
|
maxResource);
|
|
assertEquals(minMemory, ask.getCapability().getMemorySize());
|
|
assertEquals(minMemory, ask.getCapability().getMemorySize());
|
|
|
|
|
|
// case zero memory
|
|
// case zero memory
|
|
ask.setCapability(Resources.createResource(0));
|
|
ask.setCapability(Resources.createResource(0));
|
|
- SchedulerUtils.normalizeRequest(ask, resourceCalculator, null, minResource,
|
|
|
|
|
|
+ SchedulerUtils.normalizeRequest(ask, resourceCalculator, minResource,
|
|
maxResource);
|
|
maxResource);
|
|
assertEquals(minMemory, ask.getCapability().getMemorySize());
|
|
assertEquals(minMemory, ask.getCapability().getMemorySize());
|
|
|
|
|
|
// case memory is a multiple of minMemory
|
|
// case memory is a multiple of minMemory
|
|
ask.setCapability(Resources.createResource(2 * minMemory));
|
|
ask.setCapability(Resources.createResource(2 * minMemory));
|
|
- SchedulerUtils.normalizeRequest(ask, resourceCalculator, null, minResource,
|
|
|
|
|
|
+ SchedulerUtils.normalizeRequest(ask, resourceCalculator, minResource,
|
|
maxResource);
|
|
maxResource);
|
|
assertEquals(2 * minMemory, ask.getCapability().getMemorySize());
|
|
assertEquals(2 * minMemory, ask.getCapability().getMemorySize());
|
|
|
|
|
|
// case memory is not a multiple of minMemory
|
|
// case memory is not a multiple of minMemory
|
|
ask.setCapability(Resources.createResource(minMemory + 10));
|
|
ask.setCapability(Resources.createResource(minMemory + 10));
|
|
- SchedulerUtils.normalizeRequest(ask, resourceCalculator, null, minResource,
|
|
|
|
|
|
+ SchedulerUtils.normalizeRequest(ask, resourceCalculator, minResource,
|
|
maxResource);
|
|
maxResource);
|
|
assertEquals(2 * minMemory, ask.getCapability().getMemorySize());
|
|
assertEquals(2 * minMemory, ask.getCapability().getMemorySize());
|
|
|
|
|
|
// case memory is equal to max allowed
|
|
// case memory is equal to max allowed
|
|
ask.setCapability(Resources.createResource(maxMemory));
|
|
ask.setCapability(Resources.createResource(maxMemory));
|
|
- SchedulerUtils.normalizeRequest(ask, resourceCalculator, null, minResource,
|
|
|
|
|
|
+ SchedulerUtils.normalizeRequest(ask, resourceCalculator, minResource,
|
|
maxResource);
|
|
maxResource);
|
|
assertEquals(maxMemory, ask.getCapability().getMemorySize());
|
|
assertEquals(maxMemory, ask.getCapability().getMemorySize());
|
|
|
|
|
|
// case memory is just less than max
|
|
// case memory is just less than max
|
|
ask.setCapability(Resources.createResource(maxMemory - 10));
|
|
ask.setCapability(Resources.createResource(maxMemory - 10));
|
|
- SchedulerUtils.normalizeRequest(ask, resourceCalculator, null, minResource,
|
|
|
|
|
|
+ SchedulerUtils.normalizeRequest(ask, resourceCalculator, minResource,
|
|
maxResource);
|
|
maxResource);
|
|
assertEquals(maxMemory, ask.getCapability().getMemorySize());
|
|
assertEquals(maxMemory, ask.getCapability().getMemorySize());
|
|
|
|
|
|
@@ -150,14 +150,14 @@ public class TestSchedulerUtils {
|
|
maxResource = Resources.createResource(maxMemory - 10, 0);
|
|
maxResource = Resources.createResource(maxMemory - 10, 0);
|
|
ask.setCapability(Resources.createResource(maxMemory - 100));
|
|
ask.setCapability(Resources.createResource(maxMemory - 100));
|
|
// multiple of minMemory > maxMemory, then reduce to maxMemory
|
|
// multiple of minMemory > maxMemory, then reduce to maxMemory
|
|
- SchedulerUtils.normalizeRequest(ask, resourceCalculator, null, minResource,
|
|
|
|
|
|
+ SchedulerUtils.normalizeRequest(ask, resourceCalculator, minResource,
|
|
maxResource);
|
|
maxResource);
|
|
assertEquals(maxResource.getMemorySize(), ask.getCapability().getMemorySize());
|
|
assertEquals(maxResource.getMemorySize(), ask.getCapability().getMemorySize());
|
|
|
|
|
|
// ask is more than max
|
|
// ask is more than max
|
|
maxResource = Resources.createResource(maxMemory, 0);
|
|
maxResource = Resources.createResource(maxMemory, 0);
|
|
ask.setCapability(Resources.createResource(maxMemory + 100));
|
|
ask.setCapability(Resources.createResource(maxMemory + 100));
|
|
- SchedulerUtils.normalizeRequest(ask, resourceCalculator, null, minResource,
|
|
|
|
|
|
+ SchedulerUtils.normalizeRequest(ask, resourceCalculator, minResource,
|
|
maxResource);
|
|
maxResource);
|
|
assertEquals(maxResource.getMemorySize(), ask.getCapability().getMemorySize());
|
|
assertEquals(maxResource.getMemorySize(), ask.getCapability().getMemorySize());
|
|
}
|
|
}
|
|
@@ -175,13 +175,13 @@ public class TestSchedulerUtils {
|
|
// case negative memory/vcores
|
|
// case negative memory/vcores
|
|
ask.setCapability(Resources.createResource(-1024, -1));
|
|
ask.setCapability(Resources.createResource(-1024, -1));
|
|
SchedulerUtils.normalizeRequest(
|
|
SchedulerUtils.normalizeRequest(
|
|
- ask, resourceCalculator, clusterResource, minResource, maxResource);
|
|
|
|
|
|
+ ask, resourceCalculator, minResource, maxResource);
|
|
assertEquals(minResource, ask.getCapability());
|
|
assertEquals(minResource, ask.getCapability());
|
|
|
|
|
|
// case zero memory/vcores
|
|
// case zero memory/vcores
|
|
ask.setCapability(Resources.createResource(0, 0));
|
|
ask.setCapability(Resources.createResource(0, 0));
|
|
SchedulerUtils.normalizeRequest(
|
|
SchedulerUtils.normalizeRequest(
|
|
- ask, resourceCalculator, clusterResource, minResource, maxResource);
|
|
|
|
|
|
+ ask, resourceCalculator, minResource, maxResource);
|
|
assertEquals(minResource, ask.getCapability());
|
|
assertEquals(minResource, ask.getCapability());
|
|
assertEquals(1, ask.getCapability().getVirtualCores());
|
|
assertEquals(1, ask.getCapability().getVirtualCores());
|
|
assertEquals(1024, ask.getCapability().getMemorySize());
|
|
assertEquals(1024, ask.getCapability().getMemorySize());
|
|
@@ -189,7 +189,7 @@ public class TestSchedulerUtils {
|
|
// case non-zero memory & zero cores
|
|
// case non-zero memory & zero cores
|
|
ask.setCapability(Resources.createResource(1536, 0));
|
|
ask.setCapability(Resources.createResource(1536, 0));
|
|
SchedulerUtils.normalizeRequest(
|
|
SchedulerUtils.normalizeRequest(
|
|
- ask, resourceCalculator, clusterResource, minResource, maxResource);
|
|
|
|
|
|
+ ask, resourceCalculator, minResource, maxResource);
|
|
assertEquals(Resources.createResource(2048, 1), ask.getCapability());
|
|
assertEquals(Resources.createResource(2048, 1), ask.getCapability());
|
|
assertEquals(1, ask.getCapability().getVirtualCores());
|
|
assertEquals(1, ask.getCapability().getVirtualCores());
|
|
assertEquals(2048, ask.getCapability().getMemorySize());
|
|
assertEquals(2048, ask.getCapability().getMemorySize());
|