Search in sources :

Example 1 with AclSelector

use of org.apache.drill.exec.resourcemgr.config.selectors.AclSelector in project drill by apache.

the class TestRMConfigLoad method testDefaultRMConfig.

@Test
public void testDefaultRMConfig() throws Exception {
    ClusterFixtureBuilder fixtureBuilder = ClusterFixture.builder(dirTestWatcher).configProperty(ExecConstants.RM_ENABLED, true).configProperty(ExecConstants.DRILL_PORT_HUNT, true).withLocalZk();
    try (ClusterFixture cluster = fixtureBuilder.build()) {
        ResourceManager resourceManager = cluster.drillbit().getContext().getResourceManager();
        assertTrue(resourceManager instanceof DistributedResourceManager);
        ResourcePoolTree poolTree = ((DistributedResourceManager) resourceManager).getRmPoolTree();
        assertTrue("In drill-rm-default root pool is not leaf pool", poolTree.getRootPool().isLeafPool());
        assertTrue("selector in drill-rm-default is not acl selector", poolTree.getRootPool().getSelector() instanceof AclSelector);
        assertEquals("max_query_memory_per_node in drill-rm-default is not configured with expected default value", 8 * 1024L, poolTree.getRootPool().getMaxQueryMemoryPerNode());
        assertEquals("queue_selection_policy in drill-rm-default is not configured with expected default value", RMCommonDefaults.ROOT_POOL_DEFAULT_QUEUE_SELECTION_POLICY, poolTree.getSelectionPolicyInUse().getSelectionPolicy());
        assertEquals("memory share of root pool in drill-rm-default is not configured with expected default value", RMCommonDefaults.ROOT_POOL_DEFAULT_MEMORY_PERCENT, poolTree.getResourceShare(), 0);
        final QueryQueueConfig defaultQueue = poolTree.getRootPool().getQueryQueue();
        assertEquals("max_admissible in drill-rm-default is not configured with expected default value", RMCommonDefaults.MAX_ADMISSIBLE_QUERY_COUNT, defaultQueue.getMaxAdmissibleQueries());
        assertEquals("max_waiting in drill-rm-default is not configured with expected default value", RMCommonDefaults.MAX_WAITING_QUERY_COUNT, defaultQueue.getMaxWaitingQueries());
        assertEquals("max_wait_timeout in drill-rm-default is not configured with expected default value", RMCommonDefaults.MAX_WAIT_TIMEOUT_IN_MS, defaultQueue.getWaitTimeoutInMs());
        assertEquals("wait_for_preferred_nodes in drill-rm-default is not configured with expected default value", RMCommonDefaults.WAIT_FOR_PREFERRED_NODES, defaultQueue.waitForPreferredNodes());
    }
}
Also used : ClusterFixture(org.apache.drill.test.ClusterFixture) QueryQueueConfig(org.apache.drill.exec.resourcemgr.config.QueryQueueConfig) ResourcePoolTree(org.apache.drill.exec.resourcemgr.config.ResourcePoolTree) ClusterFixtureBuilder(org.apache.drill.test.ClusterFixtureBuilder) DefaultResourceManager(org.apache.drill.exec.work.foreman.rm.DefaultResourceManager) DistributedResourceManager(org.apache.drill.exec.work.foreman.rm.DistributedResourceManager) ResourceManager(org.apache.drill.exec.work.foreman.rm.ResourceManager) DistributedResourceManager(org.apache.drill.exec.work.foreman.rm.DistributedResourceManager) AclSelector(org.apache.drill.exec.resourcemgr.config.selectors.AclSelector) Test(org.junit.Test) ResourceManagerTest(org.apache.drill.categories.ResourceManagerTest) DrillTest(org.apache.drill.test.DrillTest)

Aggregations

ResourceManagerTest (org.apache.drill.categories.ResourceManagerTest)1 QueryQueueConfig (org.apache.drill.exec.resourcemgr.config.QueryQueueConfig)1 ResourcePoolTree (org.apache.drill.exec.resourcemgr.config.ResourcePoolTree)1 AclSelector (org.apache.drill.exec.resourcemgr.config.selectors.AclSelector)1 DefaultResourceManager (org.apache.drill.exec.work.foreman.rm.DefaultResourceManager)1 DistributedResourceManager (org.apache.drill.exec.work.foreman.rm.DistributedResourceManager)1 ResourceManager (org.apache.drill.exec.work.foreman.rm.ResourceManager)1 ClusterFixture (org.apache.drill.test.ClusterFixture)1 ClusterFixtureBuilder (org.apache.drill.test.ClusterFixtureBuilder)1 DrillTest (org.apache.drill.test.DrillTest)1 Test (org.junit.Test)1