SpringJolokiaServerTest.java

package org.jolokia.jvmagent.spring;

/*
 * 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.io.IOException;
import java.util.HashMap;
import java.util.Map;

import org.jolokia.jvmagent.JolokiaServerConfig;
import org.jolokia.test.util.EnvTestUtil;
import org.springframework.context.ApplicationContext;
import org.testng.annotations.Test;

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

/**
 * @author roland
 * @since 29.12.12
 */
public class SpringJolokiaServerTest extends BaseServerTest {

    @Test
    public void withoutStart() throws Exception {
        SpringJolokiaAgent server = new SpringJolokiaAgent();
        server.setConfig(getConfig(false,100));
        server.afterPropertiesSet();
        server.start();
        checkServerAndStop(server);
    }

    @Test
    public void systemProperties() throws Exception {
        checkSystemPropertyMode("fallback","/jol/","/j4p/","/j4p/");
        checkSystemPropertyMode("fallback","/jol/",null,"/jol/");
        checkSystemPropertyMode("fallback",null,null,"/jolokia/");

        checkSystemPropertyMode("override","/jol/","/j4p/","/jol/");
        checkSystemPropertyMode("override","/jol/",null,"/jol/");
        checkSystemPropertyMode("override",null,null,"/jolokia/");

        checkSystemPropertyMode(null,"/jol/",null,"/jolokia/");
    }

    private void checkSystemPropertyMode(String mode,String propContext,String configContext,String expectContext) throws IOException, InterruptedException {
        if (propContext != null) {
            System.setProperty("jolokia.agentContext", propContext);
        }
        SpringJolokiaAgent server = new SpringJolokiaAgent();
        server.setConfig(getConfig(true,100,"agentContext", configContext));
        server.setSystemPropertiesMode(mode);
        server.afterPropertiesSet();
        assertEquals(server.getServerConfig().getContextPath(), expectContext);
        System.getProperties().remove("jolokia.agentContext");
        server.stop();
        // Allow to shutdown server ...
        Thread.sleep(500);
    }

    @Test
    public void withStart() throws Exception {
        SpringJolokiaAgent server = new SpringJolokiaAgent();
        server.setConfig(getConfig(true,100));
        server.afterPropertiesSet();
        checkServerAndStop(server);
    }

    @Test
    public void withMultiConfigAndStart() throws Exception {
        SpringJolokiaAgent server = new SpringJolokiaAgent();
        server.setLookupConfig(true);
        server.setConfig(getConfig(true,100));

        ApplicationContext ctx = createMock(ApplicationContext.class);
        Map<String,SpringJolokiaConfigHolder> configs = new HashMap<String, SpringJolokiaConfigHolder>();
        configs.put("B",getConfig(false,10,"executor","single","agentContext","/j4p/"));
        configs.put("A", getConfig(true, 20, "executor", "fixed", "threadNr", "2"));
        expect(ctx.getBeansOfType(SpringJolokiaConfigHolder.class)).andReturn(configs);
        replay(ctx);
        server.setApplicationContext(ctx);
        server.afterPropertiesSet();
        JolokiaServerConfig cfg = server.getServerConfig();
        assertEquals(cfg.getExecutor(),"fixed");
        assertEquals(cfg.getThreadNr(),2);
        assertEquals(cfg.getContextPath(),"/j4p/");
        checkServerAndStop(server);
    }

    private SpringJolokiaConfigHolder getConfig(boolean autoStart, int order, String ... extraArgs) throws IOException {
        SpringJolokiaConfigHolder cfg = new SpringJolokiaConfigHolder();
        cfg.setOrder(order);
        Map<String, String> map = new HashMap<String, String>();
        map.put("autoStart","" + autoStart);
        map.put("port", EnvTestUtil.getFreePort() + "");
        map.put("host","127.0.0.1");
        for (int i = 0; i < extraArgs.length; i+=2) {
            if (extraArgs[i+1] != null) {
                map.put(extraArgs[i],extraArgs[i+1]);
            }
        }
        cfg.setConfig(map);
        return cfg;
    }

}