-
Notifications
You must be signed in to change notification settings - Fork 108
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: integrate withdraw SPL (#3134)
* withdraw spl and e2e test * changelog * fix unit test * e2e test that rent payer creates receiver ata * comments fixes * PR comments fixes * fmt * PR comments * PR comments * fix unit test
- Loading branch information
Showing
26 changed files
with
790 additions
and
138 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,73 @@ | ||
| package e2etests | ||
|
|
||
| import ( | ||
| "math/big" | ||
|
|
||
| "github.com/ethereum/go-ethereum/accounts/abi/bind" | ||
| "github.com/gagliardetto/solana-go" | ||
| "github.com/gagliardetto/solana-go/rpc" | ||
| "github.com/stretchr/testify/require" | ||
|
|
||
| "github.com/zeta-chain/node/e2e/runner" | ||
| "github.com/zeta-chain/node/e2e/utils" | ||
| crosschaintypes "github.com/zeta-chain/node/x/crosschain/types" | ||
| ) | ||
|
|
||
| func TestSPLWithdraw(r *runner.E2ERunner, args []string) { | ||
| require.Len(r, args, 1) | ||
|
|
||
| withdrawAmount := parseBigInt(r, args[0]) | ||
|
|
||
| // get SPL ZRC20 balance before withdraw | ||
| zrc20BalanceBefore, err := r.SPLZRC20.BalanceOf(&bind.CallOpts{}, r.EVMAddress()) | ||
| require.NoError(r, err) | ||
| r.Logger.Info("runner balance of SPL before withdraw: %d", zrc20BalanceBefore) | ||
|
|
||
| require.Equal(r, 1, zrc20BalanceBefore.Cmp(withdrawAmount), "Insufficient balance for withdrawal") | ||
|
|
||
| // parse withdraw amount (in lamports), approve amount is 1 SOL | ||
| approvedAmount := new(big.Int).SetUint64(solana.LAMPORTS_PER_SOL) | ||
| require.Equal( | ||
| r, | ||
| -1, | ||
| withdrawAmount.Cmp(approvedAmount), | ||
| "Withdrawal amount must be less than the %v", | ||
| approvedAmount, | ||
| ) | ||
|
|
||
| // load deployer private key | ||
| privkey := r.GetSolanaPrivKey() | ||
|
|
||
| // get receiver ata balance before withdraw | ||
| receiverAta := r.ResolveSolanaATA(privkey, privkey.PublicKey(), r.SPLAddr) | ||
| receiverBalanceBefore, err := r.SolanaClient.GetTokenAccountBalance(r.Ctx, receiverAta, rpc.CommitmentFinalized) | ||
| require.NoError(r, err) | ||
| r.Logger.Info("receiver balance of SPL before withdraw: %s", receiverBalanceBefore.Value.Amount) | ||
|
|
||
| // withdraw | ||
| tx := r.WithdrawSPLZRC20(privkey.PublicKey(), withdrawAmount, approvedAmount) | ||
|
|
||
| // wait for the cctx to be mined | ||
| cctx := utils.WaitCctxMinedByInboundHash(r.Ctx, tx.Hash().Hex(), r.CctxClient, r.Logger, r.CctxTimeout) | ||
| utils.RequireCCTXStatus(r, cctx, crosschaintypes.CctxStatus_OutboundMined) | ||
|
|
||
| // get SPL ZRC20 balance after withdraw | ||
| zrc20BalanceAfter, err := r.SPLZRC20.BalanceOf(&bind.CallOpts{}, r.EVMAddress()) | ||
| require.NoError(r, err) | ||
| r.Logger.Info("runner balance of SPL after withdraw: %d", zrc20BalanceAfter) | ||
|
|
||
| // verify balances are updated | ||
| receiverBalanceAfter, err := r.SolanaClient.GetTokenAccountBalance(r.Ctx, receiverAta, rpc.CommitmentFinalized) | ||
| require.NoError(r, err) | ||
| r.Logger.Info("receiver balance of SPL after withdraw: %s", receiverBalanceAfter.Value.Amount) | ||
|
|
||
| // verify amount is added to receiver ata | ||
| require.EqualValues( | ||
| r, | ||
| new(big.Int).Add(withdrawAmount, parseBigInt(r, receiverBalanceBefore.Value.Amount)).String(), | ||
| parseBigInt(r, receiverBalanceAfter.Value.Amount).String(), | ||
| ) | ||
|
|
||
| // verify amount is subtracted on zrc20 | ||
| require.EqualValues(r, new(big.Int).Sub(zrc20BalanceBefore, withdrawAmount).String(), zrc20BalanceAfter.String()) | ||
| } |
Oops, something went wrong.