Examples with RMCore org.ow2.proactive.resourcemanager.core.RMCore used on opensource projects

Example 1 with RMCore

use of org.ow2.proactive.resourcemanager.core.RMCore in project scheduling by ow2-proactive.

the class SelectionManagerTest method newMockedRMCore.

public static RMCore newMockedRMCore(int nbNodes) {
    RMCore mockedRMCore = Mockito.mock(RMCore.class);
    TopologyManager mockedTopologyManager = Mockito.mock(TopologyManager.class);
    when(mockedTopologyManager.getHandler(Matchers.any(TopologyDescriptor.class))).thenReturn(selectAllTopology());
    RMCore.topologyManager = mockedTopologyManager;
    if (nbNodes > 0) {
        ArrayList<RMNode> freeNodes = new ArrayList<RMNode>(nbNodes);
        for (int i = 0; i < nbNodes; i++) {
            freeNodes.add(createMockeNode("user", "mocked-node-" + (i + 1), "mocked-node-" + (i + 1)));
        }
        when(mockedRMCore.getFreeNodes()).thenReturn(freeNodes);
    }
    return mockedRMCore;
}

Example 2 with RMCore

use of org.ow2.proactive.resourcemanager.core.RMCore in project scheduling by ow2-proactive.

the class SelectionManagerTest method testSelectNodesWithNoNodes.

@Test
public void testSelectNodesWithNoNodes() {
    RMCore rmCore = newMockedRMCore(0);
    SelectionManager selectionManager = createSelectionManager(rmCore);
    Criteria crit = new Criteria(1);
    crit.setTopology(TopologyDescriptor.ARBITRARY);
    crit.setScripts(null);
    crit.setBlackList(null);
    crit.setBestEffort(false);
    NodeSet nodeSet = selectionManager.selectNodes(crit, null);
    assertEquals(0, nodeSet.size());
}

Example 3 with RMCore

use of org.ow2.proactive.resourcemanager.core.RMCore in project scheduling by ow2-proactive.

the class SelectionManagerTest method testRunScriptsWillNotBeCalled.

@Test
public void testRunScriptsWillNotBeCalled() {
    RMCore rmCore = newMockedRMCore(2);
    SelectionManager selectionManager = createSelectionManager(rmCore);
    Criteria crit = new Criteria(2);
    crit.setTopology(TopologyDescriptor.SINGLE_HOST);
    SelectionScript selectWhatever = new SelectionScript();
    crit.setScripts(Lists.newArrayList(selectWhatever));
    crit.setBlackList(null);
    crit.setBestEffort(false);
    Client mockedClient = mock(Client.class);
    selectionManager.selectNodes(crit, mockedClient);
    verify(rmCore, never()).setBusyNode(anyString(), any(Client.class));
}

Example 4 with RMCore

use of org.ow2.proactive.resourcemanager.core.RMCore in project scheduling by ow2-proactive.

the class ProbabilisticSelectionManagerTest method testSelectNodesWithNoNodes.

@Test
public void testSelectNodesWithNoNodes() {
    RMCore rmCore = SelectionManagerTest.newMockedRMCore();
    SelectionManager selectionManager = new ProbablisticSelectionManager(rmCore);
    Criteria crit = new Criteria(1);
    crit.setTopology(TopologyDescriptor.ARBITRARY);
    crit.setScripts(null);
    crit.setBlackList(null);
    crit.setBestEffort(false);
    NodeSet nodeSet = selectionManager.selectNodes(crit, null);
    assertEquals(0, nodeSet.size());
}

Example 5 with RMCore

use of org.ow2.proactive.resourcemanager.core.RMCore in project scheduling by ow2-proactive.

the class RMCore method initActivity.

/**
 * Initialization part of the RMCore active object.
 * Create RM's active objects and the default static Node Source named
 * {@link RMConstants#DEFAULT_STATIC_SOURCE_NAME}. Finally, it throws the RM
 * started event.
 *
 * @param body the active object's body.
 */
public void initActivity(Body body) {
    if (logger.isDebugEnabled()) {
        logger.debug("RMCore start : initActivity");
    }
    try {
        // setting up the policy
        logger.debug("Setting up the resource manager security policy");
        ClientsPolicy.init();
        StubObject rmCoreStub = PAActiveObject.getStubOnThis();
        PAActiveObject.registerByName(rmCoreStub, RMConstants.NAME_ACTIVE_OBJECT_RMCORE);
        dbManager = RMDBManager.getInstance();
        if (logger.isDebugEnabled()) {
            logger.debug("Creating RMAuthentication active object");
        }
        authentication = (RMAuthenticationImpl) PAActiveObject.newActive(RMAuthenticationImpl.class.getName(), new Object[] { rmCoreStub }, nodeRM);
        if (logger.isDebugEnabled()) {
            logger.debug("Creating RMMonitoring active object");
        }
        // Boot the JMX infrastructure
        this.jmxHelper.boot(authentication);
        monitoring = (RMMonitoringImpl) PAActiveObject.newActive(RMMonitoringImpl.class.getName(), new Object[] { rmCoreStub }, nodeRM);
        if (logger.isDebugEnabled()) {
            logger.debug("Creating SelectionManager active object");
        }
        selectionManager = (SelectionManager) PAActiveObject.newActive(ProbablisticSelectionManager.class.getName(), new Object[] { rmCoreStub }, nodeRM);
        if (logger.isDebugEnabled()) {
            logger.debug("Creating ClientPinger active object");
        }
        clientPinger = (ClientPinger) PAActiveObject.newActive(ClientPinger.class.getName(), new Object[] { rmCoreStub }, nodeRM);
        if (logger.isDebugEnabled()) {
            logger.debug("Creating NodeCleaner active object");
        }
        nodesCleaner = (NodesCleaner) PAActiveObject.newActive(NodesCleaner.class.getName(), new Object[] { rmCoreStub }, nodeRM);
        topologyManager = new TopologyManager();
        nodeConfigurator = (RMNodeConfigurator) PAActiveObject.newActive(RMNodeConfigurator.class.getName(), new Object[] { rmCoreStub }, nodeRM);
        // adding shutdown hook
        final RMCore rmcoreStub = (RMCore) rmCoreStub;
        Runtime.getRuntime().addShutdownHook(new Thread() {

            @Override
            public void run() {
                if (!toShutDown) {
                    rmcoreStub.shutdown(true);
                }
                synchronized (nodeRM) {
                    if (!shutedDown) {
                        try {
                            // wait for rmcore shutdown (5 min at most)
                            nodeRM.wait(5 * 60 * 60 * 1000);
                        } catch (InterruptedException e) {
                            logger.warn("shutdown hook interrupted", e);
                        }
                    }
                }
            }
        });
        // Creating RM started event
        this.monitoring.rmEvent(new RMEvent(RMEventType.STARTED));
        authentication.setActivated(true);
        clientPinger.ping();
        initiateRecoveryIfRequired();
    } catch (ActiveObjectCreationException e) {
        logger.error("", e);
    } catch (NodeException e) {
        logger.error("", e);
    } catch (ProActiveException e) {
        logger.error("", e);
    } catch (ClassNotFoundException e) {
        logger.error("", e);
    } finally {
        signalRMCoreIsInitialized();
    }
    if (logger.isDebugEnabled()) {
        logger.debug("RMCore end: initActivity");
    }
}