chore: add replay tests for agent identity fields (identity_type and effective_identity)

PiperOrigin-RevId: 836547231

Author: petjahn

tests/unit/vertexai/genai/replays/test_create_agent_engine.py

@@ -15,6 +15,7 @@
 # pylint: disable=protected-access,bad-continuation,missing-function-docstring
 
 import os
+import re
 
 from tests.unit.vertexai.genai.replays import pytest_helper
 from vertexai._genai import types
@@ -23,6 +24,10 @@
     {"name": "query", "api_mode": ""},
 ]
 
+_AGENT_IDENTITY_REGEX = re.compile(
+    "agents.global.org-[0-9]+.system.id.goog/resources/aiplatform/projects/[0-9]+/locations/us-central1/reasoningEngines/[0-9a-zA-Z]+"
+)
+
 
 def test_create_config_lightweight(client):
     agent_display_name = "test-display-name"
@@ -146,6 +151,25 @@ def test_create_with_source_packages(
         client.agent_engines.delete(name=agent_engine.api_resource.name, force=True)
 
 
+def test_create_with_identity_type(client):
+    """Tests creating an agent engine with identity type."""
+    agent_engine = client.agent_engines.create(
+        config={
+            "identity_type": types.IdentityType.AGENT_IDENTITY,
+            "http_options": {"api_version": "v1beta1"},
+        },
+    )
+    assert (
+        agent_engine.api_resource.spec.identity_type
+        == types.IdentityType.AGENT_IDENTITY
+    )
+    assert _AGENT_IDENTITY_REGEX.match(
+        agent_engine.api_resource.spec.effective_identity
+    )
+    # Clean up resources.
+    client.agent_engines.delete(name=agent_engine.api_resource.name, force=True)
+
+
 pytestmark = pytest_helper.setup(
     file=__file__,
     globals_for_file=globals(),