0.10/resource/src/test/java/org/apache/oodt/cas/resource/monitor/TestAssignmentMonitor.java

/*
 * 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;

//OODT imports
import org.apache.oodt.cas.resource.structs.ResourceNode;
import org.apache.oodt.cas.resource.structs.exceptions.MonitorException;

//JDK imports
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Iterator;
import java.util.List;
import java.util.Vector;

//JUnit imports
import junit.framework.TestCase;

/**
 * @author mattmann
 * @author bfoster
 * @author rajith
 * @version $Revision$
 * 
 * <p>
 * Test Suite for the {@link AssignmentMonitor} service
 * </p>.
 */
public class TestAssignmentMonitor extends TestCase {

    private AssignmentMonitor assgnMon = null;

    protected void setUp() throws IOException {
        generateTestConfig();
        assgnMon = new AssignmentMonitorFactory().createMonitor();
    }

    public void testGetNodes() {
        List resNodes = null;

        try {
            resNodes = assgnMon.getNodes();
        } catch (MonitorException e) {
            fail(e.getMessage());
        }

        assertNotNull(resNodes);
        assertEquals(1, resNodes.size());
    }

    public void testGetNodeById() {
        ResourceNode node = null;

        try {
            node = assgnMon.getNodeById("localhost");
        } catch (MonitorException e) {
            fail(e.getMessage());
        }

        assertNotNull(node);
        assertEquals("localhost", node.getNodeId());
    }

    public void testGetNodeInfo() {
        List resNodes = null;

        try {
            resNodes = assgnMon.getNodes();
        } catch (MonitorException e) {
            fail(e.getMessage());
        }

        assertNotNull(resNodes);

        boolean hasNode1 = false;

        for (Iterator i = resNodes.iterator(); i.hasNext();) {
            ResourceNode node = (ResourceNode) i.next();
            assertNotNull(node);
            if (node.getNodeId().equals("localhost")) {
                hasNode1 = true;
                assertEquals(node.getIpAddr().toExternalForm(),
                        "http://localhost:2001");
            }
            assertEquals(node.getCapacity(), 8);
        }

        assertTrue(hasNode1);
    }

    public void testNodeModification() throws MonitorException,
            MalformedURLException {
        List<ResourceNode> nodes = new Vector<ResourceNode>(this.assgnMon
                .getNodes());
        ResourceNode test1 = new ResourceNode("Test1", new URL(
                "http://localhost:1111"), 9);
        ResourceNode test2 = new ResourceNode("Test2", new URL(
                "http://localhost:2222"), 9);
        ResourceNode test3 = new ResourceNode("Test3", new URL(
                "http://localhost:3333"), 9);
        this.assgnMon.addNode(test1);
        nodes.add(test1);
        this.assgnMon.addNode(test2);
        nodes.add(test2);
        this.assgnMon.addNode(test3);
        nodes.add(test3);

        assertTrue(nodes.containsAll(this.assgnMon.getNodes())
                && this.assgnMon.getNodes().containsAll(nodes));
        
        this.assgnMon.removeNodeById(test1.getNodeId());
        nodes.remove(test1);
        
        assertTrue(nodes.containsAll(this.assgnMon.getNodes())
                && this.assgnMon.getNodes().containsAll(nodes));
    }


    private void generateTestConfig() throws IOException {
        String propertiesFile = "." + File.separator + "src" + File.separator +
                "test" + File.separator + "resources" + File.separator + "test.resource.properties";

        System.getProperties().load(new FileInputStream(new File(propertiesFile)));

        System.setProperty("org.apache.oodt.cas.resource.nodes.dirs",
                "file:" + new File("." + File.separator + "src" + File.separator +
                        "main" + File.separator + "resources" + File.separator +
                        "examples").getAbsolutePath());
    }

}