Remove ZMQ client from LegacyEnsemble

This commit changes LegacyEnsemble to put events directly in scheduler queue rather than sending it over ZMQ.

Author: jonathan-eq

src/ert/ensemble_evaluator/_ensemble.py

@@ -16,9 +16,7 @@
     FMEvent,
     ForwardModelStepFailure,
     ForwardModelStepSuccess,
-    event_to_json,
 )
-from _ert.forward_model_runner.client import Client
 from ert.config import ForwardModelStep, QueueConfig
 from ert.run_arg import RunArg
 from ert.scheduler import Scheduler, create_driver
@@ -185,14 +183,9 @@ def update_snapshot(self, events: Sequence[Event]) -> EnsembleSnapshot:
 
         return snapshot_mutate_event
 
-    async def send_event(
-        self,
-        event: Event,
-        retries: int = 10,
-    ) -> None:
-        assert self._config is not None
-        async with Client(self._config.get_uri(), token=self._config.token) as client:
-            await client.send(event_to_json(event), retries)
+    async def send_event(self, event: Event) -> None:
+        if self.outbound_event_queue is not None:
+            await self.outbound_event_queue.put(event)
 
     async def evaluate(
         self,
@@ -209,6 +202,7 @@ async def evaluate(
         the final result of executing all its jobs through a scheduler and driver.
         """
         self._config = config
+        self.outbound_event_queue = scheduler_queue
 
         if not self.id_:
             raise ValueError("Ensemble id not set")

tests/ert/unit_tests/ensemble_evaluator/test_ensemble_legacy.py

@@ -114,14 +114,6 @@ async def test_queue_config_properties_propagated_to_scheduler(
     monkeypatch.setattr(QueueConfig, "max_running", 44)
     ensemble._queue_config.max_submit = 55
 
-    async def mock_send_event_method(*args, **kwargs):
-        return
-
-    monkeypatch.setattr(
-        "ert.ensemble_evaluator._ensemble.LegacyEnsemble.send_event",
-        mock_send_event_method,
-    )
-
     # The function under test:
     await ensemble.evaluate(config=MagicMock())