Package eu.stratosphere.nephele.util

Source Code of eu.stratosphere.nephele.util.ManagementTestUtils

/***********************************************************************************************************************
* Copyright (C) 2010-2013 by the Stratosphere project (http://stratosphere.eu)
*
* 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 eu.stratosphere.nephele.util;

import static org.junit.Assert.fail;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.IOException;

import eu.stratosphere.core.io.IOReadableWritable;

/**
* This class contains auxiliary methods for unit tests in the Nephele management module.
*
*/
public final class ManagementTestUtils {

  /**
   * The system property key to retrieve the user directory.
   */
  private static final String USER_DIR_KEY = "user.dir";

  /**
   * The directory the configuration directory is expected in when test are executed using Eclipse.
   */
  private static final String ECLIPSE_PATH_EXTENSION = "/src/test/resources";

  /**
   * Private constructor, so class cannot be instantiated.
   */
  private ManagementTestUtils() {
  }

  /**
   * Creates a copy of the given {@link IOReadableWritable} object by an in-memory serialization and subsequent
   * deserialization.
   *
   * @param original
   *        the original object to be copied
   * @return the copy of original object created by the original object's serialization/deserialization methods
   */
  @SuppressWarnings("unchecked")
  public static IOReadableWritable createCopy(IOReadableWritable original) {

    final ByteArrayOutputStream baos = new ByteArrayOutputStream();
    final DataOutputStream dos = new DataOutputStream(baos);

    try {
      original.write(dos);
    } catch (IOException ioe) {
      fail(ioe.getMessage());
    }

    final String className = original.getClass().getName();
    if (className == null) {
      fail("Class name is null");
    }

    Class<? extends IOReadableWritable> clazz = null;

    try {
      clazz = (Class<? extends IOReadableWritable>) Class.forName(className);
    } catch (ClassNotFoundException e) {
      fail(e.getMessage());
    }

    if (clazz == null) {
      fail("Cannot find class with name " + className);
    }

    IOReadableWritable copy = null;
    try {
      copy = clazz.newInstance();
    } catch (InstantiationException e) {
      fail(e.getMessage());
    } catch (IllegalAccessException e) {
      fail(e.getMessage());
    }

    if (copy == null) {
      fail("Copy of object of type " + className + " is null");
    }

    final ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
    final DataInputStream dis = new DataInputStream(bais);

    try {
      copy.read(dis);
    } catch (IOException e) {
      fail(e.getMessage());
    }

    return copy;
  }

  /**
   * Locates a file-based resource that is used during testing. The method makes sure that the resource is always
   * located correctly, no matter if the test is executed in maven or Eclipse.
   *
   * @param resourceName
   *        the name of the resource to be located
   * @return a file object pointing to the resource or <code>null</code> if the resource could not be located
   */
  public static File locateResource(final String resourceName) {

    // This is the correct path for Maven-based tests
    File file = new File(System.getProperty(USER_DIR_KEY) + File.separator + resourceName);
    if (file.exists()) {
      return file;
    }

    file = new File(System.getProperty(USER_DIR_KEY) + ECLIPSE_PATH_EXTENSION + File.separator + resourceName);
    if (file.exists()) {
      return file;
    }

    return null;
  }
}
TOP

Related Classes of eu.stratosphere.nephele.util.ManagementTestUtils

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.