Fix/eth beacon bug (#102)

Author: erwin-wee

chainbench/test_data/ethereum.py

@@ -149,10 +149,12 @@ def get_committee(committee: dict[str, t.Any]) -> EthCommittee:
                 validators=[EthValidator(index=validator["index"]) for validator in committee["validators"]],
             )
 
-        if isinstance(data, str):
+        if isinstance(data, (str, bytes)):
             data_dict = json.loads(data)
         else:
             data_dict = data
+        logger.debug(f"data_dict type {type(data_dict)}")
+        logger.debug("Data: %s", data_dict)
 
         slot = data_dict["block_number"]
         epoch = slot // 32

chainbench/user/protocol/ethereum.py

@@ -1,18 +1,20 @@
 import logging
 import typing as t
 
-from locust import task
+from locust import tag, task
 
 from chainbench.test_data.ethereum import EthBeaconTestData
 from chainbench.user.http import HttpUser
 from chainbench.user.wss import WSSubscription
+from chainbench.util.rng import RNGManager
 
 logger = logging.getLogger(__name__)
 
 
 class EthBeaconBaseUser(HttpUser):
     abstract = True
     test_data = EthBeaconTestData()
+    rng = RNGManager()
 
     def eth_beacon_blocks_request(
         self,
@@ -327,9 +329,11 @@ def eth_v1_node_version_task(self):
 
 
 class TestEthMethod(EthBeaconUser):
+    @tag("single")
     @task
     def run_task(self) -> None:
-        self.method_to_task_function(self.environment.parsed_options.method)()
+        self.logger.info(f"Running task {self.environment.parsed_options.method}")
+        self.method_to_task_function(self.environment.parsed_options.method)(self)
 
 
 class EthSubscribe(WSSubscription):