{-# LANGUAGE AllowAmbiguousTypes #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TypeApplications #-}
{-# LANGUAGE TypeFamilies #-}

module Test.Cardano.Ledger.Core.Binary (
  decoderEquivalenceSpec,
  decoderEquivalenceEraSpec,
  txSizeSpec,
  decoderEquivalenceCoreEraTypesSpec,
  Mem,
  fullAnnCddlSpec,
  fullAnnGenCddlSpec,
  fullCddlSpec,
  fullGenCddlSpec,
) where

import Cardano.Ledger.Binary (Annotator, DecCBOR, EncCBOR, Version)
import Cardano.Ledger.Core
import Cardano.Ledger.MemoBytes (Mem)
import qualified Data.Text as T
import Lens.Micro
import Test.Cardano.Ledger.Binary (decoderEquivalenceSpec)
import Test.Cardano.Ledger.Binary.Cuddle (
  HuddleEnv,
  huddleAntiCborSpec,
  huddleDecoderEquivalenceSpec,
  huddleRoundTripAnnCborSpec,
  huddleRoundTripCborSpec,
  huddleRoundTripGenValidate,
 )
import Test.Cardano.Ledger.Common
import Test.Cardano.Ledger.Core.Arbitrary ()
import Test.Cardano.Ledger.Core.Binary.Annotator

txSizeSpec ::
  forall era.
  ( EraTx era
  , Arbitrary (Tx TopTx era)
  , SafeToHash (TxWits era)
  ) =>
  Spec
txSizeSpec :: forall era.
(EraTx era, Arbitrary (Tx TopTx era), SafeToHash (TxWits era)) =>
Spec
txSizeSpec =
  String -> Spec -> Spec
forall a. HasCallStack => String -> SpecWith a -> SpecWith a
describe String
"Transaction size" (Spec -> Spec) -> Spec -> Spec
forall a b. (a -> b) -> a -> b
$ do
    String -> (Tx TopTx era -> Expectation) -> Spec
forall prop.
(HasCallStack, Testable prop) =>
String -> prop -> Spec
prop String
"should match the size of the cbor encoding" ((Tx TopTx era -> Expectation) -> Spec)
-> (Tx TopTx era -> Expectation) -> Spec
forall a b. (a -> b) -> a -> b
$ \(Tx TopTx era
tx :: Tx TopTx era) -> do
      let txSize :: Word32
txSize = Tx TopTx era -> Word32
forall era (l :: TxLevel).
(EraTx era, HasCallStack, SafeToHash (TxWits era), Typeable l) =>
Tx l era -> Word32
forall (l :: TxLevel).
(HasCallStack, SafeToHash (TxWits era), Typeable l) =>
Tx l era -> Word32
sizeTxForFeeCalculation Tx TopTx era
tx
      Word32
txSize Word32 -> Word32 -> Expectation
forall a. (HasCallStack, Show a, Eq a) => a -> a -> Expectation
`shouldBe` Tx TopTx era
tx Tx TopTx era -> Getting Word32 (Tx TopTx era) Word32 -> Word32
forall s a. s -> Getting a s a -> a
^. Getting Word32 (Tx TopTx era) Word32
forall era (l :: TxLevel).
(EraTx era, HasCallStack) =>
SimpleGetter (Tx l era) Word32
SimpleGetter (Tx TopTx era) Word32
forall (l :: TxLevel).
HasCallStack =>
SimpleGetter (Tx l era) Word32
sizeTxF

-- | Like 'fullAnnCddlSpec' but with a custom generator.
fullAnnGenCddlSpec ::
  forall a.
  ( Eq a
  , Show a
  , EncCBOR a
  , DecCBOR a
  , DecCBOR (Annotator a)
  , HasCallStack
  ) =>
  Gen a ->
  Version ->
  T.Text ->
  SpecWith HuddleEnv
fullAnnGenCddlSpec :: forall a.
(Eq a, Show a, EncCBOR a, DecCBOR a, DecCBOR (Annotator a),
 HasCallStack) =>
Gen a -> Version -> Text -> SpecWith HuddleEnv
fullAnnGenCddlSpec Gen a
gen Version
version Text
ruleName = do
  forall a.
(Eq a, Show a, EncCBOR a, DecCBOR a, HasCallStack) =>
Gen a -> Version -> Text -> SpecWith HuddleEnv
fullGenCddlSpec @a Gen a
gen Version
version Text
ruleName
  forall a.
(HasCallStack, Eq a, Show a, EncCBOR a, DecCBOR (Annotator a)) =>
Version -> Text -> SpecWith HuddleEnv
huddleRoundTripAnnCborSpec @a Version
version Text
ruleName
  forall a.
(HasCallStack, Eq a, Show a, DecCBOR a, DecCBOR (Annotator a)) =>
Version -> Text -> SpecWith HuddleEnv
huddleDecoderEquivalenceSpec @a Version
version Text
ruleName

-- | Full CDDL codec spec for types with both plain and Annotator decoders.
fullAnnCddlSpec ::
  forall a.
  ( Eq a
  , Show a
  , Arbitrary a
  , EncCBOR a
  , DecCBOR a
  , DecCBOR (Annotator a)
  , HasCallStack
  ) =>
  Version ->
  T.Text ->
  SpecWith HuddleEnv
fullAnnCddlSpec :: forall a.
(Eq a, Show a, Arbitrary a, EncCBOR a, DecCBOR a,
 DecCBOR (Annotator a), HasCallStack) =>
Version -> Text -> SpecWith HuddleEnv
fullAnnCddlSpec = forall a.
(Eq a, Show a, EncCBOR a, DecCBOR a, DecCBOR (Annotator a),
 HasCallStack) =>
Gen a -> Version -> Text -> SpecWith HuddleEnv
fullAnnGenCddlSpec @a (forall a. Arbitrary a => Gen a
arbitrary @a)

-- | Like 'fullCddlSpec' but with a custom generator.
fullGenCddlSpec ::
  forall a.
  ( Eq a
  , Show a
  , EncCBOR a
  , DecCBOR a
  , HasCallStack
  ) =>
  Gen a ->
  Version ->
  T.Text ->
  SpecWith HuddleEnv
fullGenCddlSpec :: forall a.
(Eq a, Show a, EncCBOR a, DecCBOR a, HasCallStack) =>
Gen a -> Version -> Text -> SpecWith HuddleEnv
fullGenCddlSpec Gen a
gen Version
version Text
ruleName = do
  forall a.
(HasCallStack, Eq a, Show a, EncCBOR a, DecCBOR a) =>
Version -> Text -> SpecWith HuddleEnv
huddleRoundTripCborSpec @a Version
version Text
ruleName
  forall a.
(DecCBOR a, Show a, EncCBOR a) =>
Gen a -> Version -> Text -> SpecWith HuddleEnv
huddleRoundTripGenValidate @a Gen a
gen Version
version Text
ruleName
  forall a. DecCBOR a => Version -> Text -> SpecWith HuddleEnv
huddleAntiCborSpec @a Version
version Text
ruleName

-- | CDDL codec spec for types with only plain decoders (no Annotator).
fullCddlSpec ::
  forall a.
  ( Eq a
  , Show a
  , Arbitrary a
  , EncCBOR a
  , DecCBOR a
  , HasCallStack
  ) =>
  Version ->
  T.Text ->
  SpecWith HuddleEnv
fullCddlSpec :: forall a.
(Eq a, Show a, Arbitrary a, EncCBOR a, DecCBOR a, HasCallStack) =>
Version -> Text -> SpecWith HuddleEnv
fullCddlSpec = forall a.
(Eq a, Show a, EncCBOR a, DecCBOR a, HasCallStack) =>
Gen a -> Version -> Text -> SpecWith HuddleEnv
fullGenCddlSpec @a (forall a. Arbitrary a => Gen a
arbitrary @a)