add staking cache integration tests

Author: randy-cro

integration_tests/cosmoscli.py

@@ -19,6 +19,7 @@
 # the default initial base fee used by integration tests
 DEFAULT_GAS_PRICE = "100000000000basetcro"
 DEFAULT_GAS = "250000"
+STAKING_DEFAULT_GAS = "1000000"
 
 
 class ModuleAccount(enum.Enum):
@@ -409,24 +410,54 @@ def get_delegated_amount(self, which_addr):
             )
         )
 
-    def delegate_amount(self, to_addr, amount, from_addr, gas_price=None):
-        if gas_price is None:
-            return json.loads(
-                self.raw(
-                    "tx",
-                    "staking",
-                    "delegate",
-                    to_addr,
-                    amount,
-                    "-y",
-                    home=self.data_dir,
-                    from_=from_addr,
-                    keyring_backend="test",
-                    chain_id=self.chain_id,
-                    node=self.node_rpc,
-                )
+    def get_delegations(self, which_addr):
+        """Query all delegations made from one delegator."""
+        return json.loads(
+            self.raw(
+                "query",
+                "staking",
+                "delegations",
+                which_addr,
+                home=self.data_dir,
+                chain_id=self.chain_id,
+                node=self.node_rpc,
+                output="json",
             )
-        else:
+        )
+
+    def get_unbonding_delegations(self, which_addr):
+        """Query all unbonding delegations from a delegator."""
+        return json.loads(
+            self.raw(
+                "query",
+                "staking",
+                "unbonding-delegations",
+                which_addr,
+                home=self.data_dir,
+                chain_id=self.chain_id,
+                node=self.node_rpc,
+                output="json",
+            )
+        ).get("unbonding_responses", [])
+
+    def get_redelegations(self, delegator_addr, src_validator_addr, dst_validator_addr):
+        """Query all redelegations from a delegator."""
+        return json.loads(
+            self.raw(
+                "query",
+                "staking",
+                "redelegation",
+                delegator_addr,
+                src_validator_addr,
+                dst_validator_addr,
+                home=self.data_dir,
+                chain_id=self.chain_id,
+                node=self.node_rpc,
+                output="json",
+            )
+        ).get("redelegation_responses", [])
+
+    def delegate_amount(self, to_addr, amount, from_addr):
             return json.loads(
                 self.raw(
                     "tx",
@@ -440,7 +471,8 @@ def delegate_amount(self, to_addr, amount, from_addr, gas_price=None):
                     keyring_backend="test",
                     chain_id=self.chain_id,
                     node=self.node_rpc,
-                    gas_prices=gas_price,
+                    gas_prices=DEFAULT_GAS_PRICE,
+                    gas=STAKING_DEFAULT_GAS,
                 )
             )
 
@@ -459,6 +491,8 @@ def unbond_amount(self, to_addr, amount, from_addr):
                 keyring_backend="test",
                 chain_id=self.chain_id,
                 node=self.node_rpc,
+                gas=STAKING_DEFAULT_GAS,
+                gas_prices=DEFAULT_GAS_PRICE,
             )
         )
 
@@ -480,6 +514,8 @@ def redelegate_amount(
                 keyring_backend="test",
                 chain_id=self.chain_id,
                 node=self.node_rpc,
+                gas=STAKING_DEFAULT_GAS,
+                gas_prices=DEFAULT_GAS_PRICE,
             )
         )