162 lines
6.6 KiB
Java
162 lines
6.6 KiB
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.poi;
|
|
|
|
import static org.junit.Assert.assertArrayEquals;
|
|
import static org.junit.Assert.assertEquals;
|
|
import static org.junit.Assert.assertFalse;
|
|
import static org.junit.Assert.assertNotNull;
|
|
import static org.junit.Assert.assertTrue;
|
|
import static org.junit.Assert.fail;
|
|
|
|
import java.lang.reflect.AccessibleObject;
|
|
import java.lang.reflect.Field;
|
|
import java.lang.reflect.Method;
|
|
import java.security.AccessController;
|
|
import java.security.PrivilegedActionException;
|
|
import java.security.PrivilegedExceptionAction;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
import org.apache.poi.util.SuppressForbidden;
|
|
|
|
/**
|
|
* Util class for POI JUnit TestCases, which provide additional features
|
|
*/
|
|
public final class POITestCase {
|
|
public static void assertContains(String haystack, String needle) {
|
|
assertNotNull(haystack);
|
|
assertTrue(
|
|
"Unable to find expected text '" + needle + "' in text:\n" + haystack,
|
|
haystack.contains(needle)
|
|
);
|
|
}
|
|
|
|
public static void assertNotContained(String haystack, String needle) {
|
|
assertNotNull(haystack);
|
|
assertFalse(
|
|
"Unexpectedly found text '" + needle + "' in text:\n" + haystack,
|
|
haystack.contains(needle)
|
|
);
|
|
}
|
|
|
|
/**
|
|
* @param map haystack
|
|
* @param key needle
|
|
*/
|
|
public static <T> void assertContains(Map<T, ?> map, T key) {
|
|
if (map.containsKey(key)) {
|
|
return;
|
|
}
|
|
fail("Unable to find " + key + " in " + map);
|
|
}
|
|
|
|
/**
|
|
* Utility method to get the value of a private/protected field.
|
|
* Only use this method in test cases!!!
|
|
*/
|
|
public static <R,T> R getFieldValue(final Class<? super T> clazz, final T instance, final Class<R> fieldType, final String fieldName) {
|
|
assertTrue("Reflection of private fields is only allowed for POI classes.", clazz.getName().startsWith("org.apache.poi."));
|
|
try {
|
|
return AccessController.doPrivileged(new PrivilegedExceptionAction<R>() {
|
|
@Override
|
|
@SuppressWarnings("unchecked")
|
|
@SuppressForbidden("For test usage only")
|
|
public R run() throws Exception {
|
|
Field f = clazz.getDeclaredField(fieldName);
|
|
f.setAccessible(true);
|
|
return (R) f.get(instance);
|
|
}
|
|
});
|
|
} catch (PrivilegedActionException pae) {
|
|
throw new RuntimeException("Cannot access field '" + fieldName + "' of class " + clazz, pae.getException());
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Utility method to call a private/protected method.
|
|
* Only use this method in test cases!!!
|
|
*/
|
|
public static <R,T> R callMethod(final Class<? super T> clazz, final T instance, final Class<R> returnType, final String methodName,
|
|
final Class<?>[] parameterTypes, final Object[] parameters) {
|
|
assertTrue("Reflection of private methods is only allowed for POI classes.", clazz.getName().startsWith("org.apache.poi."));
|
|
try {
|
|
return AccessController.doPrivileged(new PrivilegedExceptionAction<R>() {
|
|
@Override
|
|
@SuppressWarnings("unchecked")
|
|
@SuppressForbidden("For test usage only")
|
|
public R run() throws Exception {
|
|
Method m = clazz.getDeclaredMethod(methodName, parameterTypes);
|
|
m.setAccessible(true);
|
|
return (R) m.invoke(instance, parameters);
|
|
}
|
|
});
|
|
} catch (PrivilegedActionException pae) {
|
|
throw new RuntimeException("Cannot access method '" + methodName + "' of class " + clazz, pae.getException());
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Utility method to shallow compare all fields of the objects
|
|
* Only use this method in test cases!!!
|
|
*/
|
|
public static void assertReflectEquals(final Object expected, Object actual) throws Exception {
|
|
final List<Field> fields;
|
|
try {
|
|
fields = AccessController.doPrivileged(new PrivilegedExceptionAction<List<Field>>() {
|
|
@Override
|
|
@SuppressForbidden("Test only")
|
|
public List<Field> run() throws Exception {
|
|
List<Field> flds = new ArrayList<Field>();
|
|
for (Class<?> c = expected.getClass(); c != null; c = c.getSuperclass()) {
|
|
Field[] fs = c.getDeclaredFields();
|
|
AccessibleObject.setAccessible(fs, true);
|
|
for (Field f : fs) {
|
|
// JaCoCo Code Coverage adds it's own field, don't look at this one here
|
|
if(f.getName().equals("$jacocoData")) {
|
|
continue;
|
|
}
|
|
|
|
flds.add(f);
|
|
}
|
|
}
|
|
return flds;
|
|
}
|
|
});
|
|
} catch (PrivilegedActionException pae) {
|
|
throw pae.getException();
|
|
}
|
|
|
|
for (Field f : fields) {
|
|
Class<?> t = f.getType();
|
|
if (t.isArray()) {
|
|
if (Object[].class.isAssignableFrom(t)) {
|
|
assertArrayEquals((Object[])f.get(expected), (Object[])f.get(actual));
|
|
} else if (byte[].class.isAssignableFrom(t)) {
|
|
assertArrayEquals((byte[])f.get(expected), (byte[])f.get(actual));
|
|
} else {
|
|
fail("Array type is not yet implemented ... add it!");
|
|
}
|
|
} else {
|
|
assertEquals(f.get(expected), f.get(actual));
|
|
}
|
|
}
|
|
}
|
|
}
|