Fix mlocked memory leak in tests

IntersectMBO · Jan 19, 2023 · 4c0f13b · 4c0f13b
1 parent cb5b787
commit 4c0f13b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cardano-crypto-tests/src/Test/Crypto/KES.hs b/cardano-crypto-tests/src/Test/Crypto/KES.hs
@@ -315,7 +315,10 @@ testKESAlgorithm lock _pm _pv n =
       , testProperty "SignKey DirectDeserialise" $
           ioPropertyWithSK @v lock $ \sk -> do
             direct <- directSerialiseToBS (fromIntegral $ sizeSignKeyKES (Proxy @v)) sk
-            prop_no_thunks_IO (directDeserialiseFromBS @(SignKeyKES v) $! direct)
+            bracket
+              (directDeserialiseFromBS @(SignKeyKES v) $! direct)
+              forgetSignKeyKES
+              (prop_no_thunks_IO . return)
       ]
 
     , testProperty "same VerKey "  $ prop_deriveVerKeyKES (Proxy @IO) (Proxy @v)