Define CBOR instances for mock Addr

IntersectMBO · Apr 16, 2024 · 62a63d8 · 62a63d8
1 parent cbfbd69
commit 62a63d8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/ouroboros-consensus/ouroboros-consensus.cabal b/ouroboros-consensus/ouroboros-consensus.cabal
@@ -464,6 +464,7 @@ library unstable-mock-block
     ouroboros-network-api,
     ouroboros-network-mock,
     serialise,
+    text,
     time,
     unstable-consensus-testlib,
 

diff --git a/ouroboros-consensus/src/unstable-mock-block/Ouroboros/Consensus/Mock/Ledger/Address.hs b/ouroboros-consensus/src/unstable-mock-block/Ouroboros/Consensus/Mock/Ledger/Address.hs
@@ -6,11 +6,13 @@ module Ouroboros.Consensus.Mock.Ledger.Address (
   , mkAddrDist
   ) where
 
+import           Cardano.Binary (FromCBOR (..), ToCBOR (..))
 import           Codec.Serialise (Serialise)
 import           Control.DeepSeq (NFData)
 import           Data.Map.Strict (Map)
 import qualified Data.Map.Strict as Map
 import           Data.String
+import           Data.Text (pack, unpack)
 import           NoThunks.Class (NoThunks)
 import           Ouroboros.Consensus.Node.ProtocolInfo
 import           Ouroboros.Consensus.NodeId (NodeId (..))
@@ -28,6 +30,12 @@ newtype Addr = Addr String
     , NoThunks
     )
 
+instance ToCBOR Addr where
+  toCBOR (Addr a) = toCBOR $ pack a
+
+instance FromCBOR Addr where
+  fromCBOR = Addr . unpack <$> fromCBOR
+
 instance Condense Addr where
   condense (Addr addr) = addr