org.jolokia.jvmagent.spring.SpringJolokiaServerTest.java Source code

Introduction

Here is the source code for org.jolokia.jvmagent.spring.SpringJolokiaServerTest.java

Source

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;
    }

}