ADK changes

google-genai-bot · copybara-github · commit 8076b2294ffa · 2025-05-23T14:41:27.000-07:00
PiperOrigin-RevId: 762573275
diff --git a/tests/unittests/cli/utils/test_cli_tools_click.py b/tests/unittests/cli/utils/test_cli_tools_click.py
@@ -28,6 +28,7 @@
 import click
 from click.testing import CliRunner
 from google.adk.cli import cli_tools_click
+from google.adk.evaluation import local_eval_set_results_manager
 import pytest
 
 
@@ -255,9 +256,17 @@ def __init__(self, metric_name: str, threshold: float) -> None:
 
   class _EvalCaseResult:
 
-    def __init__(self, eval_set_id: str, final_eval_status: str) -> None:
+    def __init__(
+        self,
+        eval_set_id: str,
+        final_eval_status: str,
+        user_id: str,
+        session_id: str,
+    ) -> None:
       self.eval_set_id = eval_set_id
       self.final_eval_status = final_eval_status
+      self.user_id = user_id
+      self.session_id = session_id
 
   class EvalCase:
 
@@ -266,9 +275,19 @@ def __init__(self, eval_id: str):
 
   class EvalSet:
 
-    def __init__(self, eval_cases: list[EvalCase]):
+    def __init__(self, eval_set_id: str, eval_cases: list[EvalCase]):
+      self.eval_set_id = eval_set_id
       self.eval_cases = eval_cases
 
+  def mock_save_eval_set_result(cls, *args, **kwargs):
+    return None
+
+  monkeypatch.setattr(
+      local_eval_set_results_manager.LocalEvalSetResultsManager,
+      "save_eval_set_result",
+      mock_save_eval_set_result,
+  )
+
   # minimal enum-like namespace
   _EvalStatus = types.SimpleNamespace(PASSED="PASSED", FAILED="FAILED")
 
@@ -283,13 +302,14 @@ def __init__(self, eval_cases: list[EvalCase]):
   stub.try_get_reset_func = lambda _p: None
   stub.parse_and_get_evals_to_run = lambda _paths: {"set1.json": ["e1", "e2"]}
   eval_sets_manager_stub.load_eval_set_from_file = lambda x, y: EvalSet(
+      "test_eval_set_id",
       [EvalCase("e1"), EvalCase("e2")]
   )
 
   # Create an async generator function for run_evals
   async def mock_run_evals(*_a, **_k):
-    yield _EvalCaseResult("set1.json", "PASSED")
-    yield _EvalCaseResult("set1.json", "FAILED")
+    yield _EvalCaseResult("set1.json", "PASSED", "user", "session1")
+    yield _EvalCaseResult("set1.json", "FAILED", "user", "session2")
 
   stub.run_evals = mock_run_evals
 
