fix balance change in timeout test

Author: mmsqe

integration_tests/cosmoscli.py

@@ -1250,7 +1250,7 @@ def transfer_tokens(self, from_, to, amount, **kwargs):
             "gas": "auto",
             "gas_adjustment": "1.5",
         }
-        return json.loads(
+        rsp = json.loads(
             self.raw(
                 "tx",
                 "cronos",
@@ -1264,6 +1264,9 @@ def transfer_tokens(self, from_, to, amount, **kwargs):
                 **(default_kwargs | kwargs),
             )
         )
+        if rsp["code"] == 0:
+            rsp = self.event_query_tx_for(rsp["txhash"])
+        return rsp
 
     def icaauth_register_account(self, connid, **kwargs):
         "execute on host chain to attach an account to the connection"

integration_tests/test_ibc_timeout.py

@@ -60,14 +60,18 @@ def test_cronos_transfer_timeout(ibc):
     )
     assert rsp["code"] == 0, rsp["raw_log"]
 
-    new_src_balance = 0
-
     def check_balance_change():
-        nonlocal new_src_balance
+        new_src_balance = get_balance(ibc.cronos, src_addr, src_denom)
+        get_balance(ibc.chainmain, dst_addr, dst_denom)
+        return old_src_balance != new_src_balance
+
+    wait_for_fn("balance has change", check_balance_change)
+
+    def check_no_change():
         new_src_balance = get_balance(ibc.cronos, src_addr, src_denom)
         get_balance(ibc.chainmain, dst_addr, dst_denom)
         return old_src_balance == new_src_balance
 
-    wait_for_fn("balance no change", check_balance_change)
+    wait_for_fn("balance no change", check_no_change)
     new_dst_balance = get_balance(ibc.chainmain, dst_addr, dst_denom)
     assert old_dst_balance == new_dst_balance