/*
* 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() {
}
}