feat(fill): set sig scheme via CLI flag for filler, default to test

Author: fselmo

packages/testing/src/consensus_testing/test_types/genesis.py

@@ -4,7 +4,7 @@
 
 from lean_spec.subspecs.containers.state import State, Validators
 from lean_spec.subspecs.containers.validator import Validator
-from lean_spec.types import Uint64
+from lean_spec.types import Bytes52, Uint64
 
 from ..keys import get_shared_key_manager
 
@@ -35,7 +35,10 @@ def generate_pre_state(**kwargs: Any) -> State:
 
     validators = Validators(
         data=[
-            Validator(pubkey=key_manager[Uint64(i)].public.encode_bytes(), index=Uint64(i))
+            Validator(
+                pubkey=Bytes52(key_manager[Uint64(i)].public.encode_bytes()),
+                index=Uint64(i),
+            )
             for i in range(num_validators)
         ]
     )

packages/testing/src/framework/cli/fill.py

@@ -1,5 +1,6 @@
 """Unified CLI command for generating Ethereum test fixtures across all layers."""
 
+import os
 import sys
 from pathlib import Path
 from typing import Sequence
@@ -37,6 +38,12 @@
     is_flag=True,
     help="Clean output directory before generating",
 )
+@click.option(
+    "--scheme",
+    type=click.Choice(["test", "prod"], case_sensitive=False),
+    default="test",
+    help="XMSS signature scheme (default: test)",
+)
 @click.pass_context
 def fill(
     ctx: click.Context,
@@ -45,6 +52,7 @@ def fill(
     fork: str,
     layer: str,
     clean: bool,
+    scheme: str,
 ) -> None:
     """
     Generate Ethereum test fixtures from test specifications.
@@ -61,7 +69,15 @@ def fill(
 
         # Default layer is consensus
         fill tests/spec_tests/devnet --fork=Devnet --clean -v
+
+        # Use specific XMSS scheme (overrides LEAN_ENV env var)
+        fill --fork=Devnet --scheme=prod --clean -v
     """
+    # Note: It's important to never import any leanSpec modules in this file, so the
+    # `LEAN_ENV` variable can be set before the config loads its value from the
+    # environment.
+    os.environ["LEAN_ENV"] = scheme.lower()
+
     config_path = Path(__file__).parent / "pytest_ini_files" / "pytest-fill.ini"
     # Find project root by looking for pyproject.toml with [tool.uv.workspace]
     project_root = Path.cwd()

tests/conftest.py

@@ -7,8 +7,6 @@
 if "LEAN_ENV" not in os.environ:
     os.environ["LEAN_ENV"] = "test"
 
-import lean_spec.subspecs.xmss as xmss
-
 # Create a profile named "no_deadline" with deadline disabled.
 settings.register_profile("no_deadline", deadline=None)
 settings.load_profile("no_deadline")