From a4baf9383b4025ce88f3af18722adf97f579c92d Mon Sep 17 00:00:00 2001
From: Vincent Hanquez <vincent@snarc.org>
Date: Sat, 23 May 2015 11:58:57 +0100
Subject: [PATCH] [test] add random number generation tests

---
 tests/Number.hs | 14 ++++++++++++++
 tests/Tests.hs  |  4 +++-
 tests/Utils.hs  | 10 ++++++++++
 3 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/tests/Number.hs b/tests/Number.hs
index d0ac4a2..0aec4fe 100644
--- a/tests/Number.hs
+++ b/tests/Number.hs
@@ -3,6 +3,7 @@ module Number (tests) where
 import Imports
 
 import Crypto.Number.Basic
+import Crypto.Number.Generate
 import Data.Bits
 
 tests = testGroup "number"
@@ -13,4 +14,17 @@ tests = testGroup "number"
             ]
     , testProperty "num-bits2" $ \(Positive i) ->
         not (i `testBit` numBits i) && (i `testBit` (numBits i - 1))
+    , testProperty "generate-param" $ \testDRG (Positive bits)  ->
+        let r = withTestDRG testDRG $ generateParams bits (Just SetHighest) False
+         in r >= 0 && numBits r == bits && testBit r (bits-1)
+    , testProperty "generate-param2" $ \testDRG (Positive m1bits) ->
+        let bits = m1bits + 1 -- make sure minimum is 2
+            r = withTestDRG testDRG $ generateParams bits (Just SetTwoHighest) False
+         in r >= 0 && numBits r == bits && testBit r (bits-1) && testBit r (bits-2)
+    , testProperty "generate-param-odd" $ \testDRG (Positive bits) ->
+        let r = withTestDRG testDRG $ generateParams bits Nothing True
+         in r >= 0 && odd r
+    , testProperty "generate-range" $ \testDRG (Positive range) ->
+        let r = withTestDRG testDRG $ generateMax range
+         in 0 <= r && r < range
     ]
diff --git a/tests/Tests.hs b/tests/Tests.hs
index 484ae33..f422de0 100644
--- a/tests/Tests.hs
+++ b/tests/Tests.hs
@@ -3,6 +3,7 @@ module Main where
 
 import Imports
 
+import qualified Number
 import qualified Hash
 import qualified Poly1305
 import qualified Salsa
@@ -24,7 +25,8 @@ import qualified KAT_TripleDES
 import qualified KAT_AFIS
 
 tests = testGroup "cryptonite"
-    [ Hash.tests
+    [ Number.tests
+    , Hash.tests
     , testGroup "MAC"
         [ Poly1305.tests
         , KAT_HMAC.tests
diff --git a/tests/Utils.hs b/tests/Utils.hs
index c8eea23..c1881ea 100644
--- a/tests/Utils.hs
+++ b/tests/Utils.hs
@@ -2,12 +2,22 @@ module Utils where
 
 import Control.Monad (replicateM)
 import Data.Char
+import Data.Word
 import Data.ByteString (ByteString)
 import qualified Data.ByteString as B
 import qualified Data.ByteString.Lazy as L
+import Crypto.Random
 
 import Test.Tasty.QuickCheck
 
+newtype TestDRG = TestDRG (Word64, Word64, Word64, Word64, Word64)
+    deriving (Show,Eq)
+
+instance Arbitrary TestDRG where
+    arbitrary = TestDRG `fmap` arbitrary
+
+withTestDRG (TestDRG l) f = fst $ withDRG (drgNewTest l) f
+
 newtype ChunkingLen = ChunkingLen [Int]
     deriving (Show,Eq)