feat: add support for custom assertoor images & use assertoor image with...

feat: add support for custom assertoor images & use assertoor image with verkle support for verkle chains (#483)

README.md

@@ -385,6 +385,10 @@ goomy_blob_params:
 
 # Configuration place for the assertoor testing tool - https:#github.com/ethpandaops/assertoor
 assertoor_params:
+  # Assertoor docker image to use
+  # Leave blank to use the default image according to your network params
+  image: ""
+
   # Check chain stability
   # This check monitors the chain and succeeds if:
   # - all clients are synced

main.star

@@ -390,6 +390,7 @@ def run(plan, args={}):
                 assertoor_config_template,
                 all_participants,
                 args_with_right_defaults.participants,
+                network_params,
                 assertoor_params,
             )
             plan.print("Successfully launched assertoor")

src/assertoor/assertoor_launcher.star

@@ -34,6 +34,7 @@ def launch_assertoor(
     config_template,
     participant_contexts,
     participant_configs,
+    network_params,
     assertoor_params,
 ):
     all_client_info = []
@@ -88,18 +89,30 @@ def launch_assertoor(
     config = get_config(
         config_files_artifact_name,
         tests_config_artifacts_name,
+        network_params,
+        assertoor_params,
     )
 
     plan.add_service(SERVICE_NAME, config)
 
 
-def get_config(config_files_artifact_name, tests_config_artifacts_name):
+def get_config(
+    config_files_artifact_name,
+    tests_config_artifacts_name,
+    network_params,
+    assertoor_params,
+):
     config_file_path = shared_utils.path_join(
         ASSERTOOR_CONFIG_MOUNT_DIRPATH_ON_SERVICE,
         ASSERTOOR_CONFIG_FILENAME,
     )
 
-    IMAGE_NAME = "ethpandaops/assertoor:master"
+    if assertoor_params.image != "":
+        IMAGE_NAME = assertoor_params.image
+    elif network_params.electra_fork_epoch != None:
+        IMAGE_NAME = "ethpandaops/assertoor:verkle-support"
+    else:
+        IMAGE_NAME = "ethpandaops/assertoor:master"
 
     return ServiceConfig(
         image=IMAGE_NAME,

src/package_io/input_parser.star

@@ -236,6 +236,7 @@ def input_parser(plan, input_args):
             goomy_blob_args=result["goomy_blob_params"]["goomy_blob_args"],
         ),
         assertoor_params=struct(
+            image=result["assertoor_params"]["image"],
             run_stability_check=result["assertoor_params"]["run_stability_check"],
             run_block_proposal_check=result["assertoor_params"][
                 "run_block_proposal_check"
@@ -614,6 +615,7 @@ def get_default_goomy_blob_params():
 
 def get_default_assertoor_params():
     return {
+        "image": "",
         "run_stability_check": True,
         "run_block_proposal_check": True,
         "run_lifecycle_test": False,