Add custom event for testing

src/ert/run_models/base_run_model.py

@@ -12,9 +12,10 @@
 from collections import defaultdict
 from collections.abc import Generator, MutableSequence
 from contextlib import contextmanager
+from dataclasses import dataclass
 from pathlib import Path
 from queue import SimpleQueue
-from typing import TYPE_CHECKING, Any, cast
+from typing import TYPE_CHECKING, Any, cast, Literal
 
 import numpy as np
 
@@ -81,6 +82,13 @@
 if TYPE_CHECKING:
     from ert.config import QueueConfig
 
+
+@dataclass
+class CustomEvent:
+    msg: str
+    event_type: Literal["CustomEvent"] = "CustomEvent"
+
+
 StatusEvents = (
     FullSnapshotEvent
     | SnapshotUpdateEvent
@@ -94,6 +102,7 @@
     | RunModelUpdateBeginEvent
     | RunModelDataEvent
     | RunModelUpdateEndEvent
+    | CustomEvent
 )
 
 
@@ -237,7 +246,10 @@ def group(cls) -> str | None:
         return None
 
     def send_event(self, event: StatusEvents) -> None:
-        self._status_queue.put(event)
+        if isinstance(event, EndEvent):
+            self._status_queue.put(event)
+        else:
+            self._status_queue.put(CustomEvent(msg="Hello world"))
 
     def send_smoother_event(
         self, iteration: int, run_id: uuid.UUID, event: AnalysisEvent