poi/src/test/java/org/apache/poi/POITestCase.java - poi - Git at Google

 /* ====================================================================
    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.hamcrest.CoreMatchers.containsString;
 import static org.hamcrest.CoreMatchers.endsWith;
 import static org.hamcrest.CoreMatchers.hasItem;
 import static org.hamcrest.CoreMatchers.not;
 import static org.hamcrest.CoreMatchers.startsWith;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 import static org.junit.jupiter.api.Assertions.fail;
 import static org.junit.jupiter.api.Assumptions.assumeTrue;

 import java.io.File;
 import java.lang.reflect.Field;
 import java.security.AccessController;
 import java.security.PrivilegedActionException;
 import java.security.PrivilegedExceptionAction;
 import java.util.Locale;
 import java.util.Map;
 import java.util.Set;

 import javax.imageio.ImageIO;

 import org.apache.poi.util.Internal;
 import org.apache.poi.util.SuppressForbidden;
 import org.mockito.internal.matchers.apachecommons.ReflectionEquals;

 /**
  * Util class for POI JUnit TestCases, which provide additional features
  */
 @Internal
 public final class POITestCase {

     private POITestCase() {
     }

     public static void assertStartsWith(String string, String prefix) {
         assertNotNull(string);
         assertNotNull(prefix);
         assertThat(string, startsWith(prefix));
     }

     public static void assertStartsWith(String message, String string, String prefix) {
         assertNotNull(message, string);
         assertNotNull(message, prefix);
         assertThat(message, string, startsWith(prefix));
     }

     public static void assertEndsWith(String string, String suffix) {
         assertNotNull(string);
         assertNotNull(suffix);
         assertThat(string, endsWith(suffix));
     }

     public static void assertContains(String haystack, String needle) {
         assertNotNull(haystack);
         assertNotNull(needle);
         assertThat(haystack, containsString(needle));
     }

     public static void assertContains(String message, String haystack, String needle) {
         assertNotNull(message, haystack);
         assertNotNull(message, needle);
         assertThat(message, haystack, containsString(needle));
     }

     public static void assertContainsIgnoreCase(String haystack, String needle, Locale locale) {
         assertNotNull(haystack);
         assertNotNull(needle);
         String hay = haystack.toLowerCase(locale);
         String n = needle.toLowerCase(locale);
         assertTrue(hay.contains(n), "Unable to find expected text '" + needle + "' in text:\n" + haystack);
     }
     public static void assertContainsIgnoreCase(String haystack, String needle) {
         assertContainsIgnoreCase(haystack, needle, Locale.ROOT);
     }

     public static void assertNotContained(String haystack, String needle) {
         assertNotNull(haystack);
         assertNotNull(needle);
         assertThat(haystack, not(containsString(needle)));
     }

     /**
      * @param map haystack
      * @param key needle
      */
     public static  <T> void assertContains(Map<T, ?> map, T key) {
         assertTrue(map.containsKey(key), "Unable to find " + key + " in " + map);
     }

     public static <T> void assertNotContained(Set<T> set, T element) {
         assertThat("Set should not contain " + element, set, not(hasItem(element)));
     }

     /**
      * Utility method to get the value of a private/protected field.
      * Only use this method in test cases!!!
      */
     @SuppressWarnings({"unused", "unchecked"})
     @SuppressForbidden("For test usage only")
     public static <R,T> R getFieldValue(final Class<? super T> clazz, final T instance, final Class<R> fieldType, final String fieldName) {
         assertTrue(clazz.getName().startsWith("org.apache.poi."), "Reflection of private fields is only allowed for POI classes.");
         try {
             return AccessController.doPrivileged((PrivilegedExceptionAction<R>) () -> {
                 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 shallow compare all fields of the objects
      * Only use this method in test cases!!!
      */
     public static void assertReflectEquals(final Object expected, Object actual) {
         // as long as ReflectionEquals is provided by Mockito, use it ... otherwise use commons.lang for the tests

         // JaCoCo Code Coverage adds its own field, don't look at this one here
         assertTrue(new ReflectionEquals(expected, "$jacocoData").matches(actual));
     }

     /**
      * Rather than adding {@literal @}Ignore to known-failing tests,
      * write the test so that it notifies us if it starts passing.
      * This is useful for closing related or forgotten bugs.
      *
      * An Example:
      * <code><pre>
      * public static int add(int a, int b) {
      *     // a known bug in behavior that has not been fixed yet
      *     raise UnsupportedOperationException("add");
      * }
      *
      * {@literal @}Test
      * void knownFailingUnitTest() {
      *     try {
      *         assertEquals(2, add(1,1));
      *         // this test fails because the assumption that this bug had not been fixed is false
      *         testPassesNow(12345);
      *     } catch (UnsupportedOperationException e) {
      *         // test is skipped because the assumption that this bug had not been fixed is true
      *         skipTest(e);
      *     }
      * }
      *
      * Once passing, this unit test can be rewritten as:
      * {@literal @}Test
      * void knownPassingUnitTest() {
      *     assertEquals(2, add(1,1));
      * }
      *
      * If you have a better idea how to simplify test code while still notifying
      * us when a previous known-failing test now passes, please improve these.
      * As a bonus, a known-failing test that fails should not be counted as a
      * passing test.
      *
      * One possible alternative is to expect the known exception, but without
      * a clear message that it is a good thing to no longer get the expected
      * exception once the test passes.
      * {@literal @}Test(expected=UnsupportedOperationException.class)
      * void knownFailingUnitTest() {
      *     assertEquals(2, add(1,1));
      * }
      *
      * @param e  the exception that was caught that will no longer
      * be raised when the bug is fixed
      */
     public static void skipTest(Throwable e) {
         assumeTrue(e != null, "This test currently fails with");
     }
     /**
      * @see #skipTest(Throwable)
      *
      * @param bug  the bug number corresponding to a known bug in bugzilla
      */
     public static void testPassesNow(int bug) {
         fail("This test passes now. Please update the unit test and bug " + bug + ".");
     }

     public static void assertBetween(String message, int value, int min, int max) {
         assertTrue(min <= value, message + ": " + value + " is less than the minimum value of " + min);
         assertTrue(value <= max, message + ": " + value + " is greater than the maximum value of " + max);
     }

 	/**
 	 * Ensures that the temporary directory is defined and exists and
 	 * ensures ImageIO uses this directory for cache-files
 	 */
 	public static void setImageIOCacheDir() {
 		final String tmpDirProperty = System.getProperty("java.io.tmpdir");
 		if(tmpDirProperty == null || "".equals(tmpDirProperty)) {
 			return;
 		}
 		// ensure that temp-dir exists because ImageIO requires it
 		File tmpDir = new File(tmpDirProperty);
 		if(!tmpDir.exists() && !tmpDir.mkdirs()) {
 			throw new IllegalStateException("Could not create temporary directory " + tmpDirProperty + ", full path " + tmpDir.getAbsolutePath());
 		}
 		ImageIO.setCacheDirectory(tmpDir);
 	}
 }
	/* ====================================================================
	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.hamcrest.CoreMatchers.containsString;
	import static org.hamcrest.CoreMatchers.endsWith;
	import static org.hamcrest.CoreMatchers.hasItem;
	import static org.hamcrest.CoreMatchers.not;
	import static org.hamcrest.CoreMatchers.startsWith;
	import static org.hamcrest.MatcherAssert.assertThat;
	import static org.junit.jupiter.api.Assertions.assertNotNull;
	import static org.junit.jupiter.api.Assertions.assertTrue;
	import static org.junit.jupiter.api.Assertions.fail;
	import static org.junit.jupiter.api.Assumptions.assumeTrue;

	import java.io.File;
	import java.lang.reflect.Field;
	import java.security.AccessController;
	import java.security.PrivilegedActionException;
	import java.security.PrivilegedExceptionAction;
	import java.util.Locale;
	import java.util.Map;
	import java.util.Set;

	import javax.imageio.ImageIO;

	import org.apache.poi.util.Internal;
	import org.apache.poi.util.SuppressForbidden;
	import org.mockito.internal.matchers.apachecommons.ReflectionEquals;

	/**
	* Util class for POI JUnit TestCases, which provide additional features
	*/
	@Internal
	public final class POITestCase {

	private POITestCase() {
	}

	public static void assertStartsWith(String string, String prefix) {
	assertNotNull(string);
	assertNotNull(prefix);
	assertThat(string, startsWith(prefix));
	}

	public static void assertStartsWith(String message, String string, String prefix) {
	assertNotNull(message, string);
	assertNotNull(message, prefix);
	assertThat(message, string, startsWith(prefix));
	}

	public static void assertEndsWith(String string, String suffix) {
	assertNotNull(string);
	assertNotNull(suffix);
	assertThat(string, endsWith(suffix));
	}

	public static void assertContains(String haystack, String needle) {
	assertNotNull(haystack);
	assertNotNull(needle);
	assertThat(haystack, containsString(needle));
	}

	public static void assertContains(String message, String haystack, String needle) {
	assertNotNull(message, haystack);
	assertNotNull(message, needle);
	assertThat(message, haystack, containsString(needle));
	}

	public static void assertContainsIgnoreCase(String haystack, String needle, Locale locale) {
	assertNotNull(haystack);
	assertNotNull(needle);
	String hay = haystack.toLowerCase(locale);
	String n = needle.toLowerCase(locale);
	assertTrue(hay.contains(n), "Unable to find expected text '" + needle + "' in text:\n" + haystack);
	}
	public static void assertContainsIgnoreCase(String haystack, String needle) {
	assertContainsIgnoreCase(haystack, needle, Locale.ROOT);
	}

	public static void assertNotContained(String haystack, String needle) {
	assertNotNull(haystack);
	assertNotNull(needle);
	assertThat(haystack, not(containsString(needle)));
	}

	/**
	* @param map haystack
	* @param key needle
	*/
	public static <T> void assertContains(Map<T, ?> map, T key) {
	assertTrue(map.containsKey(key), "Unable to find " + key + " in " + map);
	}

	public static <T> void assertNotContained(Set<T> set, T element) {
	assertThat("Set should not contain " + element, set, not(hasItem(element)));
	}

	/**
	* Utility method to get the value of a private/protected field.
	* Only use this method in test cases!!!
	*/
	@SuppressWarnings({"unused", "unchecked"})
	@SuppressForbidden("For test usage only")
	public static <R,T> R getFieldValue(final Class<? super T> clazz, final T instance, final Class<R> fieldType, final String fieldName) {
	assertTrue(clazz.getName().startsWith("org.apache.poi."), "Reflection of private fields is only allowed for POI classes.");
	try {
	return AccessController.doPrivileged((PrivilegedExceptionAction<R>) () -> {
	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 shallow compare all fields of the objects
	* Only use this method in test cases!!!
	*/
	public static void assertReflectEquals(final Object expected, Object actual) {
	// as long as ReflectionEquals is provided by Mockito, use it ... otherwise use commons.lang for the tests

	// JaCoCo Code Coverage adds its own field, don't look at this one here
	assertTrue(new ReflectionEquals(expected, "$jacocoData").matches(actual));
	}

	/**
	* Rather than adding {@literal @}Ignore to known-failing tests,
	* write the test so that it notifies us if it starts passing.
	* This is useful for closing related or forgotten bugs.
	*
	* An Example:
	* <code><pre>
	* public static int add(int a, int b) {
	* // a known bug in behavior that has not been fixed yet
	* raise UnsupportedOperationException("add");
	* }
	*
	* {@literal @}Test
	* void knownFailingUnitTest() {
	* try {
	* assertEquals(2, add(1,1));
	* // this test fails because the assumption that this bug had not been fixed is false
	* testPassesNow(12345);
	* } catch (UnsupportedOperationException e) {
	* // test is skipped because the assumption that this bug had not been fixed is true
	* skipTest(e);
	* }
	* }
	*
	* Once passing, this unit test can be rewritten as:
	* {@literal @}Test
	* void knownPassingUnitTest() {
	* assertEquals(2, add(1,1));
	* }
	*
	* If you have a better idea how to simplify test code while still notifying
	* us when a previous known-failing test now passes, please improve these.
	* As a bonus, a known-failing test that fails should not be counted as a
	* passing test.
	*
	* One possible alternative is to expect the known exception, but without
	* a clear message that it is a good thing to no longer get the expected
	* exception once the test passes.
	* {@literal @}Test(expected=UnsupportedOperationException.class)
	* void knownFailingUnitTest() {
	* assertEquals(2, add(1,1));
	* }
	*
	* @param e the exception that was caught that will no longer
	* be raised when the bug is fixed
	*/
	public static void skipTest(Throwable e) {
	assumeTrue(e != null, "This test currently fails with");
	}
	/**
	* @see #skipTest(Throwable)
	*
	* @param bug the bug number corresponding to a known bug in bugzilla
	*/
	public static void testPassesNow(int bug) {
	fail("This test passes now. Please update the unit test and bug " + bug + ".");
	}

	public static void assertBetween(String message, int value, int min, int max) {
	assertTrue(min <= value, message + ": " + value + " is less than the minimum value of " + min);
	assertTrue(value <= max, message + ": " + value + " is greater than the maximum value of " + max);
	}

	/**
	* Ensures that the temporary directory is defined and exists and
	* ensures ImageIO uses this directory for cache-files
	*/
	public static void setImageIOCacheDir() {
	final String tmpDirProperty = System.getProperty("java.io.tmpdir");
	if(tmpDirProperty == null \|\| "".equals(tmpDirProperty)) {
	return;
	}
	// ensure that temp-dir exists because ImageIO requires it
	File tmpDir = new File(tmpDirProperty);
	if(!tmpDir.exists() && !tmpDir.mkdirs()) {
	throw new IllegalStateException("Could not create temporary directory " + tmpDirProperty + ", full path " + tmpDir.getAbsolutePath());
	}
	ImageIO.setCacheDirectory(tmpDir);
	}
	}