Repository: oodt Updated Branches: refs/heads/master 038fdca69 -> df9bdc2d8
http://git-wip-us.apache.org/repos/asf/oodt/blob/287d4e89/resource/src/test/java/org/apache/oodt/cas/resource/system/TestAvroRpcResourceManager.java ---------------------------------------------------------------------- diff --git a/resource/src/test/java/org/apache/oodt/cas/resource/system/TestAvroRpcResourceManager.java b/resource/src/test/java/org/apache/oodt/cas/resource/system/TestAvroRpcResourceManager.java new file mode 100644 index 0000000..2dc867f --- /dev/null +++ b/resource/src/test/java/org/apache/oodt/cas/resource/system/TestAvroRpcResourceManager.java @@ -0,0 +1,159 @@ +/* + * 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.system; + +import junit.framework.TestCase; +import org.apache.commons.io.FileUtils; +import org.apache.oodt.cas.resource.structs.NameValueJobInput; +import org.apache.oodt.cas.resource.structs.exceptions.MonitorException; + +import java.io.File; +import java.io.FileFilter; +import java.io.FileInputStream; +import java.io.IOException; +import java.net.URL; +import java.util.Properties; + +public class TestAvroRpcResourceManager extends TestCase { + + private File tmpPolicyDir; + + private AvroRpcResourceManager rm; + + private static final int RM_PORT = 50001; + + /** + * @since OODT-182 + */ + public void testDynSetNodeCapacity() { + AvroRpcResourceManagerClient rmc = null; + try { + rmc = new AvroRpcResourceManagerClient(new URL("http://localhost:"; + + RM_PORT)); + } catch (Exception e) { + System.out.println("radu1"); + e.printStackTrace(); + fail(e.getMessage()); + } + + assertNotNull(rmc); + try { + rmc.setNodeCapacity("localhost", 8); + } catch (MonitorException e) { + System.out.println("radu2"); + e.printStackTrace(); + fail(e.getMessage()); + } + + int setCapacity = -1; + try { + setCapacity = rmc.getNodeById("localhost").getCapacity(); + + } catch (Exception e) { + System.out.println("radu3"); + e.printStackTrace(); + fail(e.getMessage()); + } + assertEquals(8, setCapacity); + + } + + /* + * (non-Javadoc) + * + * @see junit.framework.TestCase#setUp() + */ + @Override + protected void setUp() throws Exception { + try { + System.out.println(NameValueJobInput.class.getCanonicalName()); + + generateTestConfiguration(); + this.rm = new AvroRpcResourceManager(RM_PORT); + } + catch (Exception e ){ + System.out.println("radu5"); + e.printStackTrace(); + } + } + + /* + * (non-Javadoc) + * + * @see junit.framework.TestCase#tearDown() + */ + @Override + protected void tearDown() throws Exception { + this.rm.shutdown(); + deleteAllFiles(this.tmpPolicyDir.getAbsolutePath()); + } + + private void deleteAllFiles(String startDir) { + File startDirFile = new File(startDir); + File[] delFiles = startDirFile.listFiles(); + + if (delFiles != null && delFiles.length > 0) { + for (int i = 0; i < delFiles.length; i++) { + delFiles[i].delete(); + } + } + + startDirFile.delete(); + + } + + private void generateTestConfiguration() throws IOException { + Properties config = new Properties(); + + String propertiesFile = "." + File.separator + "src" + File.separator + + "test" + File.separator + "resources" + File.separator + "test.resource.properties"; + System.getProperties().load(new FileInputStream(new File("/Users/radu/gsoc/test/avro/oodt/resource/src/test/resources/test.resource.properties"))); + + // stage policy + File tmpPolicyDir = null; + try { + tmpPolicyDir = File.createTempFile("test", "ignore").getParentFile(); + } catch (Exception e) { + fail(e.getMessage()); + } + for (File policyFile : new File("/Users/radu/gsoc/test/avro/oodt/resource/src/test/resources/policy") + .listFiles(new FileFilter() { + + @Override + public boolean accept(File pathname) { + return pathname.isFile() && pathname.getName().endsWith(".xml"); + } + })) { + try { + FileUtils.copyFileToDirectory(policyFile, tmpPolicyDir); + } catch (Exception e) { + fail(e.getMessage()); + } + } + + config.setProperty("org.apache.oodt.cas.resource.nodes.dirs", tmpPolicyDir + .toURI().toString()); + config.setProperty("org.apache.oodt.cas.resource.nodetoqueues.dirs", + tmpPolicyDir.toURI().toString()); + + System.getProperties().putAll(config); + this.tmpPolicyDir = tmpPolicyDir; + } + + +} http://git-wip-us.apache.org/repos/asf/oodt/blob/287d4e89/resource/src/test/java/org/apache/oodt/cas/resource/system/TestXmlRpcResourceManager.java ---------------------------------------------------------------------- diff --git a/resource/src/test/java/org/apache/oodt/cas/resource/system/TestXmlRpcResourceManager.java b/resource/src/test/java/org/apache/oodt/cas/resource/system/TestXmlRpcResourceManager.java index e2c0307..8f27bcb 100644 --- a/resource/src/test/java/org/apache/oodt/cas/resource/system/TestXmlRpcResourceManager.java +++ b/resource/src/test/java/org/apache/oodt/cas/resource/system/TestXmlRpcResourceManager.java @@ -29,6 +29,7 @@ import java.util.Properties; import org.apache.commons.io.FileUtils; //OODT imports +import org.apache.oodt.cas.resource.structs.NameValueJobInput; import org.apache.oodt.cas.resource.structs.exceptions.MonitorException; //Junit imports @@ -87,6 +88,9 @@ public class TestXmlRpcResourceManager extends TestCase { */ @Override protected void setUp() throws Exception { + + System.out.println(NameValueJobInput.class.getCanonicalName()); + generateTestConfiguration(); this.rm = new XmlRpcResourceManager(RM_PORT); }
