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

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

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

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