google-adk 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

google/adk/tests/unittests/tools/application_integration_tool/test_application_integration_toolset.py

@@ -0,0 +1,345 @@
+# 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
+from unittest import mock
+
+from google.adk.auth.auth_credential import AuthCredential
+from google.adk.tools.application_integration_tool.application_integration_toolset import ApplicationIntegrationToolset
+from google.adk.tools.openapi_tool.openapi_spec_parser import rest_api_tool
+import pytest
+
+
+@pytest.fixture
+def mock_integration_client():
+  with mock.patch(
+      "google.adk.tools.application_integration_tool.application_integration_toolset.IntegrationClient"
+  ) as mock_client:
+    yield mock_client
+
+
+@pytest.fixture
+def mock_connections_client():
+  with mock.patch(
+      "google.adk.tools.application_integration_tool.application_integration_toolset.ConnectionsClient"
+  ) as mock_client:
+    yield mock_client
+
+
+@pytest.fixture
+def mock_openapi_toolset():
+  with mock.patch(
+      "google.adk.tools.application_integration_tool.application_integration_toolset.OpenAPIToolset"
+  ) as mock_toolset:
+    mock_toolset_instance = mock.MagicMock()
+    mock_rest_api_tool = mock.MagicMock(spec=rest_api_tool.RestApiTool)
+    mock_rest_api_tool.name = "Test Tool"
+    mock_toolset_instance.get_tools.return_value = [mock_rest_api_tool]
+    mock_toolset.return_value = mock_toolset_instance
+    yield mock_toolset
+
+
+@pytest.fixture
+def project():
+  return "test-project"
+
+
+@pytest.fixture
+def location():
+  return "us-central1"
+
+
+@pytest.fixture
+def integration_spec():
+  return {"openapi": "3.0.0", "info": {"title": "Integration API"}}
+
+
+@pytest.fixture
+def connection_spec():
+  return {"openapi": "3.0.0", "info": {"title": "Connection API"}}
+
+
+@pytest.fixture
+def connection_details():
+  return {"serviceName": "test-service", "host": "test.host"}
+
+
+def test_initialization_with_integration_and_trigger(
+    project,
+    location,
+    mock_integration_client,
+    mock_connections_client,
+    mock_openapi_toolset,
+):
+  integration_name = "test-integration"
+  trigger_name = "test-trigger"
+  toolset = ApplicationIntegrationToolset(
+      project, location, integration=integration_name, trigger=trigger_name
+  )
+  mock_integration_client.assert_called_once_with(
+      project, location, integration_name, trigger_name, None, None, None, None
+  )
+  mock_integration_client.return_value.get_openapi_spec_for_integration.assert_called_once()
+  mock_connections_client.assert_not_called()
+  mock_openapi_toolset.assert_called_once()
+  assert len(toolset.get_tools()) == 1
+  assert toolset.get_tools()[0].name == "Test Tool"
+
+
+def test_initialization_with_connection_and_entity_operations(
+    project,
+    location,
+    mock_integration_client,
+    mock_connections_client,
+    mock_openapi_toolset,
+    connection_details,
+):
+  connection_name = "test-connection"
+  entity_operations_list = ["list", "get"]
+  tool_name = "My Connection Tool"
+  tool_instructions = "Use this tool to manage entities."
+  mock_connections_client.return_value.get_connection_details.return_value = (
+      connection_details
+  )
+  toolset = ApplicationIntegrationToolset(
+      project,
+      location,
+      connection=connection_name,
+      entity_operations=entity_operations_list,
+      tool_name=tool_name,
+      tool_instructions=tool_instructions,
+  )
+  mock_integration_client.assert_called_once_with(
+      project,
+      location,
+      None,
+      None,
+      connection_name,
+      entity_operations_list,
+      None,
+      None,
+  )
+  mock_connections_client.assert_called_once_with(
+      project, location, connection_name, None
+  )
+  mock_connections_client.return_value.get_connection_details.assert_called_once()
+  mock_integration_client.return_value.get_openapi_spec_for_connection.assert_called_once_with(
+      tool_name,
+      tool_instructions
+      + f"ALWAYS use serviceName = {connection_details['serviceName']}, host ="
+      f" {connection_details['host']} and the connection name ="
+      f" projects/{project}/locations/{location}/connections/{connection_name} when"
+      " using this tool. DONOT ask the user for these values as you already"
+      " have those.",
+  )
+  mock_openapi_toolset.assert_called_once()
+  assert len(toolset.get_tools()) == 1
+  assert toolset.get_tools()[0].name == "Test Tool"
+
+
+def test_initialization_with_connection_and_actions(
+    project,
+    location,
+    mock_integration_client,
+    mock_connections_client,
+    mock_openapi_toolset,
+    connection_details,
+):
+  connection_name = "test-connection"
+  actions_list = ["create", "delete"]
+  tool_name = "My Actions Tool"
+  tool_instructions = "Perform actions using this tool."
+  mock_connections_client.return_value.get_connection_details.return_value = (
+      connection_details
+  )
+  toolset = ApplicationIntegrationToolset(
+      project,
+      location,
+      connection=connection_name,
+      actions=actions_list,
+      tool_name=tool_name,
+      tool_instructions=tool_instructions,
+  )
+  mock_integration_client.assert_called_once_with(
+      project, location, None, None, connection_name, None, actions_list, None
+  )
+  mock_connections_client.assert_called_once_with(
+      project, location, connection_name, None
+  )
+  mock_connections_client.return_value.get_connection_details.assert_called_once()
+  mock_integration_client.return_value.get_openapi_spec_for_connection.assert_called_once_with(
+      tool_name,
+      tool_instructions
+      + f"ALWAYS use serviceName = {connection_details['serviceName']}, host ="
+      f" {connection_details['host']} and the connection name ="
+      f" projects/{project}/locations/{location}/connections/{connection_name} when"
+      " using this tool. DONOT ask the user for these values as you already"
+      " have those.",
+  )
+  mock_openapi_toolset.assert_called_once()
+  assert len(toolset.get_tools()) == 1
+  assert toolset.get_tools()[0].name == "Test Tool"
+
+
+def test_initialization_without_required_params(project, location):
+  with pytest.raises(
+      ValueError,
+      match=(
+          "Either \\(integration and trigger\\) or \\(connection and"
+          " \\(entity_operations or actions\\)\\) should be provided."
+      ),
+  ):
+    ApplicationIntegrationToolset(project, location)
+
+  with pytest.raises(
+      ValueError,
+      match=(
+          "Either \\(integration and trigger\\) or \\(connection and"
+          " \\(entity_operations or actions\\)\\) should be provided."
+      ),
+  ):
+    ApplicationIntegrationToolset(project, location, integration="test")
+
+  with pytest.raises(
+      ValueError,
+      match=(
+          "Either \\(integration and trigger\\) or \\(connection and"
+          " \\(entity_operations or actions\\)\\) should be provided."
+      ),
+  ):
+    ApplicationIntegrationToolset(project, location, trigger="test")
+
+  with pytest.raises(
+      ValueError,
+      match=(
+          "Either \\(integration and trigger\\) or \\(connection and"
+          " \\(entity_operations or actions\\)\\) should be provided."
+      ),
+  ):
+    ApplicationIntegrationToolset(project, location, connection="test")
+
+
+def test_initialization_with_service_account_credentials(
+    project, location, mock_integration_client, mock_openapi_toolset
+):
+  service_account_json = json.dumps({
+      "type": "service_account",
+      "project_id": "dummy",
+      "private_key_id": "dummy",
+      "private_key": "dummy",
+      "client_email": "test@example.com",
+      "client_id": "131331543646416",
+      "auth_uri": "https://accounts.google.com/o/oauth2/auth",
+      "token_uri": "https://oauth2.googleapis.com/token",
+      "auth_provider_x509_cert_url": (
+          "https://www.googleapis.com/oauth2/v1/certs"
+      ),
+      "client_x509_cert_url": (
+          "http://www.googleapis.com/robot/v1/metadata/x509/dummy%40dummy.com"
+      ),
+      "universe_domain": "googleapis.com",
+  })
+  integration_name = "test-integration"
+  trigger_name = "test-trigger"
+  toolset = ApplicationIntegrationToolset(
+      project,
+      location,
+      integration=integration_name,
+      trigger=trigger_name,
+      service_account_json=service_account_json,
+  )
+  mock_integration_client.assert_called_once_with(
+      project,
+      location,
+      integration_name,
+      trigger_name,
+      None,
+      None,
+      None,
+      service_account_json,
+  )
+  mock_openapi_toolset.assert_called_once()
+  _, kwargs = mock_openapi_toolset.call_args
+  assert isinstance(kwargs["auth_credential"], AuthCredential)
+  assert (
+      kwargs[
+          "auth_credential"
+      ].service_account.service_account_credential.client_email
+      == "test@example.com"
+  )
+
+
+def test_initialization_without_explicit_service_account_credentials(
+    project, location, mock_integration_client, mock_openapi_toolset
+):
+  integration_name = "test-integration"
+  trigger_name = "test-trigger"
+  toolset = ApplicationIntegrationToolset(
+      project, location, integration=integration_name, trigger=trigger_name
+  )
+  mock_integration_client.assert_called_once_with(
+      project, location, integration_name, trigger_name, None, None, None, None
+  )
+  mock_openapi_toolset.assert_called_once()
+  _, kwargs = mock_openapi_toolset.call_args
+  assert isinstance(kwargs["auth_credential"], AuthCredential)
+  assert kwargs["auth_credential"].service_account.use_default_credential
+
+
+def test_get_tools(
+    project, location, mock_integration_client, mock_openapi_toolset
+):
+  integration_name = "test-integration"
+  trigger_name = "test-trigger"
+  toolset = ApplicationIntegrationToolset(
+      project, location, integration=integration_name, trigger=trigger_name
+  )
+  tools = toolset.get_tools()
+  assert len(tools) == 1
+  assert isinstance(tools[0], rest_api_tool.RestApiTool)
+  assert tools[0].name == "Test Tool"
+
+
+def test_initialization_with_connection_details(
+    project,
+    location,
+    mock_integration_client,
+    mock_connections_client,
+    mock_openapi_toolset,
+):
+  connection_name = "test-connection"
+  entity_operations_list = ["list"]
+  tool_name = "My Connection Tool"
+  tool_instructions = "Use this tool."
+  mock_connections_client.return_value.get_connection_details.return_value = {
+      "serviceName": "custom-service",
+      "host": "custom.host",
+  }
+  toolset = ApplicationIntegrationToolset(
+      project,
+      location,
+      connection=connection_name,
+      entity_operations=entity_operations_list,
+      tool_name=tool_name,
+      tool_instructions=tool_instructions,
+  )
+  mock_integration_client.return_value.get_openapi_spec_for_connection.assert_called_once_with(
+      tool_name,
+      tool_instructions
+      + "ALWAYS use serviceName = custom-service, host = custom.host and the"
+      " connection name ="
+      " projects/test-project/locations/us-central1/connections/test-connection"
+      " when using this tool. DONOT ask the user for these values as you"
+      " already have those.",
+  )

google/adk/tests/unittests/tools/google_api_tool/__init__.py

@@ -0,0 +1,13 @@
+# 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.