1 files changed, 68 insertions, 0 deletions
diff --git a/prov/src/test/java/org/spongycastle/pqc/jcajce/provider/test/FlexiTest.java b/prov/src/test/java/org/spongycastle/pqc/jcajce/provider/test/FlexiTest.java
new file mode 100644
index 00000000..c076091e
--- /dev/null
+++ b/prov/src/test/java/org/spongycastle/pqc/jcajce/provider/test/FlexiTest.java
@@ -0,0 +1,68 @@
+package org.spongycastle.pqc.jcajce.provider.test;
+
+import java.security.SecureRandom;
+import java.security.Security;
+import java.util.Arrays;
+import java.util.Random;
+
+import junit.framework.TestCase;
+import org.spongycastle.pqc.jcajce.provider.BouncyCastlePQCProvider;
+
+public abstract class FlexiTest
+    extends TestCase
+{
+
+    /**
+     * Source of randomness
+     */
+    protected Random rand;
+
+    /**
+     * Secure source of randomness
+     */
+    protected SecureRandom sr;
+
+    protected void setUp()
+    {
+        Security.addProvider(new BouncyCastlePQCProvider());
+        // initialize sources of randomness
+        rand = new Random();
+        sr = new SecureRandom();
+        // TODO need it?
+        sr.setSeed(sr.generateSeed(20));
+    }
+
+    protected static final void assertEquals(byte[] expected, byte[] actual)
+    {
+        assertTrue(Arrays.equals(expected, actual));
+    }
+
+    protected static final void assertEquals(String message, byte[] expected,
+                                             byte[] actual)
+    {
+        assertTrue(message, Arrays.equals(expected, actual));
+    }
+
+    protected static final void assertEquals(int[] expected, int[] actual)
+    {
+        assertTrue(Arrays.equals(expected, actual));
+    }
+
+    protected static final void assertEquals(String message, int[] expected,
+                                             int[] actual)
+    {
+        assertTrue(message, Arrays.equals(expected, actual));
+    }
+
+    /**
+     * Method used to report test failure when in exception is thrown.
+     *
+     * @param e the exception
+     */
+    protected static final void fail(Exception e)
+    {
+        fail("Exception thrown: " + e.getClass().getName() + ":\n"
+            + e.getMessage());
+    }
+
+}