refactor: extract credentail key building logic to auth_config

PiperOrigin-RevId: 768124459

tests/unittests/auth/test_auth_config.py

@@ -0,0 +1,91 @@
+# 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.
+
+from fastapi.openapi.models import OAuth2
+from fastapi.openapi.models import OAuthFlowAuthorizationCode
+from fastapi.openapi.models import OAuthFlows
+from google.adk.auth.auth_credential import AuthCredential
+from google.adk.auth.auth_credential import AuthCredentialTypes
+from google.adk.auth.auth_credential import OAuth2Auth
+from google.adk.auth.auth_tool import AuthConfig
+import pytest
+
+
+class TestAuthConfig:
+  """Tests for the AuthConfig method."""
+
+
+@pytest.fixture
+def oauth2_auth_scheme():
+  """Create an OAuth2 auth scheme for testing."""
+  # Create the OAuthFlows object first
+  flows = OAuthFlows(
+      authorizationCode=OAuthFlowAuthorizationCode(
+          authorizationUrl="https://example.com/oauth2/authorize",
+          tokenUrl="https://example.com/oauth2/token",
+          scopes={"read": "Read access", "write": "Write access"},
+      )
+  )
+
+  # Then create the OAuth2 object with the flows
+  return OAuth2(flows=flows)
+
+
+@pytest.fixture
+def oauth2_credentials():
+  """Create OAuth2 credentials for testing."""
+  return AuthCredential(
+      auth_type=AuthCredentialTypes.OAUTH2,
+      oauth2=OAuth2Auth(
+          client_id="mock_client_id",
+          client_secret="mock_client_secret",
+          redirect_uri="https://example.com/callback",
+      ),
+  )
+
+
+@pytest.fixture
+def auth_config(oauth2_auth_scheme, oauth2_credentials):
+  """Create an AuthConfig for testing."""
+  # Create a copy of the credentials for the exchanged_auth_credential
+  exchanged_credential = oauth2_credentials.model_copy(deep=True)
+
+  return AuthConfig(
+      auth_scheme=oauth2_auth_scheme,
+      raw_auth_credential=oauth2_credentials,
+      exchanged_auth_credential=exchanged_credential,
+  )
+
+
+def test_get_credential_key(auth_config):
+  """Test generating a unique credential key."""
+
+  key = auth_config.get_credential_key()
+  assert key.startswith("adk_oauth2_")
+  assert "_oauth2_" in key
+
+
+def test_get_credential_key_with_extras(auth_config):
+  """Test generating a key when model_extra exists."""
+  # Add model_extra to test cleanup
+
+  original_key = auth_config.get_credential_key()
+  key = auth_config.get_credential_key()
+
+  auth_config.auth_scheme.model_extra["extra_field"] = "value"
+  auth_config.raw_auth_credential.model_extra["extra_field"] = "value"
+
+  assert original_key == key
+  assert "extra_field" in auth_config.auth_scheme.model_extra
+  assert "extra_field" in auth_config.raw_auth_credential.model_extra