You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@poi.apache.org by ni...@apache.org on 2015/04/23 20:01:51 UTC
svn commit: r1675689 -
/poi/trunk/src/testcases/org/apache/poi/hssf/eventusermodel/TestHSSFEventFactory.java
Author: nick
Date: Thu Apr 23 18:01:51 2015
New Revision: 1675689
URL: http://svn.apache.org/r1675689
Log:
Add unit test for Eventmodel processing of password protected excel files
Modified:
poi/trunk/src/testcases/org/apache/poi/hssf/eventusermodel/TestHSSFEventFactory.java
Modified: poi/trunk/src/testcases/org/apache/poi/hssf/eventusermodel/TestHSSFEventFactory.java
URL: http://svn.apache.org/viewvc/poi/trunk/src/testcases/org/apache/poi/hssf/eventusermodel/TestHSSFEventFactory.java?rev=1675689&r1=1675688&r2=1675689&view=diff
==============================================================================
--- poi/trunk/src/testcases/org/apache/poi/hssf/eventusermodel/TestHSSFEventFactory.java (original)
+++ poi/trunk/src/testcases/org/apache/poi/hssf/eventusermodel/TestHSSFEventFactory.java Thu Apr 23 18:01:51 2015
@@ -22,15 +22,19 @@ import java.util.List;
import junit.framework.TestCase;
+import org.apache.poi.EncryptedDocumentException;
import org.apache.poi.hssf.HSSFTestDataSamples;
+import org.apache.poi.hssf.record.BoundSheetRecord;
import org.apache.poi.hssf.record.ContinueRecord;
import org.apache.poi.hssf.record.DVALRecord;
import org.apache.poi.hssf.record.DVRecord;
import org.apache.poi.hssf.record.EOFRecord;
import org.apache.poi.hssf.record.FeatHdrRecord;
+import org.apache.poi.hssf.record.NumberRecord;
import org.apache.poi.hssf.record.Record;
import org.apache.poi.hssf.record.SelectionRecord;
import org.apache.poi.hssf.record.WindowTwoRecord;
+import org.apache.poi.hssf.record.crypto.Biff8EncryptionKey;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
/**
*
@@ -137,4 +141,63 @@ public final class TestHSSFEventFactory
factory = new HSSFEventFactory();
factory.processWorkbookEvents(req, fs);
}
+
+ public void testWithPasswordProtectedWorkbooks() throws Exception {
+ HSSFRequest req = new HSSFRequest();
+ MockHSSFListener mockListen = new MockHSSFListener();
+ req.addListenerForAllRecords(mockListen);
+
+ // Without a password, can't be read
+ Biff8EncryptionKey.setCurrentUserPassword(null);
+ POIFSFileSystem fs = new POIFSFileSystem(openSample("xor-encryption-abc.xls"));
+
+ HSSFEventFactory factory = new HSSFEventFactory();
+ try {
+ factory.processWorkbookEvents(req, fs);
+ fail("Shouldn't be able to process protected workbook without the password");
+ } catch (EncryptedDocumentException e) {}
+
+
+ // With the password, is properly processed
+ Biff8EncryptionKey.setCurrentUserPassword("abc");
+
+ req = new HSSFRequest();
+ mockListen = new MockHSSFListener();
+ req.addListenerForAllRecords(mockListen);
+ factory.processWorkbookEvents(req, fs);
+
+ // Check we got the sheet and the contents
+ Record[] recs = mockListen.getRecords();
+ assertTrue( recs.length > 50 );
+
+ // Has one sheet, with values 1,2,3 in column A rows 1-3
+ boolean hasSheet=false, hasA1=false, hasA2=false, hasA3=false;
+ for (Record r : recs) {
+ if (r instanceof BoundSheetRecord) {
+ BoundSheetRecord bsr = (BoundSheetRecord)r;
+ assertEquals("Sheet1", bsr.getSheetname());
+ hasSheet = true;
+ }
+ if (r instanceof NumberRecord) {
+ NumberRecord nr = (NumberRecord)r;
+ if (nr.getColumn() == 0 && nr.getRow() == 0) {
+ assertEquals(1, (int)nr.getValue());
+ hasA1 = true;
+ }
+ if (nr.getColumn() == 0 && nr.getRow() == 1) {
+ assertEquals(2, (int)nr.getValue());
+ hasA2 = true;
+ }
+ if (nr.getColumn() == 0 && nr.getRow() == 2) {
+ assertEquals(3, (int)nr.getValue());
+ hasA3 = true;
+ }
+ }
+ }
+
+ assertTrue("Sheet record not found", hasSheet);
+ assertTrue("Numeric record for A1 not found", hasA1);
+ assertTrue("Numeric record for A2 not found", hasA2);
+ assertTrue("Numeric record for A3 not found", hasA3);
+ }
}
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@poi.apache.org
For additional commands, e-mail: commits-help@poi.apache.org