RMProxyUserInterfaceTest.java

/*
 * ProActive Parallel Suite(TM):
 * The Open Source library for parallel and distributed
 * Workflows & Scheduling, Orchestration, Cloud Automation
 * and Big Data Analysis on Enterprise Grids & Clouds.
 *
 * Copyright (c) 2007 - 2017 ActiveEon
 * Contact: contact@activeeon.com
 *
 * This library is free software: you can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public License
 * as published by the Free Software Foundation: version 3 of
 * the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 *
 * If needed, contact us to obtain a release under GPL Version 2 or 3
 * or a different license than the AGPL.
 */
package functionaltests.jmx;

import static org.junit.Assert.assertNotNull;

import java.security.PublicKey;
import java.util.Collections;

import org.junit.After;
import org.junit.Test;
import org.objectweb.proactive.api.PAActiveObject;
import org.ow2.proactive.authentication.crypto.CredData;
import org.ow2.proactive.authentication.crypto.Credentials;
import org.ow2.proactive.resourcemanager.authentication.RMAuthentication;
import org.ow2.proactive.resourcemanager.common.event.RMInitialState;
import org.ow2.proactive.resourcemanager.common.util.RMListenerProxy;

import functionaltests.monitor.RMMonitorEventReceiver;
import functionaltests.utils.RMFunctionalTest;
import functionaltests.utils.RMTHelper;
import functionaltests.utils.TestUsers;


public class RMProxyUserInterfaceTest extends RMFunctionalTest {

    private static final String EXISTING_OBJECT_NAME = "java.lang:type=Runtime";

    private RMListenerProxy proxyUserInterface;

    // SCHEDULING-1850
    @Test
    public void testGetNodeMBeanInfo_DisconnectionOfRemovedNodeSource() throws Exception {
        proxyUserInterface = createRMCachingProxyUserInterface();

        rmHelper.createNodeSource("NodeSource1", 1);

        RMInitialState state = ((RMMonitorEventReceiver) rmHelper.getResourceManager()).getInitialState();
        String nodeSource1_NodeJmxUrl = state.getNodesEvents().get(0).getDefaultJMXUrl();

        Object mBeanFromNodeSource1 = proxyUserInterface.getNodeMBeanInfo(nodeSource1_NodeJmxUrl,
                                                                          EXISTING_OBJECT_NAME,
                                                                          Collections.<String> emptyList());

        assertNotNull(mBeanFromNodeSource1);

        // remove nodesource and try another jmx connection
        rmHelper.getResourceManager().removeNodeSource("NodeSource1", true);

        rmHelper.createNodeSource("NodeSource2", 1);

        state = ((RMMonitorEventReceiver) rmHelper.getResourceManager()).getInitialState();
        String nodeSource2_NodeJmxUrl = state.getNodesEvents().get(0).getDefaultJMXUrl();

        Object mBeanFromNodeSource2 = proxyUserInterface.getNodeMBeanInfo(nodeSource2_NodeJmxUrl,
                                                                          EXISTING_OBJECT_NAME,
                                                                          Collections.<String> emptyList());

        assertNotNull(mBeanFromNodeSource2);

        // cleanup
        rmHelper.getResourceManager().removeNodeSource("NodeSource2", true);
    }

    @After
    public void cleanUp() {
        if (proxyUserInterface != null) {
            try {
                proxyUserInterface.disconnect();
            } catch (Exception ignored) {

            }
            PAActiveObject.terminateActiveObject(proxyUserInterface, true);
        }
    }

    private RMListenerProxy createRMCachingProxyUserInterface() throws Exception {
        RMListenerProxy proxyUserInterface = PAActiveObject.newActive(RMListenerProxy.class, new Object[] {});
        final RMAuthentication auth = rmHelper.getRMAuth();
        final PublicKey pubKey = auth.getPublicKey();
        final Credentials adminCreds = Credentials.createCredentials(new CredData(TestUsers.TEST.username,
                                                                                  TestUsers.TEST.password),
                                                                     pubKey);
        proxyUserInterface.init(RMTHelper.getLocalUrl(), adminCreds);
        return proxyUserInterface;
    }
}