1 files changed, 119 insertions, 0 deletions
diff --git a/libraries/spongycastle/core/src/test/java/org/spongycastle/util/encoders/test/UrlBase64Test.java b/libraries/spongycastle/core/src/test/java/org/spongycastle/util/encoders/test/UrlBase64Test.java
new file mode 100644
index 000000000..9822efea9
--- /dev/null
+++ b/libraries/spongycastle/core/src/test/java/org/spongycastle/util/encoders/test/UrlBase64Test.java
@@ -0,0 +1,119 @@
+package org.spongycastle.util.encoders.test;
+
+import java.io.IOException;
+
+import org.spongycastle.util.Arrays;
+import org.spongycastle.util.Strings;
+import org.spongycastle.util.encoders.DecoderException;
+import org.spongycastle.util.encoders.Hex;
+import org.spongycastle.util.encoders.UrlBase64;
+import org.spongycastle.util.encoders.UrlBase64Encoder;
+
+public class UrlBase64Test extends AbstractCoderTest
+{
+    private static final String sample1 = "mO4TyLWG7vjFWdKT8IJcVbZ_jwc.";
+    private static final byte[] sample1Bytes = Hex.decode("98ee13c8b586eef8c559d293f0825c55b67f8f07");
+    private static final String sample2 = "F4I4p8Vf_mS-Kxvri3FPoMcqmJ1f";
+    private static final byte[] sample2Bytes = Hex.decode("178238a7c55ffe64be2b1beb8b714fa0c72a989d5f");
+    private static final String sample3 = "UJmEdJYodqHJmd7Rtv6_OP29_jUEFw..";
+    private static final byte[] sample3Bytes = Hex.decode("50998474962876a1c999ded1b6febf38fdbdfe350417");
+
+    private static final String invalid1 = "%O4TyLWG7vjFWdKT8IJcVbZ_jwc.";
+    private static final String invalid2 = "F%I4p8Vf_mS-Kxvri3FPoMcqmJ1f";
+    private static final String invalid3 = "UJ%EdJYodqHJmd7Rtv6_OP29_jUEFw..";
+    private static final String invalid4 = "mO4%yLWG7vjFWdKT8IJcVbZ_jwc.";
+    private static final String invalid5 = "UJmEdJYodqHJmd7Rtv6_OP29_jUEF%..";
+    private static final String invalid6 = "mO4TyLWG7vjFWdKT8IJcVbZ_jw%.";
+    private static final String invalid7 = "F4I4p8Vf_mS-Kxvri3FPoMcqmJ1%";
+    private static final String invalid8 = "UJmEdJYodqHJmd7Rtv6_OP29_jUE%c..";
+    private static final String invalid9 = "mO4TyLWG7vjFWdKT8IJcVbZ_j%c.";
+    private static final String invalida = "F4I4p8Vf_mS-Kxvri3FPoMcqmJ%1";
+    private static final String invalidb = "UJmEdJYodqHJmd7Rtv6_OP29_jU%Fc..";
+    private static final String invalidc = "mO4TyLWG7vjFWdKT8IJcVbZ_%wc.";
+    private static final String invalidd = "F4I4p8Vf_mS-Kxvri3FPoMcqm%1c";
+
+    public UrlBase64Test(
+        String name)
+    {
+        super(name);
+    }
+    
+    protected void setUp()
+    {
+        super.setUp();
+        enc = new UrlBase64Encoder();
+    }
+
+    public void testSamples()
+        throws IOException
+    {
+        assertTrue(Arrays.areEqual(sample1Bytes, UrlBase64.decode(sample1)));
+        assertTrue(Arrays.areEqual(sample1Bytes, UrlBase64.decode(Strings.toByteArray(sample1))));
+        assertTrue(Arrays.areEqual(sample2Bytes, UrlBase64.decode(sample2)));
+        assertTrue(Arrays.areEqual(sample2Bytes, UrlBase64.decode(Strings.toByteArray(sample2))));
+        assertTrue(Arrays.areEqual(sample3Bytes, UrlBase64.decode(sample3)));
+        assertTrue(Arrays.areEqual(sample3Bytes, UrlBase64.decode(Strings.toByteArray(sample3))));
+    }
+
+    public void testInvalidInput()
+        throws IOException
+    {
+        String[] invalid = new String[] { invalid1, invalid2, invalid3, invalid4, invalid5, invalid6, invalid7, invalid8, invalid9, invalida, invalidb, invalidc, invalidd };
+
+        for (int i = 0; i != invalid.length; i++)
+        {
+            invalidTest(invalid[i]);
+            invalidTest(Strings.toByteArray(invalid[i]));
+        }
+    }
+
+    private void invalidTest(String data)
+    {
+        try
+        {
+            UrlBase64.decode(data);
+        }
+        catch (DecoderException e)
+        {
+            return;
+        }
+
+        fail("invalid String data parsed");
+    }
+
+    private void invalidTest(byte[] data)
+    {
+        try
+        {
+            UrlBase64.decode(data);
+        }
+        catch (DecoderException e)
+        {
+            return;
+        }
+
+        fail("invalid byte data parsed");
+    }
+
+    protected char paddingChar()
+    {
+        return '.';
+    }
+
+    protected boolean isEncodedChar(char c)
+    {
+        if (Character.isLetterOrDigit(c))
+        {
+            return true;
+        }
+        else if (c == '-')
+        {
+            return true;
+        }
+        else if (c == '_')
+        {
+            return true;
+        }
+        return false;
+    }
+}