package org.apache.poi.xssf.extractor;
+import static org.apache.poi.POITestCase.assertStartsWith;
+import static org.apache.poi.POITestCase.assertEndsWith;
+
import java.io.IOException;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
assertTrue(text.length() > 0);
// Check sheet names
- assertTrue(text.startsWith("Sheet1"));
- assertTrue(text.endsWith("Sheet3\n"));
+ assertStartsWith(text, "Sheet1");
+ assertEndsWith(text, "Sheet3\n");
// Now without, will have text
extractor.setIncludeSheetNames(false);
// Might not have all formatting it should do!
// TODO decide if we should really have the "null" in there
- assertTrue(text.startsWith(
+ assertStartsWith(text,
"Avgtxfull\n" +
"null\t(iii) AVERAGE TAX RATES ON ANNUAL"
- ));
+ );
extractor.close();
}
for (POITextExtractor extractor : extractors) {
String text = extractor.getText().replaceAll("[\r\t]", "");
- assertTrue(text.startsWith("First Sheet\nTest spreadsheet\n2nd row2nd row 2nd column\n"));
+ assertStartsWith(text, "First Sheet\nTest spreadsheet\n2nd row2nd row 2nd column\n");
Pattern pattern = Pattern.compile(".*13(\\.0+)?\\s+Sheet3.*", Pattern.DOTALL);
Matcher m = pattern.matcher(text);
assertTrue(m.matches());
*/
@Internal
public final class POITestCase {
+ /*
+ * Returns the first {@code length} characters from the string {@code s}
+ */
+ private static String head(String s, int length) {
+ if (length <= 0) {
+ return "";
+ }
+ if (s.length() <= length) {
+ return s;
+ }
+ final StringBuilder sb = new StringBuilder(s.substring(0, length));
+ sb.append("... (length=").append(s.length()).append(")");
+ return sb.toString();
+ }
+
+ /*
+ * Returns the last {@code length} characters from the string {@code s}
+ */
+ private static String tail(String s, int length) {
+ if (length <= 0) {
+ return "";
+ }
+ if (s.length() <= length) {
+ return s;
+ }
+ final StringBuilder sb = new StringBuilder();
+ sb.append("(length=").append(s.length()).append(") ...");
+ sb.append(s.substring(s.length() - length));
+ return sb.toString();
+ }
+
public static void assertStartsWith(String string, String prefix) {
assertNotNull(string);
assertNotNull(prefix);
-
- final int len = Math.min(string.length(), prefix.length());
- assertEquals("string does not start with prefix", prefix, string.substring(0, len));
+ assertEquals("string does not start with prefix", prefix, head(string, prefix.length()+5));
+ }
+
+ public static void assertEndsWith(String string, String suffix) {
+ assertNotNull(string);
+ assertNotNull(suffix);
+ assertEquals("string does not end with suffix", suffix, tail(string, suffix.length()+5));
}
public static void assertContains(String haystack, String needle) {
POITestCase.assertStartsWith("Apache POI", "Apache");
POITestCase.assertStartsWith("Apache POI", "Apache POI");
}
+
+ @Test
+ public void assertEndsWith() {
+ POITestCase.assertEndsWith("Apache POI", "");
+ POITestCase.assertEndsWith("Apache POI", "POI");
+ POITestCase.assertEndsWith("Apache POI", "Apache POI");
+ }
@Test
public void assertContains() {