/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.apache.oodt.cas.resource.monitor;
//Junit imports
import junit.framework.TestCase;
//OODT imports
import org.apache.oodt.cas.resource.monitor.ganglia.GangliaResourceMonitor;
import org.apache.oodt.cas.resource.monitor.ganglia.GangliaResourceMonitorFactory;
import org.apache.oodt.cas.resource.monitor.utils.MockGmetad;
import org.apache.oodt.cas.resource.structs.ResourceNode;
//JDK imports
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.List;
/**
* @author rajith
* @author mattmann
* @version $Revision$
*
* Test Suite for the {@link org.apache.oodt.cas.resource.monitor.ganglia.GangliaResourceMonitor}
*/
public class TestGangliaResourceMonitor extends TestCase {
private GangliaResourceMonitor gangliaResourceMonitor;
private ThreadLocal<MockGmetad> mockGmetad = new ThreadLocal<MockGmetad>();
@Override
protected void setUp() throws IOException {
generateTestConfig();
runMockGmetad();
gangliaResourceMonitor = (GangliaResourceMonitor)
new GangliaResourceMonitorFactory().createMonitor();
}
@Override
protected void tearDown(){
mockGmetad.remove();
}
public void testGetLoad() {
try {
ResourceNode resourceNode = new ResourceNode();
resourceNode.setId("localhost");
assertEquals(1, gangliaResourceMonitor.getLoad(resourceNode));
} catch (Exception e) {
e.printStackTrace();
fail(e.getMessage());
}
}
public void testRemoveNodeById(){
try {
gangliaResourceMonitor.removeNodeById("localhost");
assertNull(gangliaResourceMonitor.getNodeById("remotenode"));
} catch (Exception e) {
e.printStackTrace();
fail(e.getMessage());
}
}
public void testGetNodes(){
try{
List<ResourceNode> nodes = gangliaResourceMonitor.getNodes();
assertNotNull(nodes);
assertEquals(3, nodes.size());
boolean hasLocal = false;
boolean hasLocal2 = false;
boolean hasRemote = false;
for(ResourceNode node: nodes){
if(node.getNodeId().equals("localhost")){
hasLocal = true;
}
else if(node.getNodeId().equals("localhost2")){
hasLocal2 = true;
}
else if(node.getNodeId().equals("remotenode")){
hasRemote = true;
}
}
assertTrue(hasLocal&&hasLocal2&&hasRemote);
}
catch (Exception e){
e.printStackTrace();
fail(e.getMessage());
}
}
public void testGetNodeById(){
try{
ResourceNode node = gangliaResourceMonitor.getNodeById("localhost");
assertNotNull(node);
assertEquals("localhost", node.getNodeId());
node = gangliaResourceMonitor.getNodeById("localhost2");
assertNotNull(node);
assertEquals("localhost2", node.getNodeId());
node = gangliaResourceMonitor.getNodeById("remotenode");
assertNotNull(node);
assertEquals("remotenode", node.getNodeId());
}
catch(Exception e){
e.printStackTrace();
fail(e.getMessage());
}
}
private void runMockGmetad() {
int port = Integer.valueOf(System
.getProperty("org.apache.oodt.cas.resource.monitor.ganglia.gemtad.host.port"));
String sampleXMLfilePath = "." + File.separator + "src" + File.separator +
"testdata" + File.separator + "resourcemon" + File.separator + "gangliaXMLdump.xml";
mockGmetad.set(new MockGmetad(port, sampleXMLfilePath));
Thread mockGmetadServer = new Thread(mockGmetad.get());
mockGmetadServer.start();
}
private void generateTestConfig() throws IOException {
String propertiesFile = "." + File.separator + "src" + File.separator +
"testdata" + File.separator + "test.resource.properties";
System.getProperties().load(new FileInputStream(new File(propertiesFile)));
}
}