feat(plugins/consume): Check 8551 port on Engine tests

marioevz · marioevz · commit e248a3bd0094 · 2025-01-20T18:51:03.000Z
diff --git a/src/pytest_plugins/consume/hive_simulators/engine/conftest.py b/src/pytest_plugins/consume/hive_simulators/engine/conftest.py
@@ -6,7 +6,7 @@
 
 import io
 from pathlib import Path
-from typing import Mapping
+from typing import Dict, Mapping
 
 import pytest
 from hive.client import Client
@@ -20,6 +20,19 @@
 TestCase = TestCaseIndexFile | TestCaseStream
 
 
+@pytest.fixture(scope="function")
+def environment(request: pytest.FixtureRequest) -> Dict:
+    """
+    For the Engine API simulations, we need that the port 8551 is ready on the client's side,
+    instead of the default 8545, because the first request is a forkchoice update.
+
+    We signal this to hive by adding the "HIVE_CHECK_LIVE_PORT" environment variable.
+    """
+    environment: Dict = request.getfixturevalue("environment")
+    environment["HIVE_CHECK_LIVE_PORT"] = "8551"
+    return environment
+
+
 @pytest.fixture(scope="function")
 def engine_rpc(client: Client) -> EngineRPC:
     """Initialize engine RPC client for the execution client under test."""
