Fix test ensemble legacy

docs/ert/conf.py

@@ -68,7 +68,6 @@
     ("py:class", "pydantic.types.PositiveInt"),
     ("py:class", "LibresFacade"),
     ("py:class", "pandas.core.frame.DataFrame"),
-    ("py:class", "websockets.server.WebSocketServerProtocol"),
     ("py:class", "EnsembleReader"),
 ]
 nitpick_ignore_regex = [

src/ert/ensemble_evaluator/evaluator.py

@@ -269,8 +269,6 @@ async def listen_for_messages(self) -> None:
                             await self.forward_checksum(event)
                         else:
                             await self._events.put(event)
-                    # if type(event) in [EnsembleSucceeded, EnsembleFailed]:
-                    #     return
                 else:
                     logger.info(f"Connection attempt to unknown sender: {sender}.")
             except zmq.error.ZMQError as e:

src/ert/ensemble_evaluator/monitor.py

@@ -139,7 +139,6 @@ async def _receiver(self) -> None:
         except zmq.ZMQError as exc:
             logger.error(f"Failed to connect to {self._ee_con_info.router_uri} {exc}")
             raise
-
         await self._socket.send_multipart([b"", b"CONNECT"])
         self._connected.set()
 

tests/ert/unit_tests/ensemble_evaluator/test_ensemble_legacy.py

@@ -46,7 +46,7 @@ async def test_run_legacy_ensemble(
         )
         async with (
             evaluator_to_use(ensemble, config) as evaluator,
-            Monitor(config) as monitor,
+            Monitor(config.get_connection_info()) as monitor,
         ):
             async for event in monitor.track():
                 if type(event) in (
@@ -85,7 +85,7 @@ async def test_run_and_cancel_legacy_ensemble(
 
         async with (
             evaluator_to_use(ensemble, config) as evaluator,
-            Monitor(config) as monitor,
+            Monitor(config.get_connection_info()) as monitor,
         ):
             # on lesser hardware the realizations might be killed by max_runtime
             # and the ensemble is set to STOPPED