From 215b0896b3d57849b0af79f1edaa30a5c026e9ed Mon Sep 17 00:00:00 2001
From: Oliver Bundalo <oliver.bundalo@ethernal.tech>
Date: Fri, 23 Feb 2024 09:35:26 +0100
Subject: [PATCH] e2e storage test

---
 e2e-polybft/e2e/storage_test.go | 189 ++++++++++++++++++++++++++++++++
 1 file changed, 189 insertions(+)
 create mode 100644 e2e-polybft/e2e/storage_test.go

diff --git a/e2e-polybft/e2e/storage_test.go b/e2e-polybft/e2e/storage_test.go
new file mode 100644
index 0000000000..b4bde540a6
--- /dev/null
+++ b/e2e-polybft/e2e/storage_test.go
@@ -0,0 +1,189 @@
+package e2e
+
+import (
+	"math/big"
+	"reflect"
+	"sync"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"github.com/umbracle/ethgo"
+	"github.com/umbracle/ethgo/jsonrpc"
+	"github.com/umbracle/ethgo/wallet"
+
+	"github.com/0xPolygon/polygon-edge/consensus/polybft"
+	"github.com/0xPolygon/polygon-edge/e2e-polybft/framework"
+	"github.com/0xPolygon/polygon-edge/types"
+)
+
+type txObject struct {
+	bNumber uint64
+	txHash  ethgo.Hash
+	to      ethgo.Address
+}
+
+func TestE2E_Storage(t *testing.T) {
+	// premine an account in the genesis file
+	sender, err := wallet.GenerateKey()
+	require.NoError(t, err)
+
+	cluster := framework.NewTestCluster(t, 5,
+		framework.WithPremine(types.Address(sender.Address())),
+		framework.WithBurnContract(&polybft.BurnContractInfo{BlockNumber: 0, Address: types.ZeroAddress}),
+	)
+	defer cluster.Stop()
+
+	cluster.WaitForReady(t)
+
+	client := cluster.Servers[0].JSONRPC().Eth()
+
+	num := 20
+
+	receivers := []ethgo.Address{}
+
+	for i := 0; i < num; i++ {
+		key, err := wallet.GenerateKey()
+		require.NoError(t, err)
+
+		receivers = append(receivers, key.Address())
+	}
+
+	var wg sync.WaitGroup
+
+	txs := []txObject{}
+
+	for i := 0; i < num; i++ {
+		wg.Add(1)
+
+		func(i int, to ethgo.Address) {
+			defer wg.Done()
+
+			txn := &ethgo.Transaction{
+				From:  sender.Address(),
+				To:    &to,
+				Gas:   30000, // enough to send a transfer
+				Value: big.NewInt(int64(i)),
+				Nonce: uint64(i),
+			}
+
+			// Send every second transaction as a dynamic fees one
+			if i%2 == 0 {
+				txn.Type = ethgo.TransactionDynamicFee
+				txn.MaxFeePerGas = big.NewInt(1000000000)
+				txn.MaxPriorityFeePerGas = big.NewInt(100000000)
+			} else {
+				txn.Type = ethgo.TransactionLegacy
+				txn.GasPrice = ethgo.Gwei(2).Uint64()
+			}
+
+			bn, th := sendTx(t, client, sender, txn)
+			txs = append(txs, txObject{bNumber: bn, txHash: th, to: to})
+		}(i, receivers[i])
+	}
+
+	wg.Wait()
+
+	err = cluster.WaitUntil(2*time.Minute, 2*time.Second, func() bool {
+		for i, receiver := range receivers {
+			balance, err := client.GetBalance(receiver, ethgo.Latest)
+			if err != nil {
+				return true
+			}
+
+			t.Logf("Balance %s %s", receiver, balance)
+
+			if balance.Uint64() != uint64(i) {
+				return false
+			}
+		}
+
+		return true
+	})
+	require.NoError(t, err)
+
+	checkStorage(t, txs, client)
+}
+
+// sendTx is a helper function which signs transaction with provided private key and sends it
+func sendTx(t *testing.T, client *jsonrpc.Eth, sender *wallet.Key, txn *ethgo.Transaction) (uint64, ethgo.Hash) {
+	t.Helper()
+
+	chainID, err := client.ChainID()
+	require.NoError(t, err)
+
+	if txn.Type == ethgo.TransactionDynamicFee {
+		txn.ChainID = chainID
+	}
+
+	signer := wallet.NewEIP155Signer(chainID.Uint64())
+	signedTxn, err := signer.SignTx(txn, sender)
+	require.NoError(t, err)
+
+	txnRaw, err := signedTxn.MarshalRLPTo(nil)
+	require.NoError(t, err)
+
+	h, err := client.SendRawTransaction(txnRaw)
+	require.NoError(t, err)
+
+	bn, err := client.BlockNumber()
+	require.NoError(t, err)
+
+	return bn, h
+}
+
+func checkStorage(t *testing.T, txs []txObject, client *jsonrpc.Eth) {
+	t.Helper()
+
+	for i, tx := range txs {
+		bn, err := client.GetBlockByNumber(ethgo.BlockNumber(tx.bNumber), true)
+		require.NoError(t, err)
+		assert.NotNil(t, bn)
+
+		bh, err := client.GetBlockByHash(bn.Hash, true)
+		require.NoError(t, err)
+		assert.NotNil(t, bh)
+
+		if !reflect.DeepEqual(bn, bh) {
+			t.Fatal("blocks dont match")
+		}
+
+		bt, err := client.GetTransactionByHash(tx.txHash)
+		require.NoError(t, err)
+		assert.NotNil(t, bt)
+		assert.Equal(t, uint64(i), bt.Value.Uint64())
+		assert.Equal(t, uint64(30000), bt.Gas)
+		assert.Equal(t, uint64(i), bt.Nonce)
+		assert.Equal(t, uint64(i), bt.TxnIndex)
+		assert.NotEmpty(t, bt.V)
+		assert.NotEmpty(t, bt.R)
+		assert.NotEmpty(t, bt.S)
+
+		if !reflect.DeepEqual(tx.to, *bt.To) {
+			t.Fatal("tx to dont match")
+		}
+
+		if i%2 == 0 {
+			assert.Equal(t, ethgo.TransactionDynamicFee, bt.Type)
+			assert.Equal(t, uint64(0), bt.GasPrice)
+			assert.NotNil(t, bt.ChainID)
+		} else {
+			assert.Equal(t, ethgo.TransactionLegacy, bt.Type)
+			assert.Equal(t, ethgo.Gwei(2).Uint64(), bt.GasPrice)
+		}
+
+		r, err := client.GetTransactionReceipt(tx.txHash)
+		require.NoError(t, err)
+		assert.NotNil(t, r)
+		assert.Equal(t, bt.TxnIndex, r.TransactionIndex)
+		assert.Equal(t, bt.Hash, r.TransactionHash)
+		assert.Equal(t, bt.BlockHash, r.BlockHash)
+		assert.Equal(t, bt.BlockNumber, r.BlockNumber)
+		assert.NotEmpty(t, r.LogsBloom)
+
+		if !reflect.DeepEqual(*bt.To, *r.To) {
+			t.Fatal("receipt to dont match")
+		}
+	}
+}