sig-submission test: integrate validateSig

Author: crocodile-dentist

dmq-node/dmq-node.cabal

@@ -179,20 +179,25 @@ test-suite dmq-tests
   build-depends:
     QuickCheck,
     base >=4.14 && <4.22,
+    binary,
     bytestring,
     cardano-crypto-class,
     cardano-crypto-tests,
+    cardano-ledger-core,
     cborg,
+    containers,
     contra-tracer,
     dmq-node,
-    io-classes,
+    io-classes:{io-classes, strict-stm},
     io-sim,
     kes-agent-crypto,
     ouroboros-network:{api, framework, protocols, protocols-tests-lib, tests-lib},
     quickcheck-instances,
     serialise,
     tasty,
     tasty-quickcheck,
+    time,
+    transformers,
     typed-protocols:{typed-protocols, codec-properties},
     with-utf8,
 

dmq-node/test/DMQ/Protocol/SigSubmission/Test.hs

@@ -26,12 +26,18 @@ module DMQ.Protocol.SigSubmission.Test (tests) where
 import Codec.CBOR.Encoding qualified as CBOR
 import Codec.CBOR.Read qualified as CBOR
 import Codec.CBOR.Write qualified as CBOR
+import Control.Concurrent.Class.MonadSTM.Strict
 import Control.Monad (zipWithM, (>=>))
 import Control.Monad.ST (runST)
+import Control.Monad.Trans.Except
 import Data.Bifunctor (second)
+import Data.Binary qualified as Binary
 import Data.ByteString (ByteString)
+import Data.ByteString qualified as BS
 import Data.ByteString.Lazy qualified as BL
 import Data.List.NonEmpty qualified as NonEmpty
+import Data.Map qualified as Map
+import Data.Time.Clock.POSIX (posixSecondsToUTCTime)
 import Data.Typeable
 import Data.Word (Word32)
 import GHC.TypeNats (KnownNat)
@@ -46,16 +52,20 @@ import Cardano.Crypto.DSIGN.Class qualified as DSIGN
 import Cardano.Crypto.KES.Class (KESAlgorithm (..), VerKeyKES, encodeSigKES)
 import Cardano.Crypto.KES.Class qualified as KES
 import Cardano.Crypto.PinnedSizedBytes (PinnedSizedBytes, psbToByteString)
+import Cardano.Crypto.Hash (castHash, hashWith)
 import Cardano.Crypto.Seed (mkSeedFromBytes)
+import Cardano.Ledger.Hashes (KeyHash (..))
 import Cardano.KESAgent.KES.Crypto (Crypto (..))
 import Cardano.KESAgent.KES.Evolution qualified as KES
 import Cardano.KESAgent.KES.OCert (OCert (..))
 import Cardano.KESAgent.KES.OCert qualified as KES
 import Cardano.KESAgent.Protocols.StandardCrypto (MockCrypto, StandardCrypto)
 import Test.Crypto.Instances
 
+import DMQ.Diffusion.NodeKernel (PoolValidationCtx(..))
 import DMQ.Protocol.SigSubmission.Codec
 import DMQ.Protocol.SigSubmission.Type
+import DMQ.Protocol.SigSubmission.Validate
 
 import Ouroboros.Network.Protocol.TxSubmission2.Test (labelMsg)
 
@@ -830,15 +840,20 @@ prop_validateSig
      )
   => WithConstrKES size kesCrypt (Sig crypto)
   -> Property
-prop_validateSig constr = ioProperty $ do
+prop_validateSig constr = ioProperty do
     sig <- runWithConstr constr
-    return $ case validateSig KES.defEvolutionConfig sig of
+    countersVar <- newTVarIO Map.empty
+    let validationCtx =
+          DMQPoolValidationCtx (posixSecondsToUTCTime 0) Nothing Map.empty countersVar
+        dummyHash = KeyHash . castHash . hashWith (BS.toStrict . Binary.encode . const (0 :: Int))
+    result <- runExceptT $ validateSig KES.defEvolutionConfig dummyHash [sig] validationCtx
+    return case result of
       Left err -> counterexample ("KES seed: " ++ show (ctx constr))
                 . counterexample ("KES vk key: " ++ show (ocertVkHot . getSigOpCertificate . sigOpCertificate $ sig))
                 . counterexample (show sig)
                 . counterexample (show err)
                 $ False
-      Right () -> property True
+      Right {} -> property True
 
 prop_validateSig_mockcrypto
   :: Blind (WithConstrKES (SeedSizeKES (KES MockCrypto)) (KES MockCrypto) (Sig MockCrypto))