org.ngrinder.agent.service.AgentManagerServiceTest.java Source code

Introduction

Here is the source code for org.ngrinder.agent.service.AgentManagerServiceTest.java

Source

/* 
 * 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. 
 */
package org.ngrinder.agent.service;

import junit.framework.Assert;
import net.grinder.message.console.AgentControllerState;
import org.apache.commons.io.FileUtils;
import org.apache.commons.lang.ArrayUtils;
import org.apache.commons.lang.mutable.MutableInt;
import org.junit.Test;
import org.ngrinder.AbstractNGrinderTransactionalTest;
import org.ngrinder.agent.repository.AgentManagerRepository;
import org.ngrinder.infra.config.Config;
import org.ngrinder.model.AgentInfo;
import org.springframework.beans.factory.annotation.Autowired;

import java.io.File;
import java.io.IOException;
import java.net.URISyntaxException;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.List;
import java.util.Map;

import static org.hamcrest.Matchers.is;
import static org.junit.Assert.assertThat;
import static org.ngrinder.common.util.TypeConvertUtils.cast;

/**
 * Agent service test.
 *
 * @author Tobi
 * @since 3.0
 */
public class AgentManagerServiceTest extends AbstractNGrinderTransactionalTest {

    @Autowired
    private AgentManagerService agentManagerService;

    @Autowired
    private AgentPackageService agentPackageService;

    @Autowired
    private AgentManagerRepository agentRepository;

    @Autowired
    private LocalAgentService localAgentService;

    @Autowired
    private Config config;

    @Test
    public void testSaveGetDeleteAgent() {
        AgentInfo agent = saveAgent("save");
        AgentInfo agent2 = agentManagerService.getOne(agent.getId());
        Assert.assertNotNull(agent2);

        List<AgentInfo> agentListDB = agentManagerService.getAllLocal();
        Assert.assertNotNull(agentListDB);

        agentManagerService.approve(agent.getId(), true);

        agentRepository.delete(agent.getId());
        agent2 = agentManagerService.getOne(agent.getId());
        Assert.assertNull(agent2);
    }

    private AgentInfo saveAgent(String key) {
        AgentInfo agent = new AgentInfo();
        agent.setIp("1.1.1.1");
        agent.setName("testAppName" + key);
        agent.setPort(8080);
        agent.setRegion("testRegion" + key);
        agent.setState(AgentControllerState.BUSY);
        agentRepository.save(agent);
        return agent;
    }

    @Test
    public void testGetUserAvailableAgentCount() {
        Map<String, MutableInt> countMap = agentManagerService.getAvailableAgentCountMap(getTestUser());
        String currRegion = config.getRegion();
        int oriCount = countMap.get(currRegion).intValue();

        AgentInfo agentInfo = new AgentInfo();
        agentInfo.setName("localhost");
        agentInfo.setRegion(config.getRegion());
        agentInfo.setIp("127.127.127.127");
        agentInfo.setPort(1);
        agentInfo.setState(AgentControllerState.READY);
        agentInfo.setApproved(true);
        agentRepository.save(agentInfo);
        localAgentService.expireCache();
        countMap = agentManagerService.getAvailableAgentCountMap(getTestUser());

        int newCount = countMap.get(config.getRegion()).intValue();
        assertThat(newCount, is(oriCount + 1));
    }

    @Test
    public void testCheckAgentState() {
        AgentInfo agentInfo = new AgentInfo();
        agentInfo.setName("localhost");
        agentInfo.setRegion(config.getRegion());
        agentInfo.setIp("127.127.127.127");
        agentInfo.setPort(1);
        agentInfo.setState(AgentControllerState.READY);
        agentRepository.save(agentInfo);
        localAgentService.expireCache();
        agentManagerService.checkAgentState();

        AgentInfo agentInDB = agentRepository.findOne(agentInfo.getId());
        assertThat(agentInDB.getIp(), is(agentInfo.getIp()));
        assertThat(agentInDB.getName(), is(agentInfo.getName()));
        assertThat(agentInDB.getState(), is(AgentControllerState.INACTIVE));
    }

    @Test
    public void testCompressAgentFolder() throws IOException, URISyntaxException {
        URLClassLoader loader = (URLClassLoader) this.getClass().getClassLoader();
        URL core = this.getClass().getClassLoader().getResource("lib/ngrinder-core-test.jar");
        URL sh = this.getClass().getClassLoader().getResource("lib/ngrinder-sh-test.jar");
        URL[] ls = { core, sh };
        URL[] urls = loader.getURLs();
        URL[] allLib = cast(ArrayUtils.addAll(urls, ls));
        URLClassLoader child = new URLClassLoader(allLib, this.getClass().getClassLoader());
        File agentUpgrade = agentPackageService.createAgentPackage(child, null, null, 10000, null);
        FileUtils.deleteQuietly(agentUpgrade);
    }

    @Test
    public void testOther() {
    }

}