add unit test for bug 58779 when the problem gets fixed

git-svn-id: https://svn.apache.org/repos/asf/poi/trunk@1722091 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Javen O'Neal 2015-12-29 10:40:20 +00:00
parent dba7fcdd4e
commit 44c5423be4

View File

@ -21,6 +21,8 @@ import static org.junit.Assert.*;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import org.apache.poi.EmptyFileException;
@ -36,7 +38,7 @@ import org.apache.poi.util.TempFile;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.openxml4j.opc.PackageAccess;
import org.junit.Test;
public final class TestWorkbookFactory {
@ -49,20 +51,44 @@ public final class TestWorkbookFactory {
private static final POILogger LOGGER = POILogFactory.getLogger(TestWorkbookFactory.class);
/**
* // TODO: close() re-writes the sample-file?! Resort to revert() for now to close file handle...
* Revert the changes that were made to the workbook rather than closing the workbook.
* This allows the file handle to be closed to avoid the file handle leak detector.
* This is a temporary fix until we figure out why wb.close() writes changes to disk.
* Closes the sample workbook read in from filename.
* Throws an exception if closing the workbook results in the file on disk getting modified.
*
* @param filename the sample workbook to read in
* @param wb the workbook to close
* @throws IOException
*/
private static void assertCloseDoesNotModifyFile(String filename, Workbook wb) throws IOException {
final byte[] before = HSSFTestDataSamples.getTestDataFileContent(filename);
closeOrRevert(wb);
final byte[] after = HSSFTestDataSamples.getTestDataFileContent(filename);
assertArrayEquals(filename + " sample file was modified as a result of closing the workbook",
before, after);
}
/**
* bug 58779: Closing an XSSFWorkbook that was created with WorkbookFactory modifies the file
* FIXME: replace this method with wb.close() when bug 58779 is resolved.
*
* @param wb
* @throws IOException
*/
private static void revert(Workbook wb) {
private static void closeOrRevert(Workbook wb) throws IOException {
// TODO: close() re-writes the sample-file?! Resort to revert() for now to close file handle...
LOGGER.log(POILogger.WARN,
"reverting XSSFWorkbook rather than closing it to avoid close() modifying the file on disk." +
"This is a separate bug that isn't tested by this unit test.");
if (wb instanceof XSSFWorkbook) {
((XSSFWorkbook) wb).getPackage().revert();
if (wb instanceof HSSFWorkbook) {
wb.close();
}
else if (wb instanceof XSSFWorkbook) {
final XSSFWorkbook xwb = (XSSFWorkbook) wb;
if (PackageAccess.READ == xwb.getPackage().getPackageAccess()) {
xwb.close();
}
else {
LOGGER.log(POILogger.WARN,
"reverting XSSFWorkbook rather than closing it to avoid close() modifying the file on disk. " +
"Refer to bug 58779.");
xwb.getPackage().revert();
}
} else {
throw new RuntimeException("Unsupported workbook type");
}
@ -78,7 +104,7 @@ public final class TestWorkbookFactory {
);
assertNotNull(wb);
assertTrue(wb instanceof HSSFWorkbook);
wb.close();
assertCloseDoesNotModifyFile(xls, wb);
// Package -> xssf
wb = WorkbookFactory.create(
@ -87,8 +113,7 @@ public final class TestWorkbookFactory {
);
assertNotNull(wb);
assertTrue(wb instanceof XSSFWorkbook);
// TODO: this re-writes the sample-file?! wb.close();
revert(wb);
assertCloseDoesNotModifyFile(xlsx, wb);
}
@Test
@ -99,13 +124,13 @@ public final class TestWorkbookFactory {
wb = WorkbookFactory.create(HSSFTestDataSamples.getSampleFile(xls), null, true);
assertNotNull(wb);
assertTrue(wb instanceof HSSFWorkbook);
wb.close();
assertCloseDoesNotModifyFile(xls, wb);
// Package -> xssf
wb = WorkbookFactory.create(HSSFTestDataSamples.getSampleFile(xlsx), null, true);
assertNotNull(wb);
assertTrue(wb instanceof XSSFWorkbook);
wb.close();
assertCloseDoesNotModifyFile(xlsx, wb);
}
/**
@ -123,15 +148,14 @@ public final class TestWorkbookFactory {
);
assertNotNull(wb);
assertTrue(wb instanceof HSSFWorkbook);
wb.close();
assertCloseDoesNotModifyFile(xls, wb);
wb = WorkbookFactory.create(
HSSFTestDataSamples.openSampleFileStream(xlsx)
);
assertNotNull(wb);
assertTrue(wb instanceof XSSFWorkbook);
// TODO: this re-writes the sample-file?! wb.close();
revert(wb);
assertCloseDoesNotModifyFile(xlsx, wb);
// File -> either
wb = WorkbookFactory.create(
@ -139,17 +163,17 @@ public final class TestWorkbookFactory {
);
assertNotNull(wb);
assertTrue(wb instanceof HSSFWorkbook);
wb.close();
assertCloseDoesNotModifyFile(xls, wb);
wb = WorkbookFactory.create(
HSSFTestDataSamples.getSampleFile(xlsx)
);
assertNotNull(wb);
assertTrue(wb instanceof XSSFWorkbook);
// TODO: close() re-writes the sample-file?! Resort to revert() for now to close file handle...
revert(wb);
assertCloseDoesNotModifyFile(xlsx, wb);
// Invalid type -> exception
final byte[] before = HSSFTestDataSamples.getTestDataFileContent(txt);
try {
InputStream stream = HSSFTestDataSamples.openSampleFileStream(txt);
try {
@ -161,6 +185,9 @@ public final class TestWorkbookFactory {
} catch(InvalidFormatException e) {
// Good
}
final byte[] after = HSSFTestDataSamples.getTestDataFileContent(txt);
assertArrayEquals("Invalid type file was modified after trying to open the file as a spreadsheet",
before, after);
}
/**
@ -177,14 +204,14 @@ public final class TestWorkbookFactory {
);
assertNotNull(wb);
assertTrue(wb instanceof HSSFWorkbook);
wb.close();
assertCloseDoesNotModifyFile(xls, wb);
wb = WorkbookFactory.create(
HSSFTestDataSamples.openSampleFileStream(xlsx), null
);
assertNotNull(wb);
assertTrue(wb instanceof XSSFWorkbook);
revert(wb);
assertCloseDoesNotModifyFile(xlsx, wb);
// Unprotected, wrong password, opens normally
@ -193,14 +220,14 @@ public final class TestWorkbookFactory {
);
assertNotNull(wb);
assertTrue(wb instanceof HSSFWorkbook);
wb.close();
assertCloseDoesNotModifyFile(xls, wb);
wb = WorkbookFactory.create(
HSSFTestDataSamples.openSampleFileStream(xlsx), "wrong"
);
assertNotNull(wb);
assertTrue(wb instanceof XSSFWorkbook);
revert(wb);
assertCloseDoesNotModifyFile(xlsx, wb);
// Protected, correct password, opens fine
@ -209,14 +236,14 @@ public final class TestWorkbookFactory {
);
assertNotNull(wb);
assertTrue(wb instanceof HSSFWorkbook);
wb.close();
assertCloseDoesNotModifyFile(xls_prot[0], wb);
wb = WorkbookFactory.create(
HSSFTestDataSamples.openSampleFileStream(xlsx_prot[0]), xlsx_prot[1]
);
assertNotNull(wb);
assertTrue(wb instanceof XSSFWorkbook);
revert(wb);
assertCloseDoesNotModifyFile(xlsx_prot[0], wb);
// Protected, wrong password, throws Exception
@ -224,7 +251,7 @@ public final class TestWorkbookFactory {
wb = WorkbookFactory.create(
HSSFTestDataSamples.openSampleFileStream(xls_prot[0]), "wrong"
);
wb.close();
assertCloseDoesNotModifyFile(xls_prot[0], wb);
fail("Shouldn't be able to open with the wrong password");
} catch (EncryptedDocumentException e) {}
@ -232,7 +259,7 @@ public final class TestWorkbookFactory {
wb = WorkbookFactory.create(
HSSFTestDataSamples.openSampleFileStream(xlsx_prot[0]), "wrong"
);
revert(wb);
assertCloseDoesNotModifyFile(xlsx_prot[0], wb);
fail("Shouldn't be able to open with the wrong password");
} catch (EncryptedDocumentException e) {}
}
@ -250,14 +277,14 @@ public final class TestWorkbookFactory {
);
assertNotNull(wb);
assertTrue(wb instanceof HSSFWorkbook);
wb.close();
assertCloseDoesNotModifyFile(xls, wb);
wb = WorkbookFactory.create(
HSSFTestDataSamples.getSampleFile(xlsx), null
);
assertNotNull(wb);
assertTrue(wb instanceof XSSFWorkbook);
revert(wb);
assertCloseDoesNotModifyFile(xlsx, wb);
// Unprotected, wrong password, opens normally
wb = WorkbookFactory.create(
@ -265,14 +292,14 @@ public final class TestWorkbookFactory {
);
assertNotNull(wb);
assertTrue(wb instanceof HSSFWorkbook);
wb.close();
assertCloseDoesNotModifyFile(xls, wb);
wb = WorkbookFactory.create(
HSSFTestDataSamples.getSampleFile(xlsx), "wrong"
);
assertNotNull(wb);
assertTrue(wb instanceof XSSFWorkbook);
revert(wb);
assertCloseDoesNotModifyFile(xlsx, wb);
// Protected, correct password, opens fine
wb = WorkbookFactory.create(
@ -280,21 +307,21 @@ public final class TestWorkbookFactory {
);
assertNotNull(wb);
assertTrue(wb instanceof HSSFWorkbook);
wb.close();
assertCloseDoesNotModifyFile(xls_prot[0], wb);
wb = WorkbookFactory.create(
HSSFTestDataSamples.getSampleFile(xlsx_prot[0]), xlsx_prot[1]
);
assertNotNull(wb);
assertTrue(wb instanceof XSSFWorkbook);
revert(wb);
assertCloseDoesNotModifyFile(xlsx_prot[0], wb);
// Protected, wrong password, throws Exception
try {
wb = WorkbookFactory.create(
HSSFTestDataSamples.getSampleFile(xls_prot[0]), "wrong"
);
wb.close();
assertCloseDoesNotModifyFile(xls_prot[0], wb);
fail("Shouldn't be able to open with the wrong password");
} catch (EncryptedDocumentException e) {}
@ -302,30 +329,33 @@ public final class TestWorkbookFactory {
wb = WorkbookFactory.create(
HSSFTestDataSamples.getSampleFile(xlsx_prot[0]), "wrong"
);
revert(wb);
assertCloseDoesNotModifyFile(xlsx_prot[0], wb);
fail("Shouldn't be able to open with the wrong password");
} catch (EncryptedDocumentException e) {}
}
/**
* Check that a helpful exception is given on an empty file / stream
* Check that a helpful exception is given on an empty input stream
*/
@Test
public void testEmptyFile() throws Exception {
public void testEmptyInputStream() throws Exception {
InputStream emptyStream = new ByteArrayInputStream(new byte[0]);
File emptyFile = TempFile.createTempFile("empty", ".poi");
try {
WorkbookFactory.create(emptyStream);
fail("Shouldn't be able to create for an empty stream");
} catch (EmptyFileException e) {
}
} catch (final EmptyFileException expected) {}
}
/**
* Check that a helpful exception is given on an empty file
*/
@Test
public void testEmptyFile() throws Exception {
File emptyFile = TempFile.createTempFile("empty", ".poi");
try {
WorkbookFactory.create(emptyFile);
fail("Shouldn't be able to create for an empty file");
} catch (EmptyFileException e) {
}
} catch (final EmptyFileException expected) {}
emptyFile.delete();
}
}