[Test] [history server] [collector] Ensure event type coverage

historyserver/config/rayjob.yaml

@@ -6,6 +6,7 @@ spec:
   entrypoint: |
     python -c "
     import ray
+    import time
     ray.init()
 
     @ray.remote(num_cpus=0.5)
@@ -35,6 +36,10 @@ spec:
     final_count = ray.get(counter.get_count.remote())
     print(f'Final count: {final_count}')
     print(f'Cluster resources: {ray.cluster_resources()}')
+
+    # For now, events on the worker nodes aren't sent to the collector when calling ray.shutdown().
+    # As a workaround, we explicitly wait for 5 seconds to ensure events are sent.
+    time.sleep(5)
     "
   # Select the existing Ray cluster running the collector.
   clusterSelector:

historyserver/pkg/eventserver/types/event.go

@@ -1,7 +1,10 @@
 package types
 
+// EventType is the Ray event type.
 type EventType string
 
+// There are 11 potential Ray event types:
+// https://github.com/ray-project/ray/blob/3b41c97fa90c58b0b72c0026f57005b92310160d/src/ray/protobuf/public/events_base_event.proto#L49-L61
 const (
 	EVENT_TYPE_UNSPECIFIED      EventType = "EVENT_TYPE_UNSPECIFIED"
 	TASK_DEFINITION_EVENT       EventType = "TASK_DEFINITION_EVENT"
@@ -15,3 +18,18 @@ const (
 	ACTOR_DEFINITION_EVENT      EventType = "ACTOR_DEFINITION_EVENT"
 	ACTOR_LIFECYCLE_EVENT       EventType = "ACTOR_LIFECYCLE_EVENT"
 )
+
+// AllEventTypes includes all potential event types defined in Ray.
+var AllEventTypes = []EventType{
+	EVENT_TYPE_UNSPECIFIED,
+	TASK_DEFINITION_EVENT,
+	TASK_LIFECYCLE_EVENT,
+	ACTOR_TASK_DEFINITION_EVENT,
+	TASK_PROFILE_EVENT,
+	DRIVER_JOB_DEFINITION_EVENT,
+	DRIVER_JOB_LIFECYCLE_EVENT,
+	NODE_DEFINITION_EVENT,
+	NODE_LIFECYCLE_EVENT,
+	ACTOR_DEFINITION_EVENT,
+	ACTOR_LIFECYCLE_EVENT,
+}