Skip to content

Commit 3a186b9

Browse files
committed
feat: add deposit amount to stake addr deregistration
- Updated `deregister_stake_addr` to include `deposit_amt` parameter. - Modified `cleanup` function to retrieve and pass the stake deposit amount. - Adjusted transaction creation to handle deposit amount.
1 parent 5be6cf6 commit 3a186b9

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

cardano_node_tests/utils/testnet_cleanup.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -45,12 +45,17 @@ def withdraw_reward(
4545

4646

4747
def deregister_stake_addr(
48-
cluster_obj: clusterlib.ClusterLib, pool_user: clusterlib.PoolUser, name_template: str
48+
cluster_obj: clusterlib.ClusterLib,
49+
pool_user: clusterlib.PoolUser,
50+
name_template: str,
51+
deposit_amt: int,
4952
) -> None:
5053
"""Deregister stake address."""
5154
# files for deregistering stake address
5255
stake_addr_dereg_cert = cluster_obj.g_stake_address.gen_stake_addr_deregistration_cert(
53-
addr_name=f"rf_{name_template}_addr0_dereg", stake_vkey_file=pool_user.stake.vkey_file
56+
addr_name=f"rf_{name_template}_addr0_dereg",
57+
deposit_amt=deposit_amt,
58+
stake_vkey_file=pool_user.stake.vkey_file,
5459
)
5560
tx_files_deregister = clusterlib.TxFiles(
5661
certificate_files=[stake_addr_dereg_cert],
@@ -63,6 +68,7 @@ def deregister_stake_addr(
6368
src_address=pool_user.payment.address,
6469
tx_name=f"{name_template}_dereg_stake_addr",
6570
tx_files=tx_files_deregister,
71+
deposit=-deposit_amt,
6672
)
6773

6874

@@ -166,6 +172,7 @@ def cleanup(
166172
faucet_addr_file = cluster_env.state_dir / "shelley" / "faucet.addr"
167173
faucet_payment = create_addr_record(faucet_addr_file)
168174
files_found = group_files(find_files(location))
175+
stake_deposit_amt = cluster_obj.g_query.get_address_deposit()
169176

170177
def _run(files: tp.List[pl.Path]) -> None:
171178
for fpath in files:
@@ -200,7 +207,10 @@ def _run(files: tp.List[pl.Path]) -> None:
200207
)
201208

202209
deregister_stake_addr(
203-
cluster_obj=cluster_obj, pool_user=pool_user, name_template=f_name
210+
cluster_obj=cluster_obj,
211+
pool_user=pool_user,
212+
name_template=f_name,
213+
deposit_amt=stake_deposit_amt,
204214
)
205215
else:
206216
try:

0 commit comments

Comments
 (0)