1 files changed, 91 insertions, 0 deletions
diff --git a/pkix/src/test/java/org/spongycastle/tsp/test/CMSTimeStampedDataParserTest.java b/pkix/src/test/java/org/spongycastle/tsp/test/CMSTimeStampedDataParserTest.java
new file mode 100644
index 00000000..5fa14db7
--- /dev/null
+++ b/pkix/src/test/java/org/spongycastle/tsp/test/CMSTimeStampedDataParserTest.java
@@ -0,0 +1,91 @@
+package org.spongycastle.tsp.test;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.InputStream;
+
+import junit.framework.TestCase;
+import org.spongycastle.operator.DigestCalculator;
+import org.spongycastle.operator.DigestCalculatorProvider;
+import org.spongycastle.operator.bc.BcDigestCalculatorProvider;
+import org.spongycastle.tsp.TimeStampToken;
+import org.spongycastle.tsp.cms.CMSTimeStampedDataParser;
+import org.spongycastle.util.io.Streams;
+
+public class CMSTimeStampedDataParserTest
+    extends TestCase
+{
+
+    CMSTimeStampedDataParser cmsTimeStampedData = null;
+    String fileInput = "FileDaFirmare.txt.tsd.der";
+    private byte[] baseData;
+
+    protected void setUp()
+        throws Exception
+    {
+        ByteArrayOutputStream origStream = new ByteArrayOutputStream();
+        InputStream in = this.getClass().getResourceAsStream(fileInput);
+        int ch;
+
+        while ((ch = in.read()) >= 0)
+        {
+            origStream.write(ch);
+        }
+
+        origStream.close();
+
+        this.baseData = origStream.toByteArray();
+
+        cmsTimeStampedData = new CMSTimeStampedDataParser(baseData);
+    }
+
+    protected void tearDown()
+        throws Exception
+    {
+        cmsTimeStampedData = null;
+    }
+
+    public void testGetTimeStampTokens()
+        throws Exception
+    {
+        TimeStampToken[] tokens = cmsTimeStampedData.getTimeStampTokens();
+        assertEquals(3, tokens.length);
+    }
+
+    public void testValidateAllTokens()
+        throws Exception
+    {
+        DigestCalculatorProvider digestCalculatorProvider = new BcDigestCalculatorProvider();
+        ByteArrayOutputStream bOut = new ByteArrayOutputStream();
+
+        Streams.pipeAll(cmsTimeStampedData.getContent(), bOut);
+
+        DigestCalculator imprintCalculator = cmsTimeStampedData.getMessageImprintDigestCalculator(digestCalculatorProvider);
+
+        Streams.pipeAll(new ByteArrayInputStream(bOut.toByteArray()), imprintCalculator.getOutputStream());
+
+        byte[] digest = imprintCalculator.getDigest();
+
+        TimeStampToken[] tokens = cmsTimeStampedData.getTimeStampTokens();
+        for (int i = 0; i < tokens.length; i++)
+        {
+            cmsTimeStampedData.validate(digestCalculatorProvider, digest, tokens[i]);
+        }
+    }
+
+    public void testValidate()
+        throws Exception
+    {
+        DigestCalculatorProvider digestCalculatorProvider = new BcDigestCalculatorProvider();
+        ByteArrayOutputStream bOut = new ByteArrayOutputStream();
+
+        Streams.pipeAll(cmsTimeStampedData.getContent(), bOut);
+
+        DigestCalculator imprintCalculator = cmsTimeStampedData.getMessageImprintDigestCalculator(digestCalculatorProvider);
+
+        Streams.pipeAll(new ByteArrayInputStream(bOut.toByteArray()), imprintCalculator.getOutputStream());
+
+        cmsTimeStampedData.validate(digestCalculatorProvider, imprintCalculator.getDigest());
+    }
+
+}