refactor(workflow): Remove run ID handling in tests

Removed node name mapping, run ID splitting, and related parameters
from testing utils as unique names are no longer generated for dynamic
nodes.

Change-Id: I2a1e96b0b37d62800b847f71720ddf51a43eb5bc

Author: boyangsvl

tests/unittests/workflow/testing_utils.py

@@ -141,37 +141,17 @@ def append_user_content(
   return event
 
 
-def _is_run_id_suffix(part: str) -> bool:
-  """Checks if a string part looks like an auto-generated run ID suffix."""
-  return len(part) == 15 or part.isdigit()
-
-
-def _split_name_and_run_id(name: str) -> tuple[str, str | None]:
-  """Splits a node name into base name and run ID suffix if present."""
-  for sep in ('@', '_'):
-    if sep in name:
-      parts = name.split(sep)
-      if _is_run_id_suffix(parts[-1]):
-        return sep.join(parts[:-1]), parts[-1]
-  return name, None
-
-
 # Extracts the contents from the events and transform them into a list of
 # (author, simplified_content) tuples.
-def simplify_events(
-    events: list[Event], *, strip_run_id: bool = False
-) -> list[tuple[str, types.Part]]:
+def simplify_events(events: list[Event]) -> list[tuple[str, types.Part]]:
   res = []
   for event in events:
     if event.content:
       author = event.author
-      if strip_run_id:
-        author, _ = _split_name_and_run_id(author)
       res.append((author, simplify_content(event.content)))
   return res
 
 
-
 END_OF_AGENT = 'end_of_agent'
 
 

tests/unittests/workflow/workflow_testing_utils.py

@@ -41,7 +41,6 @@
 from pydantic import Field
 from typing_extensions import override
 
-from .testing_utils import _split_name_and_run_id
 from .testing_utils import END_OF_AGENT
 from .testing_utils import simplify_content
 
@@ -183,19 +182,6 @@ async def create_parent_invocation_context(
   )
 
 
-def _build_node_name_map(events: list[Event]) -> dict[str, str]:
-  """Builds a map from node name to source node name from state updates."""
-  node_name_map = {}
-  for event in events:
-    if event.actions.agent_state:
-      nodes = event.actions.agent_state.get('nodes', {})
-      for node_name, node_state in nodes.items():
-        source_name = node_state.get('source_node_name')
-        if source_name:
-          node_name_map[node_name] = source_name
-  return node_name_map
-
-
 def simplify_event_with_node(
     event: Event,
     include_state_delta: bool = False,
@@ -240,14 +226,9 @@ def simplify_events_with_node(
     events: list[Event],
     *,
     include_state_delta: bool = False,
-    map_dynamic_node_to_the_source: bool = False,
     include_workflow_output: bool = False,
 ) -> list[tuple[str, Any]]:
   results = []
-  node_name_map = {}
-
-  if map_dynamic_node_to_the_source:
-    node_name_map = _build_node_name_map(events)
 
   # Second pass: Simplify events
   for event in events:
@@ -268,11 +249,7 @@ def simplify_events_with_node(
       if hasattr(event, 'node_info') and event.node_info.path:
         author = event.node_info.path
       else:
-        author = node_name_map.get(event.author, event.author)
-        # Strip auto-generated dynamic execution run IDs or counter numeric indices
-        # from dynamic child/standalone parallel execution authors so assertions
-        # match base node definition names consistently.
-        author, _ = _split_name_and_run_id(author)
+        author = event.author
       results.append((author, simplified_event))
   return results
 
@@ -283,7 +260,6 @@ def simplify_events_with_node_and_agent_state(
     include_state_delta: bool = False,
     include_inputs_and_triggers: bool = False,
     include_resume_inputs: bool = False,
-    map_dynamic_node_to_the_source: bool = False,
     include_workflow_output: bool = False,
 ):
   fields_to_exclude = {'run_id'}
@@ -293,10 +269,6 @@ def simplify_events_with_node_and_agent_state(
     fields_to_exclude.add('resume_inputs')
 
   results = []
-  node_name_map = {}
-
-  if map_dynamic_node_to_the_source:
-    node_name_map = _build_node_name_map(events)
 
   for event in events:
     # Optionally skip top-level workflow output events.
@@ -313,7 +285,7 @@ def simplify_events_with_node_and_agent_state(
     if hasattr(event, 'node_info') and event.node_info.path:
       author = event.node_info.path
     else:
-      author = node_name_map.get(event.author, event.author)
+      author = event.author
 
     if simplified_event:
       results.append((author, simplified_event))