No public description

PiperOrigin-RevId: 748777998

tests/integration/test_context_variable.py

@@ -0,0 +1,67 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import json
+
+import pytest
+
+# Skip until fixed.
+pytest.skip(allow_module_level=True)
+
+from .fixture import context_variable_agent
+from .utils import TestRunner
+
+
+@pytest.mark.parametrize(
+    "agent_runner",
+    [{"agent": context_variable_agent.agent.state_variable_echo_agent}],
+    indirect=True,
+)
+def test_context_variable_missing(agent_runner: TestRunner):
+  with pytest.raises(KeyError) as e_info:
+    agent_runner.run("Hi echo my customer id.")
+  assert "customerId" in str(e_info.value)
+
+
+@pytest.mark.parametrize(
+    "agent_runner",
+    [{"agent": context_variable_agent.agent.state_variable_update_agent}],
+    indirect=True,
+)
+def test_context_variable_update(agent_runner: TestRunner):
+  _call_function_and_assert(
+      agent_runner,
+      "update_fc",
+      ["RRRR", "3.141529", ["apple", "banana"], [1, 3.14, "hello"]],
+      "successfully",
+  )
+
+
+def _call_function_and_assert(
+    agent_runner: TestRunner, function_name: str, params, expected
+):
+  param_section = (
+      " with params"
+      f" {params if isinstance(params, str) else json.dumps(params)}"
+      if params is not None
+      else ""
+  )
+  agent_runner.run(
+      f"Call {function_name}{param_section} and show me the result"
+  )
+
+  model_response_event = agent_runner.get_events()[-1]
+  assert model_response_event.author == "context_variable_update_agent"
+  assert model_response_event.content.role == "model"
+  assert expected in model_response_event.content.parts[0].text.strip()