fix: get_erc20_balance method to Foundry

Author: romanzac

tests-functional/clients/foundry.py

@@ -200,3 +200,11 @@ def generate_tokens(self, controller_address, to_address, token_amount, private_
         )
         result = self.container.exec_run(generate_cmd)
         return result
+
+    def get_erc20_balance(self, token_address, owner_address):
+        if not self.container:
+            raise Exception("Container not found")
+
+        balance_cmd = f"cast call {token_address} 'balanceOf(address)' {owner_address} --rpc-url http://anvil:8545"
+        result = self.container.exec_run(balance_cmd)
+        return result

tests-functional/tests/test_wakuext_community_token_permissions.py

@@ -110,8 +110,7 @@ def test_membership_with_valid_tokens(self, foundry_client):
         member_address = accounts[0]["address"] if accounts else self.fake_address
 
         # Verify the balance directly with cast
-        balance_cmd = f"cast call {self.snt_address} 'balanceOf(address)' {member_address} --rpc-url http://anvil:8545"
-        balance_result = foundry_client.container.exec_run(balance_cmd)
+        balance_result = foundry_client.get_erc20_balance(self.snt_address, member_address)
         logger.debug(f"SNT balance check: exit_code={balance_result.exit_code}, output={balance_result.output.decode()}")
 
         token_address = self.snt_address
@@ -174,8 +173,7 @@ def test_admin_token_permissions_with_valid_tokens(self, foundry_client):
         member_address = accounts[0]["address"] if accounts else self.fake_address
 
         # Verify the balance directly with cast
-        balance_cmd = f"cast call {self.snt_address} 'balanceOf(address)' {member_address} --rpc-url http://anvil:8545"
-        balance_result = foundry_client.container.exec_run(balance_cmd)
+        balance_result = foundry_client.get_erc20_balance(self.snt_address, member_address)
         logger.debug(f"SNT balance check: exit_code={balance_result.exit_code}, output={balance_result.output.decode()}")
 
         # Owner creates token-gated community with admin permission