{-# LANGUAGE DataKinds #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE RankNTypes #-}
{-# LANGUAGE TypeApplications #-}

module Test.Cardano.Ledger.Mary.Translation (
  maryTranslationTests,
  maryEncodeDecodeTests,
) where

import Cardano.Ledger.Allegra (AllegraEra)
import Cardano.Ledger.Binary
import Cardano.Ledger.Core
import Cardano.Ledger.Genesis (NoGenesis (..))
import Cardano.Ledger.Mary (MaryEra)
import Cardano.Ledger.Mary.Translation ()
import qualified Cardano.Ledger.Shelley.API as S
import Test.Cardano.Ledger.AllegraEraGen ()
import Test.Cardano.Ledger.Binary.RoundTrip
import Test.Cardano.Ledger.Mary.Arbitrary ()
import Test.Cardano.Ledger.Mary.Binary.Annotator ()
import Test.Cardano.Ledger.Shelley.Generator.ShelleyEraGen ()
import Test.Cardano.Ledger.Shelley.Serialisation.EraIndepGenerators ()
import Test.Cardano.Ledger.Shelley.Serialisation.Generators ()
import Test.Cardano.Ledger.ShelleyMA.Serialisation.Generators ()
import Test.Cardano.Ledger.TranslationTools (translateEraEncCBOR, translateEraEncoding)
import Test.Tasty (TestTree, testGroup)
import Test.Tasty.HUnit (Assertion)
import Test.Tasty.QuickCheck (testProperty)

maryEncodeDecodeTests :: TestTree
maryEncodeDecodeTests :: TestTree
maryEncodeDecodeTests =
  TestName -> [TestTree] -> TestTree
testGroup
    TestName
"encoded allegra types can be decoded as mary types"
    [ TestName -> (AllegraTxAuxData AllegraEra -> Assertion) -> TestTree
forall a. Testable a => TestName -> a -> TestTree
testProperty
        TestName
"decoding metadata (Annotator)"
        ( forall a b.
(ToCBOR a, DecCBOR (Annotator b), HasCallStack) =>
Version -> Version -> (b -> a -> Assertion) -> a -> Assertion
embedTripAnnExpectation @(TxAuxData AllegraEra) @(TxAuxData MaryEra)
            (forall era. Era era => Version
eraProtVerLow @AllegraEra)
            (forall era. Era era => Version
eraProtVerLow @MaryEra)
            (\TxAuxData MaryEra
_ TxAuxData AllegraEra
_ -> () -> Assertion
forall a. a -> IO a
forall (f :: * -> *) a. Applicative f => a -> f a
pure ())
        )
    , TestName -> (AllegraTxAuxData AllegraEra -> Assertion) -> TestTree
forall a. Testable a => TestName -> a -> TestTree
testProperty
        TestName
"decoding metadata"
        ( forall a b.
(Typeable b, Eq b, HasCallStack) =>
Version
-> Version -> Trip a b -> (b -> a -> Assertion) -> a -> Assertion
embedTripExpectation @(TxAuxData AllegraEra) @(TxAuxData MaryEra)
            (forall era. Era era => Version
eraProtVerLow @AllegraEra)
            (forall era. Era era => Version
eraProtVerLow @MaryEra)
            Trip (AllegraTxAuxData AllegraEra) (AllegraTxAuxData MaryEra)
Trip (TxAuxData AllegraEra) (TxAuxData MaryEra)
forall a b. (EncCBOR a, DecCBOR b) => Trip a b
cborTrip
            (\TxAuxData MaryEra
_ TxAuxData AllegraEra
_ -> () -> Assertion
forall a. a -> IO a
forall (f :: * -> *) a. Applicative f => a -> f a
pure ())
        )
    ]

maryTranslationTests :: TestTree
maryTranslationTests :: TestTree
maryTranslationTests =
  TestName -> [TestTree] -> TestTree
testGroup
    TestName
"Mary translation binary compatibiliby tests"
    [ TestName
-> (ShelleyTx (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a. Testable a => TestName -> a -> TestTree
testProperty TestName
"Tx compatibility" ((ShelleyTx (PreviousEra MaryEra) -> Assertion) -> TestTree)
-> (ShelleyTx (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a b. (a -> b) -> a -> b
$
        forall era (f :: * -> *).
(HasCallStack, TranslateEra era f,
 Show (TranslationError era f)) =>
TranslationContext era
-> (f era -> Encoding)
-> (f (PreviousEra era) -> Encoding)
-> f (PreviousEra era)
-> Assertion
translateEraEncoding @MaryEra @S.ShelleyTx TranslationContext MaryEra
NoGenesis MaryEra
forall era. NoGenesis era
NoGenesis ShelleyTx MaryEra -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR ShelleyTx AllegraEra -> Encoding
ShelleyTx (PreviousEra MaryEra) -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR
    , TestName -> (ProposedPPUpdates AllegraEra -> Assertion) -> TestTree
forall a. Testable a => TestName -> a -> TestTree
testProperty TestName
"ProposedPPUpdates compatibility" (forall (f :: * -> *).
(EncCBOR (f AllegraEra), EncCBOR (f MaryEra),
 TranslateEra MaryEra f, Show (TranslationError MaryEra f)) =>
f AllegraEra -> Assertion
test @S.ProposedPPUpdates)
    , TestName
-> (ShelleyGovState (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a. Testable a => TestName -> a -> TestTree
testProperty TestName
"ShelleyGovState compatibility" ((ShelleyGovState (PreviousEra MaryEra) -> Assertion) -> TestTree)
-> (ShelleyGovState (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a b. (a -> b) -> a -> b
$
        forall era (f :: * -> *).
(HasCallStack, TranslateEra era f,
 Show (TranslationError era f)) =>
TranslationContext era
-> (f era -> Encoding)
-> (f (PreviousEra era) -> Encoding)
-> f (PreviousEra era)
-> Assertion
translateEraEncoding @MaryEra @S.ShelleyGovState TranslationContext MaryEra
NoGenesis MaryEra
forall era. NoGenesis era
NoGenesis ShelleyGovState MaryEra -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR ShelleyGovState AllegraEra -> Encoding
ShelleyGovState (PreviousEra MaryEra) -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR
    , TestName -> (ShelleyTxOut AllegraEra -> Assertion) -> TestTree
forall a. Testable a => TestName -> a -> TestTree
testProperty TestName
"TxOut compatibility" (forall (f :: * -> *).
(EncCBOR (f AllegraEra), EncCBOR (f MaryEra),
 TranslateEra MaryEra f, Show (TranslationError MaryEra f)) =>
f AllegraEra -> Assertion
test @S.ShelleyTxOut)
    , TestName -> (UTxO (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a. Testable a => TestName -> a -> TestTree
testProperty TestName
"UTxO compatibility" ((UTxO (PreviousEra MaryEra) -> Assertion) -> TestTree)
-> (UTxO (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a b. (a -> b) -> a -> b
$
        forall era (f :: * -> *).
(HasCallStack, TranslateEra era f,
 Show (TranslationError era f)) =>
TranslationContext era
-> (f era -> Encoding)
-> (f (PreviousEra era) -> Encoding)
-> f (PreviousEra era)
-> Assertion
translateEraEncoding @MaryEra @S.UTxO TranslationContext MaryEra
NoGenesis MaryEra
forall era. NoGenesis era
NoGenesis UTxO MaryEra -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR UTxO AllegraEra -> Encoding
UTxO (PreviousEra MaryEra) -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR
    , TestName
-> (UTxOState (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a. Testable a => TestName -> a -> TestTree
testProperty TestName
"UTxOState compatibility" ((UTxOState (PreviousEra MaryEra) -> Assertion) -> TestTree)
-> (UTxOState (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a b. (a -> b) -> a -> b
$
        forall era (f :: * -> *).
(HasCallStack, TranslateEra era f,
 Show (TranslationError era f)) =>
TranslationContext era
-> (f era -> Encoding)
-> (f (PreviousEra era) -> Encoding)
-> f (PreviousEra era)
-> Assertion
translateEraEncoding @MaryEra @S.UTxOState TranslationContext MaryEra
NoGenesis MaryEra
forall era. NoGenesis era
NoGenesis UTxOState MaryEra -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR UTxOState AllegraEra -> Encoding
UTxOState (PreviousEra MaryEra) -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR
    , TestName
-> (LedgerState (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a. Testable a => TestName -> a -> TestTree
testProperty TestName
"LedgerState compatibility" ((LedgerState (PreviousEra MaryEra) -> Assertion) -> TestTree)
-> (LedgerState (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a b. (a -> b) -> a -> b
$
        forall era (f :: * -> *).
(HasCallStack, TranslateEra era f,
 Show (TranslationError era f)) =>
TranslationContext era
-> (f era -> Encoding)
-> (f (PreviousEra era) -> Encoding)
-> f (PreviousEra era)
-> Assertion
translateEraEncoding @MaryEra @S.LedgerState TranslationContext MaryEra
NoGenesis MaryEra
forall era. NoGenesis era
NoGenesis LedgerState MaryEra -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR LedgerState AllegraEra -> Encoding
LedgerState (PreviousEra MaryEra) -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR
    , TestName
-> (EpochState (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a. Testable a => TestName -> a -> TestTree
testProperty TestName
"EpochState compatibility" ((EpochState (PreviousEra MaryEra) -> Assertion) -> TestTree)
-> (EpochState (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a b. (a -> b) -> a -> b
$
        forall era (f :: * -> *).
(HasCallStack, TranslateEra era f,
 Show (TranslationError era f)) =>
TranslationContext era
-> (f era -> Encoding)
-> (f (PreviousEra era) -> Encoding)
-> f (PreviousEra era)
-> Assertion
translateEraEncoding @MaryEra @S.EpochState TranslationContext MaryEra
NoGenesis MaryEra
forall era. NoGenesis era
NoGenesis EpochState MaryEra -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR EpochState AllegraEra -> Encoding
EpochState (PreviousEra MaryEra) -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR
    , TestName
-> (ShelleyTxWits (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a. Testable a => TestName -> a -> TestTree
testProperty TestName
"ShelleyTxWits compatibility" ((ShelleyTxWits (PreviousEra MaryEra) -> Assertion) -> TestTree)
-> (ShelleyTxWits (PreviousEra MaryEra) -> Assertion) -> TestTree
forall a b. (a -> b) -> a -> b
$
        forall era (f :: * -> *).
(HasCallStack, TranslateEra era f,
 Show (TranslationError era f)) =>
TranslationContext era
-> (f era -> Encoding)
-> (f (PreviousEra era) -> Encoding)
-> f (PreviousEra era)
-> Assertion
translateEraEncoding @MaryEra @S.ShelleyTxWits TranslationContext MaryEra
NoGenesis MaryEra
forall era. NoGenesis era
NoGenesis ShelleyTxWits MaryEra -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR ShelleyTxWits AllegraEra -> Encoding
ShelleyTxWits (PreviousEra MaryEra) -> Encoding
forall a. ToCBOR a => a -> Encoding
toCBOR
    , TestName -> (Update AllegraEra -> Assertion) -> TestTree
forall a. Testable a => TestName -> a -> TestTree
testProperty TestName
"Update compatibility" (forall (f :: * -> *).
(EncCBOR (f AllegraEra), EncCBOR (f MaryEra),
 TranslateEra MaryEra f, Show (TranslationError MaryEra f)) =>
f AllegraEra -> Assertion
test @S.Update)
    ]

test ::
  forall f.
  ( EncCBOR (f AllegraEra)
  , EncCBOR (f MaryEra)
  , TranslateEra MaryEra f
  , Show (TranslationError MaryEra f)
  ) =>
  f AllegraEra ->
  Assertion
test :: forall (f :: * -> *).
(EncCBOR (f AllegraEra), EncCBOR (f MaryEra),
 TranslateEra MaryEra f, Show (TranslationError MaryEra f)) =>
f AllegraEra -> Assertion
test = [MaryEra]
-> TranslationContext MaryEra
-> f (PreviousEra MaryEra)
-> Assertion
forall (proxy :: * -> *) era (f :: * -> *).
(HasCallStack, TranslateEra era f, EncCBOR (f era),
 EncCBOR (f (PreviousEra era)), Show (TranslationError era f)) =>
proxy era
-> TranslationContext era -> f (PreviousEra era) -> Assertion
translateEraEncCBOR ([] :: [MaryEra]) TranslationContext MaryEra
NoGenesis MaryEra
forall era. NoGenesis era
NoGenesis