/*
* 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.flink.runtime.testutils;
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 org.apache.flink.core.io.IOReadableWritable;
import org.apache.flink.core.memory.InputViewDataInputStreamWrapper;
import org.apache.flink.core.memory.OutputViewDataOutputStreamWrapper;
/**
* 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(new OutputViewDataOutputStreamWrapper(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(new InputViewDataInputStreamWrapper(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;
}
}