doReturn("test").when(queue).getQueueName(); AppSchedulingInfo appSchedulingInfo = new AppSchedulingInfo(appAttemptId, "test", queue, null, 0, new ResourceUsage(), new HashMap<String, String>(), null);
public AbstractCSQueue(CapacitySchedulerContext cs, CapacitySchedulerConfiguration configuration, String queueName, CSQueue parent, CSQueue old) { this.labelManager = cs.getRMContext().getNodeLabelManager(); this.parent = parent; this.queueName = queueName; this.queuePath = ((parent == null) ? "" : (parent.getQueuePath() + ".")) + this.queueName; this.resourceCalculator = cs.getResourceCalculator(); this.activitiesManager = cs.getActivitiesManager(); // must be called after parent and queueName is set this.metrics = old != null ? (CSQueueMetrics) old.getMetrics() : CSQueueMetrics.forQueue(getQueuePath(), parent, cs.getConfiguration().getEnableUserMetrics(), cs.getConf()); this.csContext = cs; this.minimumAllocation = csContext.getMinimumResourceCapability(); // initialize ResourceUsage queueUsage = new ResourceUsage(); queueEntity = new PrivilegedEntity(EntityType.QUEUE, getQueuePath()); // initialize QueueCapacities queueCapacities = new QueueCapacities(parent == null); // initialize queueResourceQuotas queueResourceQuotas = new QueueResourceQuotas(); ReentrantReadWriteLock lock = new ReentrantReadWriteLock(); readLock = lock.readLock(); writeLock = lock.writeLock(); }
ResourceUsage ru = new ResourceUsage(); ru.setPending(pending[i]); ru.setUsed(used[i]);
Deque<ParentQueue> pqs = new LinkedList<ParentQueue>(); ParentQueue root = mockParentQueue(null, queues[0], pqs); ResourceUsage resUsage = new ResourceUsage(); resUsage.setUsed(used[0]); when(root.getQueueName()).thenReturn("/"); if (queues[i] > 0) { q = mockParentQueue(p, queues[i], pqs); ResourceUsage resUsagePerQueue = new ResourceUsage(); resUsagePerQueue.setUsed(used[i]); when(q.getQueueResourceUsage()).thenReturn(resUsagePerQueue);
ResourceUsage ru = Mockito.spy(new ResourceUsage()); ru.setUsed(label, used); when(ru.getCachedUsed(anyString())).thenReturn(used);
usage = new ResourceUsage(); userResourceUsage.put(app.getUser(), usage);
ResourceUsage ru = new ResourceUsage(); ru.setPending(pending[i]); ru.setUsed(used[i]);
ResourceUsage ru = new ResourceUsage(); QueueResourceQuotas qr = new QueueResourceQuotas();
public AbstractCSQueue(CapacitySchedulerContext cs, String queueName, CSQueue parent, CSQueue old) throws IOException { this.labelManager = cs.getRMContext().getNodeLabelManager(); this.parent = parent; this.queueName = queueName; this.resourceCalculator = cs.getResourceCalculator(); // must be called after parent and queueName is set this.metrics = old != null ? old.getMetrics() : QueueMetrics.forQueue(getQueuePath(), parent, cs.getConfiguration().getEnableUserMetrics(), cs.getConf()); this.csContext = cs; this.minimumAllocation = csContext.getMinimumResourceCapability(); // initialize ResourceUsage queueUsage = new ResourceUsage(); queueEntity = new PrivilegedEntity(EntityType.QUEUE, getQueuePath()); // initialize QueueCapacities queueCapacities = new QueueCapacities(parent == null); }
AppSchedulingInfo info = new AppSchedulingInfo( appAttemptId, "test", queue, mock(ActiveUsersManager.class), 0, new ResourceUsage(), new HashMap<>(), mock(RMContext.class)); Assert.assertEquals(0, info.getSchedulerKeys().size());
public AbstractCSQueue(CapacitySchedulerContext cs, String queueName, CSQueue parent, CSQueue old) throws IOException { this.labelManager = cs.getRMContext().getNodeLabelManager(); this.parent = parent; this.queueName = queueName; this.resourceCalculator = cs.getResourceCalculator(); // must be called after parent and queueName is set this.metrics = old != null ? old.getMetrics() : QueueMetrics.forQueue(getQueuePath(), parent, cs.getConfiguration().getEnableUserMetrics(), cs.getConf()); this.csContext = cs; this.minimumAllocation = csContext.getMinimumResourceCapability(); // initialize ResourceUsage queueUsage = new ResourceUsage(); queueEntity = new PrivilegedEntity(EntityType.QUEUE, getQueuePath()); // initialize QueueCapacities queueCapacities = new QueueCapacities(parent == null); }
Deque<ParentQueue> pqs = new LinkedList<ParentQueue>(); ParentQueue root = mockParentQueue(null, queues[0], pqs); ResourceUsage resUsage = new ResourceUsage(); resUsage.setUsed(used[0]); resUsage.setReserved(reserved[0]); if (queues[i] > 0) { q = mockParentQueue(p, queues[i], pqs); ResourceUsage resUsagePerQueue = new ResourceUsage(); resUsagePerQueue.setUsed(used[i]); resUsagePerQueue.setReserved(reserved[i]);
private void internalTestModifyAndRead(String label) throws Exception { ResourceUsage usage = new ResourceUsage(); Resource res; // First get returns 0 always res = get(usage, suffix, label); check(0, 0, res); // Add 1,1 should returns 1,1 inc(usage, suffix, Resource.newInstance(1, 1), label); check(1, 1, get(usage, suffix, label)); // Set 2,2 set(usage, suffix, Resource.newInstance(2, 2), label); check(2, 2, get(usage, suffix, label)); // dec 2,2 dec(usage, suffix, Resource.newInstance(2, 2), label); check(0, 0, get(usage, suffix, label)); }
private void internalTestModifyAndRead(String label) throws Exception { ResourceUsage usage = new ResourceUsage(); Resource res; // First get returns 0 always res = get(usage, suffix, label); check(0, 0, res); // Add 1,1 should returns 1,1 try { inc(usage, suffix, Resource.newInstance(1, 1), label); check(1, 1, get(usage, suffix, label)); } catch (NoSuchMethodException e) { // Few operations need not have to be verified as some resources doesn't // inc/dec apis exposed (For Eg: CachedUsed and CachedPending). } // Set 2,2 set(usage, suffix, Resource.newInstance(2, 2), label); check(2, 2, get(usage, suffix, label)); // dec 2,2 try { dec(usage, suffix, Resource.newInstance(2, 2), label); check(0, 0, get(usage, suffix, label)); } catch (NoSuchMethodException e) { // Few operations need not have to be verified, as some resources doesn't // inc/dec apis exposed (For Eg: CachedUsed and CachedPending). } }