add BASE support (#35)

- add base mainnet and testnet to test data
- enable test to automatically pull recent blocks for test data if chain ID not defined in CHAIN_INFO
- upgrade locust and click versions, and change psycopg2 to psycopg2-binary

Author: erwin-wee

chainbench/profile/ethereum/general.py

@@ -32,8 +32,8 @@ def call_task(self):
             method="eth_call",
             params=[
                 {
-                    "to": "0xdAC17F958D2ee523a2206206994597C13D831ec7",
-                    "data": "0x70a082310000000000000000000000009696f59E4d72E237BE84fFD425DCaD154Bf96976",  # noqa: E501
+                    "to": "0x8BB4C975Ff3c250e0ceEA271728547f3802B36Fd",
+                    "data": "0x734e7fb90000000000000000000000000000000000000000000000000000000064d40c600000000000000000000000000000000000000000000000000000000064d43690",  # noqa: E501
                 },
                 "latest",
             ],

chainbench/test_data/evm.py

@@ -1,8 +1,11 @@
+import logging
 from typing import Mapping
 
 from chainbench.test_data.base import BaseTestData, BlockchainData, Blocks, ChainInfo
 from chainbench.util.rng import get_rng
 
+logger = logging.getLogger(__name__)
+
 
 class EVMTestData(BaseTestData):
     TXS_REQUIRED = 100
@@ -35,6 +38,16 @@ class EVMTestData(BaseTestData):
             "start_block": 20000000,
             "end_block": 32000000,
         },
+        8453: {
+            "name": "base-mainnet",
+            "start_block": 1,
+            "end_block": 2400000,
+        },
+        84531: {
+            "name": "base-testnet",
+            "start_block": 1,
+            "end_block": 8200000,
+        },
     }
 
     def _fetch_chain_id(self) -> int:
@@ -57,23 +70,29 @@ def _fetch_random_block(self, start, end, return_txs=True) -> tuple[int, dict]:
         block_number = rng.random.randint(start, end)
         return self._fetch_block(block_number, return_txs=return_txs)
 
+    def _get_start_and_end_blocks(self, use_recent_blocks: bool) -> tuple[int, int]:
+        chain_id: int = self._fetch_chain_id()
+        if not use_recent_blocks and chain_id in self.CHAIN_INFO:
+            start_block_number = self.CHAIN_INFO[chain_id]["start_block"]
+            end_block_number = self.CHAIN_INFO[chain_id]["end_block"]
+            logger.info("Using blocks from %s to %s as test data", start_block_number, end_block_number)
+            return start_block_number, end_block_number
+        else:
+            end_block_number = int(self._make_call("eth_blockNumber"), 0)
+            start_block_number = end_block_number - 20
+            logger.info("Using recent blocks from %s to %s as test data", start_block_number, end_block_number)
+            return start_block_number, end_block_number
+
     # get initial data from blockchain
     def _get_init_data(self, parsed_options) -> BlockchainData:
         txs: list[dict] = []
         tx_hashes: list[str] = []
         accounts: set[str] = set()
         blocks: Blocks = []
-        chain_id: int = self._fetch_chain_id()
         start_block_number: int
         end_block_number: int
         return_txs: bool
-
-        if parsed_options.use_recent_blocks:
-            end_block_number = int(self._make_call("eth_blockNumber"), 0)
-            start_block_number = end_block_number - 20
-        else:
-            start_block_number = self.CHAIN_INFO[chain_id]["start_block"]
-            end_block_number = self.CHAIN_INFO[chain_id]["end_block"]
+        start_block_number, end_block_number = self._get_start_and_end_blocks(parsed_options.use_recent_blocks)
 
         while self.TXS_REQUIRED > len(txs) or self.ACCOUNTS_REQUIRED > len(accounts) or self.SAVE_BLOCKS > len(blocks):
             if self.ACCOUNTS_REQUIRED > len(accounts) or self.TXS_REQUIRED > len(txs):