Merge pull request #398 from mkoura/cardano_mode_default

Make cardano mode default
IntersectMBO · Feb 26, 2021 · b3212e3 · b3212e3
2 parents 52d9f23 + 6fa5b86
commit b3212e3
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/cardano_node_tests/utils/cluster_nodes.py b/cardano_node_tests/utils/cluster_nodes.py
@@ -75,7 +75,13 @@ def __init__(self) -> None:
 
     def get_cluster_obj(self) -> clusterlib.ClusterLib:
         """Return instance of `ClusterLib` (cluster_obj)."""
-        cluster_obj = clusterlib.ClusterLib(get_cluster_env().state_dir)
+        cluster_env = get_cluster_env()
+        cluster_obj = clusterlib.ClusterLib(
+            state_dir=cluster_env.state_dir,
+            protocol=clusterlib.Protocols.SHELLEY,
+            era=cluster_env.cluster_era,
+            tx_era=cluster_env.tx_era,
+        )
         return cluster_obj
 
     def create_addrs_data(

diff --git a/cardano_node_tests/utils/clusterlib.py b/cardano_node_tests/utils/clusterlib.py
@@ -179,7 +179,7 @@ class ClusterLib:
     def __init__(
         self,
         state_dir: FileType,
-        protocol: str = Protocols.SHELLEY,
+        protocol: str = Protocols.CARDANO,
         era: str = "",
         tx_era: str = "",
         slots_offset: int = 0,
@@ -1981,3 +1981,6 @@ def create_stake_pool(
             tx_raw_output=tx_raw_output,
             kes_key_pair=node_kes,
         )
+
+    def __repr__(self) -> str:
+        return f"<ClusterLib: protocol={self.protocol}, era={self.era}, tx_era={self.tx_era}>"