moparisthebest/poi
1
0
Fork 0
Code Issues Pull Requests Releases 1 Wiki Activity

Add a deep re-write NPOIFS test

Browse Source 
git-svn-id: https://svn.apache.org/repos/asf/poi/trunk@1678781 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Nick Burch 2015-05-11 18:00:45 +00:00
parent 1945dfc00c
commit bd6012d9b9
1 changed files with 33 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
src/testcases/org/apache/poi/poifs/filesystem/TestFileSystemBugs.java
View File

@ -21,8 +21,10 @@ import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import junit.framework.TestCase;
@ -125,9 +127,11 @@ public final class TestFileSystemBugs extends TestCase {
/**
* With heavily nested documents, ensure we still re-write the same
*/
public void IGNOREDtestHeavilyNestedReWrite() throws Exception {
public void testHeavilyNestedReWrite() throws Exception {
for (DirectoryNode root : openSSSample("ex42570-20305.xls", false)) {
// TODO Record the structure
// Record the structure
Map<String,Integer> entries = new HashMap<String, Integer>();
fetchSizes("/", root, entries);
// Prepare to copy
DirectoryNode dest;
@ -150,7 +154,33 @@ public final class TestFileSystemBugs extends TestCase {
NPOIFSFileSystem read = new NPOIFSFileSystem(
new ByteArrayInputStream(baos.toByteArray()));
// TODO Check the structure matches
// Check the structure matches
checkSizes("/", read.getRoot(), entries);
}
}
private void fetchSizes(String path, DirectoryNode dir, Map<String,Integer> entries) {
for (Entry entry : dir) {
if (entry instanceof DirectoryNode) {
String ourPath = path + entry.getName() + "/";
entries.put(ourPath, -1);
fetchSizes(ourPath, (DirectoryNode)entry, entries);
} else {
DocumentNode doc = (DocumentNode)entry;
entries.put(path+entry.getName(), doc.getSize());
}
}
}
private void checkSizes(String path, DirectoryNode dir, Map<String,Integer> entries) {
for (Entry entry : dir) {
if (entry instanceof DirectoryNode) {
String ourPath = path + entry.getName() + "/";
assertTrue(entries.containsKey(ourPath));
assertEquals(-1, entries.get(ourPath).intValue());
checkSizes(ourPath, (DirectoryNode)entry, entries);
} else {
DocumentNode doc = (DocumentNode)entry;
assertEquals(entries.get(path+entry.getName()).intValue(), doc.getSize());
}
}
}
}