diff --git a/src/scratchpad/testcases/org/apache/poi/hslf/record/TestDocumentEncryption.java b/src/scratchpad/testcases/org/apache/poi/hslf/record/TestDocumentEncryption.java index 37e7e31db..a53173a8a 100644 --- a/src/scratchpad/testcases/org/apache/poi/hslf/record/TestDocumentEncryption.java +++ b/src/scratchpad/testcases/org/apache/poi/hslf/record/TestDocumentEncryption.java @@ -20,6 +20,7 @@ package org.apache.poi.hslf.record; import static org.junit.Assert.assertArrayEquals; import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertNotNull; import static org.junit.Assert.assertTrue; import static org.junit.Assert.fail; @@ -72,7 +73,7 @@ public class TestDocumentEncryption { "Password_Protected-hello.ppt", "Password_Protected-np-hello.ppt", }; - + for (String pptFile : encPpts) { try { NPOIFSFileSystem fs = new NPOIFSFileSystem(slTests.getFile(pptFile), true); @@ -83,6 +84,7 @@ public class TestDocumentEncryption { fail(pptFile+" can't be decrypted"); } } + // password is reset in @After } @Test @@ -94,24 +96,27 @@ public class TestDocumentEncryption { // need to cache data (i.e. read all data) before changing the key size List picsExpected = hss.getPictureData(); hss.getDocumentSummaryInformation(); - EncryptionInfo ei = hss.getDocumentEncryptionAtom().getEncryptionInfo(); - ((CryptoAPIEncryptionHeader)ei.getHeader()).setKeySize(0x78); - + DocumentEncryptionAtom documentEncryptionAtom = hss.getDocumentEncryptionAtom(); + assertNotNull(documentEncryptionAtom); + EncryptionInfo ei = documentEncryptionAtom.getEncryptionInfo(); + ((CryptoAPIEncryptionHeader) ei.getHeader()).setKeySize(0x78); + ByteArrayOutputStream bos = new ByteArrayOutputStream(); hss.write(bos); hss.close(); fs.close(); - + fs = new NPOIFSFileSystem(new ByteArrayInputStream(bos.toByteArray())); hss = new HSLFSlideShowImpl(fs); List picsActual = hss.getPictureData(); - + assertEquals(picsExpected.size(), picsActual.size()); - for (int i=0; i