assertEquals(rootAInfo.getId(), expectedRootAInfo.getId()); assertEquals(rootAInfo.getState(), expectedRootAInfo.getState()); assertEquals(rootAInfo.getNumRunningQueries(), expectedRootAInfo.getNumRunningQueries()); assertEquals(rootAInfo.getNumQueuedQueries(), expectedRootAInfo.getNumQueuedQueries()); assertEquals(actualRootInfo.getId(), expectedRootInfo.getId()); assertEquals(actualRootInfo.getState(), expectedRootInfo.getState()); assertEquals(actualRootInfo.getNumRunningQueries(), expectedRootInfo.getNumRunningQueries()); assertEquals(actualRootInfo.getNumQueuedQueries(), expectedRootInfo.getNumQueuedQueries());
private static void assertGroupInfoEquals(ResourceGroupInfo actual, ResourceGroupInfo expected) { assertTrue(actual.getSchedulingWeight() == expected.getSchedulingWeight() && actual.getSoftConcurrencyLimit() == expected.getSoftConcurrencyLimit() && actual.getHardConcurrencyLimit() == expected.getHardConcurrencyLimit() && actual.getMaxQueuedQueries() == expected.getMaxQueuedQueries() && actual.getNumQueuedQueries() == expected.getNumQueuedQueries() && actual.getNumRunningQueries() == expected.getNumRunningQueries() && actual.getNumEligibleSubGroups() == expected.getNumEligibleSubGroups() && Objects.equals(actual.getId(), expected.getId()) && actual.getState() == expected.getState() && actual.getSchedulingPolicy() == expected.getSchedulingPolicy() && Objects.equals(actual.getSoftMemoryLimit(), expected.getSoftMemoryLimit()) && Objects.equals(actual.getMemoryUsage(), expected.getMemoryUsage())); } }
assertEquals(rootInfo.getSubGroups().get(0).getMaxQueuedQueries(), rootA.getMaxQueuedQueries()); assertEquals(rootInfo.getSubGroups().get(0).getNumEligibleSubGroups(), 2); assertEquals(rootInfo.getSubGroups().get(0).getNumRunningQueries(), 0); assertEquals(rootInfo.getSubGroups().get(0).getNumQueuedQueries(), 10); assertGroupInfoEquals(rootInfo.getSubGroups().get(1), rootB.getInfo()); assertEquals(rootInfo.getSubGroups().get(1).getMaxQueuedQueries(), rootB.getMaxQueuedQueries()); assertEquals(rootInfo.getSubGroups().get(1).getNumEligibleSubGroups(), 0); assertEquals(rootInfo.getSubGroups().get(1).getNumRunningQueries(), 1); assertEquals(rootInfo.getSubGroups().get(1).getNumQueuedQueries(), 9); assertEquals(rootInfo.getSoftConcurrencyLimit(), root.getSoftConcurrencyLimit());
assertEquals(info.getNumRunningQueries(), 0); assertEquals(info.getNumQueuedQueries(), 40); root.processQueuedQueries(); info = root.getInfo(); assertEquals(info.getNumRunningQueries(), 4); assertEquals(info.getNumQueuedQueries(), 36); assertEquals(info.getNumRunningQueries(), 4); assertEquals(info.getNumQueuedQueries(), 32); root.processQueuedQueries(); info = root.getInfo(); assertEquals(info.getNumRunningQueries(), 4); assertEquals(info.getNumQueuedQueries(), 32); root.processQueuedQueries(); info = root.getInfo(); assertEquals(info.getNumRunningQueries(), 10); assertEquals(info.getNumQueuedQueries(), 26);