1 files changed, 143 insertions, 0 deletions
diff --git a/libraries/spongycastle/pg/src/test/java/org/spongycastle/openpgp/test/PGPCompressionTest.java b/libraries/spongycastle/pg/src/test/java/org/spongycastle/openpgp/test/PGPCompressionTest.java
new file mode 100644
index 000000000..6b884f901
--- /dev/null
+++ b/libraries/spongycastle/pg/src/test/java/org/spongycastle/openpgp/test/PGPCompressionTest.java
@@ -0,0 +1,143 @@
+package org.spongycastle.openpgp.test;
+
+import org.spongycastle.jce.provider.BouncyCastleProvider;
+import org.spongycastle.openpgp.PGPCompressedData;
+import org.spongycastle.openpgp.PGPCompressedDataGenerator;
+import org.spongycastle.openpgp.PGPException;
+import org.spongycastle.openpgp.PGPObjectFactory;
+import org.spongycastle.util.test.SimpleTest;
+import org.spongycastle.util.test.UncloseableOutputStream;
+
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.security.Security;
+
+public class PGPCompressionTest 
+    extends SimpleTest
+{
+    public void performTest()
+        throws Exception
+    {
+        testCompression(PGPCompressedData.UNCOMPRESSED);
+        testCompression(PGPCompressedData.ZIP);
+        testCompression(PGPCompressedData.ZLIB);
+        testCompression(PGPCompressedData.BZIP2);
+
+        //
+        // new style - using stream close
+        //
+        ByteArrayOutputStream bOut = new ByteArrayOutputStream();
+        PGPCompressedDataGenerator cPacket = new PGPCompressedDataGenerator(
+                PGPCompressedData.ZIP);
+
+        OutputStream out = cPacket.open(new UncloseableOutputStream(bOut), new byte[4]);
+
+        out.write("hello world! !dlrow olleh".getBytes());
+
+        out.close();
+
+        validateData(bOut.toByteArray());
+
+        try
+        {
+            out.close();
+            cPacket.close();
+        }
+        catch (Exception e)
+        {
+            fail("Redundant close() should be ignored");
+        }
+
+        //
+        // new style - using generator close
+        //
+        bOut = new ByteArrayOutputStream();
+        cPacket = new PGPCompressedDataGenerator(
+                PGPCompressedData.ZIP);
+
+        out = cPacket.open(new UncloseableOutputStream(bOut), new byte[4]);
+
+        out.write("hello world! !dlrow olleh".getBytes());
+
+        cPacket.close();
+
+        validateData(bOut.toByteArray());
+
+        try
+        {
+            out.close();
+            cPacket.close();
+        }
+        catch (Exception e)
+        {
+            fail("Redundant close() should be ignored");
+        }
+    }
+
+    private void validateData(byte[] data)
+        throws IOException, PGPException
+    {
+        PGPObjectFactory pgpFact = new PGPObjectFactory(data);
+        PGPCompressedData c1 = (PGPCompressedData)pgpFact.nextObject();
+        InputStream pIn = c1.getDataStream();
+
+        ByteArrayOutputStream bOut = new ByteArrayOutputStream();
+
+        int ch;
+        while ((ch = pIn.read()) >= 0)
+        {
+            bOut.write(ch);
+        }
+
+        if (!areEqual(bOut.toByteArray(), "hello world! !dlrow olleh".getBytes()))
+        {
+            fail("compression test failed");
+        }
+    }
+
+    private void testCompression(
+        int type)
+        throws IOException, PGPException
+    {
+        ByteArrayOutputStream bOut = new ByteArrayOutputStream();
+        PGPCompressedDataGenerator cPacket = new PGPCompressedDataGenerator(type);
+
+        OutputStream out = cPacket.open(new UncloseableOutputStream(bOut));
+
+        out.write("hello world!".getBytes());
+
+        out.close();
+
+        PGPObjectFactory pgpFact = new PGPObjectFactory(bOut.toByteArray());
+        PGPCompressedData c1 = (PGPCompressedData)pgpFact.nextObject();
+        InputStream pIn = c1.getDataStream();
+
+        bOut.reset();
+
+        int ch;
+        while ((ch = pIn.read()) >= 0)
+        {
+            bOut.write(ch);
+        }
+
+        if (!areEqual(bOut.toByteArray(), "hello world!".getBytes()))
+        {
+            fail("compression test failed");
+        }
+    }
+
+    public String getName()
+    {
+        return "PGPCompressionTest";
+    }
+
+    public static void main(
+        String[]    args)
+    {
+        Security.addProvider(new BouncyCastleProvider());
+
+        runTest(new PGPCompressionTest());
+    }
+}