From a4235e2b3d06ed041b59d0a0037c522f27536ce1 Mon Sep 17 00:00:00 2001 From: Aayush Date: Tue, 9 May 2023 12:35:09 -0400 Subject: [PATCH] feat: deflake eth_balance_test --- itests/eth_balance_test.go | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/itests/eth_balance_test.go b/itests/eth_balance_test.go index d5f15122715..1165879026d 100644 --- a/itests/eth_balance_test.go +++ b/itests/eth_balance_test.go @@ -121,15 +121,24 @@ func TestEthBalanceCorrectLookup(t *testing.T) { require.NoError(t, err) require.True(t, ml.Receipt.ExitCode.IsSuccess()) - bal, err := client.EVM().EthGetBalance(ctx, ethAddr, strconv.FormatInt(int64(ml.Height-2), 10)) + execTs, err := client.ChainGetTipSet(ctx, ml.TipSet) + require.NoError(t, err) + + inclTs, err := client.ChainGetTipSet(ctx, execTs.Parents()) + require.NoError(t, err) + + inclTsParents, err := client.ChainGetTipSet(ctx, inclTs.Parents()) + require.NoError(t, err) + + bal, err := client.EVM().EthGetBalance(ctx, ethAddr, strconv.FormatInt(int64(inclTsParents.Height()), 10)) require.NoError(t, err) require.Equal(t, int64(0), bal.Int64()) - bal, err = client.EVM().EthGetBalance(ctx, ethAddr, strconv.FormatInt(int64(ml.Height-1), 10)) + bal, err = client.EVM().EthGetBalance(ctx, ethAddr, strconv.FormatInt(int64(inclTs.Height()), 10)) require.NoError(t, err) require.Equal(t, val, bal.Int64()) - bal, err = client.EVM().EthGetBalance(ctx, ethAddr, strconv.FormatInt(int64(ml.Height), 10)) + bal, err = client.EVM().EthGetBalance(ctx, ethAddr, strconv.FormatInt(int64(execTs.Height()), 10)) require.NoError(t, err) require.Equal(t, val, bal.Int64()) }