org.jolokia.detector.ServerHandleTest.java Source code

Introduction

Here is the source code for org.jolokia.detector.ServerHandleTest.java

Source

package org.jolokia.detector;

/*
 * Copyright 2009-2013 Roland Huss
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import java.net.MalformedURLException;
import java.util.HashMap;
import java.util.Map;

import javax.management.*;

import org.easymock.EasyMock;
import org.jolokia.backend.Config;
import org.jolokia.config.ConfigKey;
import org.jolokia.config.Configuration;
import org.jolokia.util.LogHandler;
import org.json.simple.JSONObject;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import static org.easymock.EasyMock.*;
import static org.testng.Assert.*;

/**
 * @author roland
 * @since 06.06.12
 */
public class ServerHandleTest {

    private ServerHandle serverHandle;
    private String vendor;
    private String product;
    private String version;
    private Map<String, String> extraInfo;

    @BeforeMethod
    public void setup() throws MalformedURLException {
        extraInfo = new HashMap<String, String>();
        extraInfo.put("extra1", "value1");
        vendor = "acim";
        product = "dukeNukem";
        version = "forEver";
        serverHandle = new ServerHandle(vendor, product, version, extraInfo);
    }

    @Test
    public void basics() throws MalformedURLException {
        assertEquals(serverHandle.getProduct(), product);
        assertEquals(serverHandle.getVendor(), vendor);
        assertEquals(serverHandle.getExtraInfo(null).get("extra1"), "value1");
        assertEquals(serverHandle.getVersion(), version);
    }

    @Test
    public void toJson() {
        JSONObject json = serverHandle.toJSONObject(null);
        assertEquals(json.get("vendor"), vendor);
        assertEquals(json.get("product"), product);
        assertEquals(json.get("version"), version);
        assertEquals(((JSONObject) json.get("extraInfo")).get("extra1"), "value1");
    }

    @Test
    public void allNull() {
        ServerHandle handle = new ServerHandle(null, null, null, null);
        assertNull(handle.getVendor());
        assertNull(handle.toJSONObject(null).get("extraInfo"));
    }

    @Test
    public void detectorOptions() {
        Configuration opts = new Configuration(ConfigKey.DETECTOR_OPTIONS, "{\"dukeNukem\" : {\"doIt\" : true }}");
        JSONObject config = serverHandle.getDetectorOptions(opts, null);
        assertTrue((Boolean) config.get("doIt"));
    }

    @Test
    public void detectorOptionsEmpty() {
        JSONObject config = serverHandle.getDetectorOptions(new Configuration(), null);
        assertNull(config);
    }

    @Test
    public void detectOptionsFail() {
        LogHandler handler = EasyMock.createMock(LogHandler.class);
        handler.error(matches("^.*parse detector options.*"), isA(Exception.class));
        replay(handler);

        Configuration opts = new Configuration(ConfigKey.DETECTOR_OPTIONS, "blub: bla");
        JSONObject config = serverHandle.getDetectorOptions(opts, handler);
        assertNull(config);
        verify(handler);
    }

    @Test
    public void registerAtMBeanServer() throws MalformedObjectNameException, MBeanRegistrationException,
            InstanceAlreadyExistsException, NotCompliantMBeanException {
        Config config = new Config(null, null, null);
        ObjectName oName = new ObjectName("jolokia:type=Config");
        ObjectInstance oInstance = new ObjectInstance(oName, Config.class.getName());
        MBeanServer server = EasyMock.createMock(MBeanServer.class);
        expect(server.registerMBean(eq(config), eq(oName))).andReturn(oInstance);
        replay(server);

        ObjectName resName = serverHandle.registerMBeanAtServer(server, config, "jolokia:type=Config");
        assertEquals(resName, oName);

        verify(server);
    }

    @Test
    public void registerAtMBeanServer2() throws MalformedObjectNameException, MBeanRegistrationException,
            InstanceAlreadyExistsException, NotCompliantMBeanException {
        Config config = new Config(null, null, null);
        ObjectInstance oInstance = new ObjectInstance("jolokia:type=dummy", Config.class.getName());
        MBeanServer server = EasyMock.createMock(MBeanServer.class);
        expect(server.registerMBean(config, null)).andReturn(oInstance);
        replay(server);

        ObjectName resName = serverHandle.registerMBeanAtServer(server, config, null);
        assertEquals(resName, new ObjectName("jolokia:type=dummy"));

        verify(server);
    }
}