From 6ac734db6da4ab89af42d34c5a84328c4ada0b65 Mon Sep 17 00:00:00 2001
From: lumtis <lucas.bertrand.22@gmail.com>
Date: Fri, 24 Jan 2025 14:36:34 +0100
Subject: [PATCH] add spl balance in command

---
 e2e/runner/balances.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/e2e/runner/balances.go b/e2e/runner/balances.go
index 4c2cf6a9b4..c47a1d6292 100644
--- a/e2e/runner/balances.go
+++ b/e2e/runner/balances.go
@@ -23,6 +23,7 @@ type AccountBalances struct {
 	ZetaERC20 *big.Int
 	ZetaBTC   *big.Int
 	ZetaSOL   *big.Int
+	ZetaSPL   *big.Int
 	EvmETH    *big.Int
 	EvmZETA   *big.Int
 	EvmERC20  *big.Int
@@ -72,6 +73,10 @@ func (r *E2ERunner) GetAccountBalances(skipBTC bool) (AccountBalances, error) {
 	if err != nil {
 		return AccountBalances{}, fmt.Errorf("get sol zrc20 balance: %w", err)
 	}
+	zetaSPL, err := r.getZRC20BalanceSafe(r.SPLZRC20)
+	if err != nil {
+		return AccountBalances{}, fmt.Errorf("get spl zrc20 balance: %w", err)
+	}
 
 	// evm
 	evmEth, err := r.EVMClient.BalanceAt(r.Ctx, r.EVMAddress(), nil)
@@ -140,6 +145,7 @@ func (r *E2ERunner) GetAccountBalances(skipBTC bool) (AccountBalances, error) {
 		ZetaERC20: zetaErc20,
 		ZetaBTC:   zetaBtc,
 		ZetaSOL:   zetaSol,
+		ZetaSPL:   zetaSPL,
 		EvmETH:    evmEth,
 		EvmZETA:   evmZeta,
 		EvmERC20:  evmErc20,
@@ -190,6 +196,7 @@ func (r *E2ERunner) PrintAccountBalances(balances AccountBalances) {
 	r.Logger.Print("* ERC20 balance: %s", balances.ZetaERC20.String())
 	r.Logger.Print("* BTC balance:   %s", balances.ZetaBTC.String())
 	r.Logger.Print("* SOL balance: %s", balances.ZetaSOL.String())
+	r.Logger.Print("* SPL balance: %s", balances.ZetaSPL.String())
 
 	// evm
 	r.Logger.Print("EVM:")