Fixes codahale#15 generator test uses random source

simbo1905 · Jun 15, 2019 · b4a268f · b4a268f
1 parent f44e1ce
commit b4a268f
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/test/java/com/codahale/shamir/GF256Test.java b/src/test/java/com/codahale/shamir/GF256Test.java
@@ -88,9 +88,22 @@ void eval() {
     assertThat(GF256.eval(new byte[] {1, 0, 2, 3}, (byte) 2)).isEqualTo((byte) 17);
   }
 
+  private static class ZeroLastByteFirstAttemptRandom extends SecureRandom {
+    private int counter = 2;
+
+    @Override
+    public void nextBytes(byte[] b) {
+      super.nextBytes(b);
+      if (counter > 0) {
+        b[b.length - 1] = 0;
+        counter--;
+      }
+    }
+  }
+
   @Test
   void generate() {
-    final SecureRandom random = new SecureRandom();
+    final SecureRandom random = new ZeroLastByteFirstAttemptRandom();
     final byte[] p = GF256.generate(random, 5, (byte) 20);
     assertThat(p[0]).isEqualTo((byte) 20);
     assertThat(p.length).isEqualTo(6);