WIP

hydra-node/src/Hydra/Network/Authenticate.hs

@@ -18,6 +18,10 @@ data Authenticated msg = Authenticated
   deriving stock (Eq, Show, Generic)
   deriving anyclass (ToJSON, FromJSON)
 
+instance (Arbitrary msg, SignableRepresentation msg) => Arbitrary (Authenticated msg) where
+  arbitrary = genericArbitrary
+  shrink = genericShrink
+
 instance ToCBOR msg => ToCBOR (Authenticated msg) where
   toCBOR (Authenticated msg sig) = toCBOR msg <> toCBOR sig
 

hydra-node/test/Hydra/Network/AuthenticateSpec.hs

@@ -1,16 +1,20 @@
+{-# LANGUAGE TypeApplications #-}
+
 module Hydra.Network.AuthenticateSpec where
 
 import Hydra.Prelude
 import Test.Hydra.Prelude
 
 import Control.Concurrent.Class.MonadSTM (MonadSTM (readTVarIO), modifyTVar', newTVarIO)
 import Control.Monad.IOSim (runSimOrThrow)
+import Data.ByteString (pack)
 import Hydra.Crypto (sign)
 import Hydra.Network (Network (..))
+import Hydra.Network.Authenticate (Authenticated (Authenticated), withAuthentication)
 import Hydra.Network.HeartbeatSpec (noop)
+import Hydra.NetworkSpec (prop_canRoundtripCBOREncoding)
 import Test.Hydra.Fixture (aliceSk, bob, bobSk)
-import Test.QuickCheck (generate)
-import Hydra.Network.Authenticate (Authenticated (Authenticated), withAuthentication)
+import Test.QuickCheck (generate, listOf, vectorOf)
 
 spec :: Spec
 spec = parallel $ do
@@ -72,3 +76,11 @@ spec = parallel $ do
 
     sentMsgs `shouldBe` [Authenticated "1" (sign signingKey "1")]
 
+  describe "Serialization" $ do
+    prop "can roundtrip CBOR encoding/decoding of Authenticated Hydra Message" $ prop_canRoundtripCBOREncoding @(Authenticated Msg)
+
+newtype Msg = Msg ByteString
+  deriving newtype (Eq, Show, ToCBOR, FromCBOR)
+
+instance Arbitrary Msg where
+  arbitrary = Msg . pack <$> listOf arbitrary