cardano-testnet | Add test for providing reference input datums

Author: carbolymer

cardano-testnet/cardano-testnet.cabal

@@ -185,7 +185,8 @@ test-suite cardano-testnet-test
 
   main-is:              cardano-testnet-test.hs
 
-  other-modules:        Cardano.Testnet.Test.Cli.Conway.StakeSnapshot
+  other-modules:        Cardano.Testnet.Test.Api.TxReferenceInputDatum
+                        Cardano.Testnet.Test.Cli.Conway.StakeSnapshot
                         Cardano.Testnet.Test.Cli.KesPeriodInfo
                         Cardano.Testnet.Test.Cli.LeadershipSchedule
                         Cardano.Testnet.Test.Cli.Query

cardano-testnet/src/Testnet/Components/Query.hs

@@ -36,6 +36,7 @@ module Testnet.Components.Query
   , checkDRepsNumber
   , checkDRepState
   , assertNewEpochState
+  , getProtocolParams
   , getGovActionLifetime
   , getKeyDeposit
   , getDelegationState
@@ -44,8 +45,9 @@ module Testnet.Components.Query
 
 import           Cardano.Api as Api
 import           Cardano.Api.Ledger (Credential, DRepState, EpochInterval (..), KeyRole (DRepRole))
-import           Cardano.Api.Shelley (ShelleyLedgerEra)
 import qualified Cardano.Api.Ledger as L
+import           Cardano.Api.Shelley (LedgerProtocolParameters (..), ShelleyLedgerEra)
+import qualified Cardano.Api.Tx.UTxO as Utxo
 
 import           Cardano.Crypto.Hash (hashToStringAsHex)
 import           Cardano.Ledger.Api (ConwayGovState)
@@ -65,7 +67,6 @@ import           Data.IORef
 import           Data.List (sortOn)
 import qualified Data.Map as Map
 import           Data.Map.Strict (Map)
-import qualified Data.Map.Strict as M
 import           Data.Maybe
 import           Data.Ord (Down (..))
 import           Data.Text (Text)
@@ -301,11 +302,11 @@ findAllUtxos
   => MonadTest m
   => EpochStateView
   -> ShelleyBasedEra era
-  -> m (Map TxIn (TxOut CtxUTxO era))
+  -> m (UTxO era)
 findAllUtxos epochStateView sbe = withFrozenCallStack $ do
   AnyNewEpochState sbe' _ tbs <- getEpochState epochStateView
   Refl <- H.leftFail $ assertErasEqual sbe sbe'
-  pure $ getLedgerTablesUTxOValues sbe' tbs
+  pure . UTxO $ getLedgerTablesUTxOValues sbe' tbs
 
 -- | Retrieve utxos from the epoch state view for an address.
 findUtxosWithAddress
@@ -316,7 +317,7 @@ findUtxosWithAddress
   => EpochStateView
   -> ShelleyBasedEra era
   -> Text -- ^ Address
-  -> m (Map TxIn (TxOut CtxUTxO era))
+  -> m (UTxO era)
 findUtxosWithAddress epochStateView sbe address = withFrozenCallStack $ do
   utxos <- findAllUtxos epochStateView sbe
   H.note_ $ "Finding UTxOs for " <> T.unpack address
@@ -327,7 +328,7 @@ findUtxosWithAddress epochStateView sbe address = withFrozenCallStack $ do
       maybeToEither ("Could not deserialize address: " <> T.unpack address)
         (deserialiseAddress AsAddressAny address)
 
-  let utxos' = M.filter (\(TxOut txAddr _ _ _)  -> txAddr == address') utxos
+  let utxos' = Utxo.filter (\(TxOut txAddr _ _ _)  -> txAddr == address') utxos
   H.note_ $ unlines (map show $ toList utxos')
   pure utxos'
   where
@@ -344,7 +345,7 @@ findLargestUtxoWithAddress
   -> Text -- ^ Address
   -> m (Maybe (TxIn, TxOut CtxUTxO era))
 findLargestUtxoWithAddress epochStateView sbe address = withFrozenCallStack $ do
-  utxos <- M.assocs <$> findUtxosWithAddress epochStateView sbe address
+  utxos <- toList <$> findUtxosWithAddress epochStateView sbe address
   pure
     . listToMaybe
     $ sortOn (\(_, TxOut _ txOutValue _ _) -> Down $ txOutValueToLovelace txOutValue) utxos
@@ -558,6 +559,15 @@ assertNewEpochState epochStateView sbe maxWait lens expected = withFrozenCallSta
         Refl <- H.leftFail $ assertErasEqual sbe actualEra
         pure $ newEpochState ^. lens
 
+-- | Return current protocol parameters from the governance state
+getProtocolParams :: (H.MonadAssertion m, MonadTest m, MonadIO m)
+  => EpochStateView
+  -> ConwayEraOnwards era
+  -> m (LedgerProtocolParameters era)
+getProtocolParams epochStateView ceo = conwayEraOnwardsConstraints ceo $ do
+   govState :: ConwayGovState era <- getGovState epochStateView ceo
+   pure . LedgerProtocolParameters $ govState ^. L.cgsCurPParamsL
+
 
 -- | Obtains the @govActionLifetime@ from the protocol parameters.
 -- The @govActionLifetime@ or governance action maximum lifetime in epochs is

cardano-testnet/src/Testnet/Types.hs

@@ -21,6 +21,7 @@ module Testnet.Types
   , testnetSprockets
   , TestnetNode(..)
   , nodeSocketPath
+  , nodeConnectionInfo
   , isTestnetNodeSpo
   , SpoNodeKeys(..)
   , Delegator(..)
@@ -62,18 +63,20 @@ import           Data.List (intercalate)
 import           Data.Maybe
 import           Data.MonoTraversable (Element, MonoFunctor (..))
 import           Data.Text (Text)
-import           Data.Time.Clock (UTCTime)
 import           GHC.Exts (IsString (..))
 import           GHC.Generics (Generic)
 import qualified GHC.IO.Handle as IO
 import           GHC.Stack
+import           Lens.Micro (ix, (^?))
 import           Network.Socket (HostAddress, PortNumber, hostAddressToTuple, tupleToHostAddress)
 import           System.FilePath
 import qualified System.Process as IO
 
 import           Testnet.Start.Types
 
+import           Hedgehog (MonadTest)
 import qualified Hedgehog as H
+import qualified Hedgehog.Extras as H
 import qualified Hedgehog.Extras.Stock as H
 import           Hedgehog.Extras.Stock.IO.Network.Sprocket (Sprocket (..))
 
@@ -115,7 +118,7 @@ data SKey k
 data TestnetRuntime = TestnetRuntime
   { configurationFile :: !(NodeConfigFile In)
   , shelleyGenesisFile :: !FilePath
-  , testnetMagic :: !Int
+  , testnetMagic :: !Int -- TODO change to Word32
   , testnetNodes :: ![TestnetNode]
   , wallets :: ![PaymentKeyInfo]
   , delegators :: ![Delegator]
@@ -148,6 +151,23 @@ isTestnetNodeSpo = isJust . poolKeys
 nodeSocketPath :: TestnetNode -> SocketPath
 nodeSocketPath = File . H.sprocketSystemName . nodeSprocket
 
+-- | Connection data for a node in the testnet
+nodeConnectionInfo :: MonadTest m
+                   => TestnetRuntime
+                   -> Int -- ^ node index, starting from 0
+                   -> m LocalNodeConnectInfo -- ^ fails when there's no node with requested index
+nodeConnectionInfo TestnetRuntime{testnetMagic, testnetNodes} index = do
+  case testnetNodes ^? ix index of
+    Nothing -> do
+      H.note_ $ "There is no node in the testnet with index: " <> show index <> ". Number of nodes: " <> show (length testnetNodes)
+      H.failure
+    Just node ->
+        pure LocalNodeConnectInfo
+              { localNodeSocketPath= nodeSocketPath node
+              , localNodeNetworkId=Testnet (NetworkMagic $ fromIntegral testnetMagic)
+              , localConsensusModeParams=CardanoModeParams $ EpochSlots 21600}
+
+
 data SpoNodeKeys = SpoNodeKeys
   { poolNodeKeysCold :: KeyPair StakePoolKey
   , poolNodeKeysVrf :: KeyPair VrfKey
@@ -187,14 +207,14 @@ getStartTime
   => HasCallStack
   => FilePath
   -> TestnetRuntime
-  -> m UTCTime
+  -> m SystemStart
 getStartTime tempRootPath TestnetRuntime{configurationFile} = withFrozenCallStack $ H.evalEither <=< H.evalIO . runExceptT $ do
   byronGenesisFile <-
     decodeNodeConfiguration configurationFile >>= \case
       NodeProtocolConfigurationCardano NodeByronProtocolConfiguration{npcByronGenesisFile} _ _ _ _ _ ->
         pure $ unGenesisFile npcByronGenesisFile
   let byronGenesisFilePath = tempRootPath </> byronGenesisFile
-  G.gdStartTime . G.configGenesisData <$> decodeGenesisFile byronGenesisFilePath
+  SystemStart . G.gdStartTime . G.configGenesisData <$> decodeGenesisFile byronGenesisFilePath
   where
     decodeNodeConfiguration :: File NodeConfig In -> ExceptT String IO NodeProtocolConfiguration
     decodeNodeConfiguration (File file) = do