ParentQueue buildMockRootQueue(Random r, String[][] resData, int[]... queueData) { Resource[] abs = parseResourceDetails(resData[0]); Resource[] used = parseResourceDetails(resData[1]); Resource[] pending = parseResourceDetails(resData[2]); Resource[] reserved = parseResourceDetails(resData[3]); Resource[] gran = parseResourceDetails(resData[4]); int[] maxCap = queueData[0]; int[] apps = queueData[1]; int[] queues = queueData[2]; return mockNested(abs, maxCap, used, pending, reserved, apps, gran, queues); }
private ParentQueue buildMockRootQueue(Random rand2, String[][] resData, int[][] queueData) { Resource[] abs = parseResourceDetails(resData[0]); Resource[] used = parseResourceDetails(resData[1]); Resource[] pending = parseResourceDetails(resData[2]); Resource[] reserved = parseResourceDetails(resData[3]); Resource[] gran = parseResourceDetails(resData[4]); int[] maxCap = queueData[0]; int[] apps = queueData[1]; int[] queues = queueData[2]; return mockNested(abs, maxCap, used, pending, reserved, apps, gran, queues); }
ProportionalCapacityPreemptionPolicy buildPolicy(int[][] qData, String[][] resData, boolean useDominantResourceCalculator) { if (useDominantResourceCalculator) { when(mCS.getResourceCalculator()).thenReturn( new DominantResourceCalculator()); } ProportionalCapacityPreemptionPolicy policy = new ProportionalCapacityPreemptionPolicy(rmContext, mCS, mClock); clusterResources = leafAbsCapacities(parseResourceDetails(resData[0]), qData[2]); ParentQueue mRoot = buildMockRootQueue(rand, resData, qData); when(mCS.getRootQueue()).thenReturn(mRoot); setResourceAndNodeDetails(); return policy; }
ProportionalCapacityPreemptionPolicy buildPolicy(int[][] qData, String[][] resData, boolean useDominantResourceCalculator) { if (useDominantResourceCalculator) { when(mCS.getResourceCalculator()).thenReturn( new DominantResourceCalculator()); } ProportionalCapacityPreemptionPolicy policy = new ProportionalCapacityPreemptionPolicy( conf, rmContext, mCS, mClock); clusterResources = leafAbsCapacities(parseResourceDetails(resData[0]), qData[2]); ParentQueue mRoot = buildMockRootQueue(rand, resData, qData); when(mCS.getRootQueue()).thenReturn(mRoot); setResourceAndNodeDetails(); return policy; }