Skip to content

Commit 8e833ad

Browse files
This PR refactors test_pools.py align with the new build Method
1 parent b999519 commit 8e833ad

File tree

1 file changed

+21
-6
lines changed

1 file changed

+21
-6
lines changed

cardano_node_tests/tests/test_pools.py

Lines changed: 21 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,15 @@ def _register_stake_pool_w_build_common(
137137
138138
Args:
139139
build_method: One of `BuildMethods.BUILD` or `BuildMethods.BUILD_EST`.
140+
cluster_obj: An instance of `clusterlib.ClusterLib`.
141+
pool_data: A `PoolData` tuple containing info about the stake pool.
142+
pool_owners: A list of `PoolUser` structures containing pool user addresses and keys.
143+
vrf_vkey_file: A path to node VRF vkey file.
144+
cold_key_pair: A `ColdKeyPair` tuple containing the key pair and the counter.
145+
tx_name: A name of the transaction.
146+
reward_account_vkey_file: A path to reward account vkey file (optional).
147+
deposit: A deposit amount needed by the transaction (optional).
148+
destination_dir: A path to directory for storing artifacts (optional).
140149
"""
141150
tx_name = f"{tx_name}_reg_pool"
142151
pool_reg_cert_file = cluster_obj.g_stake_pool.gen_pool_registration_cert(
@@ -213,6 +222,7 @@ def _create_stake_pool_w_build_common(
213222
"""Create and register a stake pool using a `transaction build` command.
214223
215224
Args:
225+
build_method: One of `BuildMethods.BUILD` or `BuildMethods.BUILD_EST`.
216226
cluster_obj: An instance of `clusterlib.ClusterLib`.
217227
pool_data: A `PoolData` tuple containing info about the stake pool.
218228
pool_owners: A list of `PoolUser` structures containing pool user addresses and keys.
@@ -341,6 +351,7 @@ def _deregister_stake_pool_w_build(
341351

342352
return pool_dereg_cert_file, tx_raw_output
343353

354+
344355
def _create_register_pool(
345356
cluster_obj: clusterlib.ClusterLib,
346357
temp_template: str,
@@ -363,8 +374,8 @@ def _create_register_pool(
363374

364375
# Create and register pool
365376
if build_method in (
366-
clusterlib_utils.BuildMethods.BUILD,
367-
clusterlib_utils.BuildMethods.BUILD_EST,
377+
clusterlib_utils.BuildMethods.BUILD,
378+
clusterlib_utils.BuildMethods.BUILD_EST,
368379
):
369380
pool_creation_out = _create_stake_pool_w_build_common(
370381
cluster_obj=cluster_obj,
@@ -1450,8 +1461,10 @@ def test_update_stake_pool_metadata(
14501461
update_epoch = cluster.g_query.get_epoch()
14511462

14521463
# Update the pool metadata by resubmitting the pool registration certificate
1453-
if build_method in (clusterlib_utils.BuildMethods.BUILD,
1454-
clusterlib_utils.BuildMethods.BUILD_EST):
1464+
if build_method in (
1465+
clusterlib_utils.BuildMethods.BUILD,
1466+
clusterlib_utils.BuildMethods.BUILD_EST,
1467+
):
14551468
_register_stake_pool_w_build_common(
14561469
cluster_obj=cluster,
14571470
pool_data=pool_data_updated,
@@ -1572,8 +1585,10 @@ def test_update_stake_pool_parameters(
15721585
update_epoch = cluster.g_query.get_epoch()
15731586

15741587
# Update the pool parameters by resubmitting the pool registration certificate
1575-
if build_method in (clusterlib_utils.BuildMethods.BUILD,
1576-
clusterlib_utils.BuildMethods.BUILD_EST):
1588+
if build_method in (
1589+
clusterlib_utils.BuildMethods.BUILD,
1590+
clusterlib_utils.BuildMethods.BUILD_EST,
1591+
):
15771592
_register_stake_pool_w_build_common(
15781593
cluster_obj=cluster,
15791594
pool_data=pool_data_updated,

0 commit comments

Comments
 (0)