chore: import upstream snapshot with attribution

tests/unittests/utils/test_env_utils.py

@@ -0,0 +1,84 @@
+# Copyright 2026 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 warnings
+
+from google.adk.utils.env_utils import is_enterprise_mode_enabled
+from google.adk.utils.env_utils import is_env_enabled
+import pytest
+
+
+@pytest.mark.parametrize(
+    'env_value,expected',
+    [
+        ('true', True),
+        ('TRUE', True),
+        ('TrUe', True),
+        ('1', True),
+        ('false', False),
+        ('FALSE', False),
+        ('0', False),
+        ('', False),
+    ],
+)
+def test_is_env_enabled(monkeypatch, env_value, expected):
+  """Test is_env_enabled with various environment variable values."""
+  monkeypatch.setenv('TEST_FLAG', env_value)
+  assert is_env_enabled('TEST_FLAG') is expected
+
+
+@pytest.mark.parametrize(
+    'default,expected',
+    [
+        ('0', False),
+        ('1', True),
+        ('true', True),
+    ],
+)
+def test_is_env_enabled_with_defaults(monkeypatch, default, expected):
+  """Test is_env_enabled when env var is not set with different defaults."""
+  monkeypatch.delenv('TEST_FLAG', raising=False)
+  assert is_env_enabled('TEST_FLAG', default=default) is expected
+
+
+def test_is_enterprise_mode_enabled_via_enterprise_env(monkeypatch):
+  """Enterprise mode is on when GOOGLE_GENAI_USE_ENTERPRISE is truthy."""
+  monkeypatch.setenv('GOOGLE_GENAI_USE_ENTERPRISE', 'true')
+
+  assert is_enterprise_mode_enabled() is True
+
+
+def test_is_enterprise_mode_enabled_falls_back_to_vertexai_with_warning(
+    monkeypatch,
+):
+  """The deprecated GOOGLE_GENAI_USE_VERTEXAI still enables enterprise mode and warns."""
+  monkeypatch.delenv('GOOGLE_GENAI_USE_ENTERPRISE', raising=False)
+  monkeypatch.setenv('GOOGLE_GENAI_USE_VERTEXAI', 'true')
+
+  with warnings.catch_warnings(record=True) as caught:
+    warnings.simplefilter('always')
+    result = is_enterprise_mode_enabled()
+
+  assert result is True
+  assert len(caught) == 1
+  assert issubclass(caught[-1].category, DeprecationWarning)
+  assert 'GOOGLE_GENAI_USE_VERTEXAI is deprecated' in str(caught[-1].message)
+
+
+def test_is_enterprise_mode_enabled_defaults_to_false(monkeypatch):
+  """Enterprise mode is off when no relevant env var is set."""
+  monkeypatch.delenv('GOOGLE_GENAI_USE_ENTERPRISE', raising=False)
+  monkeypatch.delenv('GOOGLE_GENAI_USE_VERTEXAI', raising=False)
+
+  assert is_enterprise_mode_enabled() is False