@Override public QueueMetrics load(String tenantId) { int numQueued = Iterators.size(taskQueues.getQueued(tenantId)); int numInProgress = Iterators.size(taskQueues.getBeingConsumed(tenantId)); return new QueueMetrics(numQueued, numInProgress); } });
expected.put("tenantA", new QueueMetrics(2, 1)); expected.put("tenantB", new QueueMetrics(1, 0)); expected.put("tenantC", new QueueMetrics(0, 1)); expected.put("tenantD", new QueueMetrics(1, 1)); expected.put(TENANT, new QueueMetrics(0, 0)); expected.put(Constants.SUPERADMIN_TENANT, new QueueMetrics(0, 0)); assertQueueMetrics("tenantA", ImmutableMap.of("tenantA", new QueueMetrics(2, 1))); assertQueueMetrics("tenantB", ImmutableMap.of("tenantB", new QueueMetrics(1, 0))); assertQueueMetrics("tenantC", ImmutableMap.of("tenantC", new QueueMetrics(0, 1))); assertQueueMetrics("tenantD", ImmutableMap.of("tenantD", new QueueMetrics(1, 1))); } finally { provisionerQueues.removeAll();
expected.put("tenant1", new QueueMetrics(2, 1)); expected.put("tenant2", new QueueMetrics(1, 0)); expected.put("tenant3", new QueueMetrics(0, 1)); expected.put("tenant4", new QueueMetrics(1, 1)); expected.put(Constants.SUPERADMIN_TENANT, new QueueMetrics(0, 0)); Assert.assertEquals(expected.get("tenant3"), service.getTaskQueueMetricsSnapshot("id3")); Assert.assertEquals(expected.get("tenant4"), service.getTaskQueueMetricsSnapshot("id4")); Assert.assertEquals(new QueueMetrics(0, 0), service.getTaskQueueMetricsSnapshot("non-id")); Assert.assertEquals(expected, service.getTaskQueueMetricsSnapshot());