hmac-drbg

commit 6811d20dcf29586ee894c6bdacd7567aadc36621
parent ba4c5d3c8726faa7e07f81bcf65d74b782ea53bd
Author: Jared Tobin <jared@jtobin.io>
Date:   Sat, 10 Jan 2026 01:14:27 +0400

test+bench: handle either

Diffstat:
M
bench/Main.hs
 | 
8
++++----
M
test/Main.hs
 | 
4
++--

2 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/bench/Main.hs b/bench/Main.hs
@@ -22,14 +22,14 @@ suite drbg256 drbg512 =
     bgroup "HMAC-SHA256" [
       bench "new" $ whnfAppIO (DRBG.new SHA256.hmac mempty mempty) mempty
     , bench "reseed" $ whnfAppIO (DRBG.reseed mempty mempty) drbg256
-    , bench "gen (32B)"  $ nfAppIO (DRBG.gen mempty 32) drbg256
-    , bench "gen (256B)" $ nfAppIO (DRBG.gen mempty 256) drbg256
+    , bench "gen (32B)"  $ whnfAppIO (DRBG.gen mempty 32) drbg256
+    , bench "gen (256B)" $ whnfAppIO (DRBG.gen mempty 256) drbg256
     ]
   , bgroup "HMAC-SHA512" [
       bench "new" $ whnfAppIO (DRBG.new SHA512.hmac mempty mempty) mempty
     , bench "reseed" $ whnfAppIO (DRBG.reseed mempty mempty) drbg512
-    , bench "gen (32B)"  $ nfAppIO (DRBG.gen mempty 32) drbg512
-    , bench "gen (256B)" $ nfAppIO (DRBG.gen mempty 256) drbg512
+    , bench "gen (32B)"  $ whnfAppIO (DRBG.gen mempty 32) drbg512
+    , bench "gen (256B)" $ whnfAppIO (DRBG.gen mempty 256) drbg512
     ]
   ]
 
diff --git a/test/Main.hs b/test/Main.hs
@@ -103,7 +103,7 @@ execute hmac Case {..} = testCase ("count " <> show caseCount) $ do
   assertEqual "k0" k0 caseK0
 
   DRBG.reseed caseEntropy1 caseAddl1 drbg
-  _ <- DRBG.gen mempty bytes drbg
+  Right _ <- DRBG.gen mempty bytes drbg
   v1 <- DRBG._read_v drbg
   k1 <- DRBG._read_k drbg
 
@@ -111,7 +111,7 @@ execute hmac Case {..} = testCase ("count " <> show caseCount) $ do
   assertEqual "k1" k1 caseK1
 
   DRBG.reseed caseEntropy2 caseAddl2 drbg
-  returned <- DRBG.gen mempty bytes drbg
+  Right returned <- DRBG.gen mempty bytes drbg
   v2 <- DRBG._read_v drbg
   k2 <- DRBG._read_k drbg