From 612c53fe8d71528c66c7ae992251354449248c19 Mon Sep 17 00:00:00 2001 From: Martin Kourim Date: Wed, 27 Jan 2021 09:03:57 +0100 Subject: [PATCH] Make it work with the new deposit keys --- cardano_node_tests/utils/clusterlib.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/cardano_node_tests/utils/clusterlib.py b/cardano_node_tests/utils/clusterlib.py index 03b1ac710..4486dd60e 100644 --- a/cardano_node_tests/utils/clusterlib.py +++ b/cardano_node_tests/utils/clusterlib.py @@ -934,11 +934,13 @@ def get_stake_addr_info(self, stake_addr: str) -> StakeAddrInfo: def get_key_deposit(self) -> int: """Return key deposit amount.""" - return int(self.get_protocol_params()["keyDeposit"]) + pparams = self.get_protocol_params() + return int(pparams.get("keyDeposit") or pparams.get("stakeAddressDeposit") or 0) def get_pool_deposit(self) -> int: """Return pool deposit amount.""" - return int(self.get_protocol_params()["poolDeposit"]) + pparams = self.get_protocol_params() + return int(pparams.get("poolDeposit") or pparams.get("stakePoolDeposit") or 0) def get_stake_distribution(self) -> dict: """Return current aggregated stake distribution per stake pool.""" @@ -996,8 +998,8 @@ def get_tx_deposit(self, tx_files: TxFiles) -> int: return 0 pparams = self.get_protocol_params() - key_deposit = pparams["keyDeposit"] - pool_deposit = pparams["poolDeposit"] + key_deposit = pparams.get("keyDeposit") or pparams.get("stakeAddressDeposit") or 0 + pool_deposit = pparams.get("poolDeposit") or pparams.get("stakePoolDeposit") or 0 deposit = 0 for cert in tx_files.certificate_files: