feat(load): add trie stress test (#4137)

Author: RodrigoVillar

tests/load/contracts/TrieStressTest.bindings.go

@@ -0,0 +1,13 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.0;
+
+contract TrieStressTest {
+    bytes32[] private data;
+
+    function writeValues(uint value) public {
+        bytes32 dataToPush = bytes32(uint256(uint160(msg.sender)) << 96);
+        for (uint i = 0; i < value; i++) {
+            data.push(dataToPush);
+        }
+    }
+}

tests/load/contracts/TrieStressTest.sol

@@ -42,7 +42,18 @@ func NewRandomTest(
 		return nil, err
 	}
 
-	_, tx, contract, err := contracts.DeployEVMLoadSimulator(txOpts, worker.Client)
+	_, tx, simulatorContract, err := contracts.DeployEVMLoadSimulator(txOpts, worker.Client)
+	if err != nil {
+		return nil, err
+	}
+
+	if _, err := bind.WaitDeployed(ctx, worker.Client, tx); err != nil {
+		return nil, err
+	}
+
+	worker.Nonce++
+
+	_, tx, trieContract, err := contracts.DeployTrieStressTest(txOpts, worker.Client)
 	if err != nil {
 		return nil, err
 	}
@@ -74,66 +85,73 @@ func NewRandomTest(
 		},
 		{
 			Test: ReadTest{
-				Contract: contract,
+				Contract: simulatorContract,
 				Count:    count,
 			},
 			Weight: weight,
 		},
 		{
 			Test: WriteTest{
-				Contract: contract,
+				Contract: simulatorContract,
 				Count:    count,
 			},
 			Weight: weight,
 		},
 		{
 			Test: StateModificationTest{
-				Contract: contract,
+				Contract: simulatorContract,
 				Count:    count,
 			},
 			Weight: weight,
 		},
 		{
 			Test: HashingTest{
-				Contract: contract,
+				Contract: simulatorContract,
 				Count:    count,
 			},
 			Weight: weight,
 		},
 		{
 			Test: MemoryTest{
-				Contract: contract,
+				Contract: simulatorContract,
 				Count:    count,
 			},
 			Weight: weight,
 		},
 		{
 			Test: CallDepthTest{
-				Contract: contract,
+				Contract: simulatorContract,
 				Count:    count,
 			},
 			Weight: weight,
 		},
 		{
-			Test:   ContractCreationTest{Contract: contract},
+			Test:   ContractCreationTest{Contract: simulatorContract},
 			Weight: weight,
 		},
 		{
 			Test: PureComputeTest{
-				Contract:      contract,
+				Contract:      simulatorContract,
 				NumIterations: count,
 			},
 			Weight: weight,
 		},
 		{
 			Test: LargeEventTest{
-				Contract:  contract,
+				Contract:  simulatorContract,
 				NumEvents: count,
 			},
 			Weight: weight,
 		},
 		{
-			Test:   ExternalCallTest{Contract: contract},
+			Test:   ExternalCallTest{Contract: simulatorContract},
+			Weight: weight,
+		},
+		{
+			Test: TrieStressTest{
+				Contract:  trieContract,
+				NumValues: count,
+			},
 			Weight: weight,
 		},
 	}
@@ -383,6 +401,21 @@ func (e ExternalCallTest) Run(
 	executeContractTx(tc, ctx, wallet, e.Contract.SimulateExternalCall)
 }
 
+type TrieStressTest struct {
+	Contract  *contracts.TrieStressTest
+	NumValues *big.Int
+}
+
+func (t TrieStressTest) Run(
+	tc tests.TestContext,
+	ctx context.Context,
+	wallet *Wallet,
+) {
+	executeContractTx(tc, ctx, wallet, func(txOpts *bind.TransactOpts) (*types.Transaction, error) {
+		return t.Contract.WriteValues(txOpts, t.NumValues)
+	})
+}
+
 func executeContractTx(
 	tc tests.TestContext,
 	ctx context.Context,