Files
Artemis-Eapil/tests/unit/test_gemini_role.py
T
somew1nd 9a5aaf99b7 feat: introduce MCP server, cross-platform runtime, and Showcase UI reorganization
- Showcase UI Reorganization:
  * Move web UI dashboard from root directory into apps/showcase_ui/ for cleaner project structure
  * Add 'artemis ui' CLI command (artemis/interfaces/cli/commands/ui.py) and launcher scripts for the interactive showcase console
- Standalone MCP Server & Notifiers (mcp_server/):
  * Implement dedicated MCP server package with mobile automation tools (device_state, inspect_trace, task_manager, task_runner)
  * Add multi-channel notifiers (AgentAPI, Desktop notifications, Webhooks, File output, and Composite routing)
  * Introduce background task runner and trace store for autonomous mobile execution monitoring
- Cross-Platform Runtime & Toolchain Support:
  * Add artemis/platform/ with OS-specific implementations for Linux, macOS (Darwin), and Windows
  * Add artemis/runtime/supervisor.py for resilient process and agent runtime supervision
  * Add artemis/toolchain/ with dependency resolution and SDK descriptors
  * Add cross-platform launcher scripts (start.sh, start.bat) and installer scripts (scripts/install_deps.*)
- Core Enhancements & Diagnostics:
  * Enhance coordinate normalization, OCR API fallback, and perception pipeline
  * Add credential validation (credentials_validator.py) and system diagnostics
  * Update bilingual documentation (README.md, README_CN.md) and architecture diagrams
- Test Suite Expansion:
  * Add integration tests for cloud gateway, remote ADB, and Artemis login
  * Add unit tests covering core diagnostics, platform detection, toolchain, credentials validator, optional OCR, and MCP tools
2026-08-18 13:59:27 -07:00

80 lines
2.5 KiB
Python

# 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.
from google import genai
from google.genai import types
from google.genai.errors import APIError
import pytest
pytestmark = pytest.mark.skip(
reason="Exploratory live API verification test for Gemini SDK role support"
)
@pytest.fixture
def client():
"""Initializes the Gemini client. Requires GEMINI_API_KEY in environment."""
import os
from artemis.config.settings import is_placeholder_key
key = os.environ.get("GEMINI_API_KEY") or os.environ.get("GOOGLE_API_KEY")
if not key or is_placeholder_key(key):
pytest.skip(
"Valid GEMINI_API_KEY / GOOGLE_API_KEY not configured for live Gemini role tests"
)
try:
return genai.Client()
except Exception as e:
pytest.skip(f"Could not initialize genai.Client: {e}")
def test_gemini_role_tool_fails(client):
"""Confirm that using role='tool' results in a 400 APIError from Gemini."""
contents = [
types.Content(
role="tool",
parts=[types.Part.from_text(text="This is a test payload for tool execution.")],
)
]
with pytest.raises(APIError) as exc_info:
client.models.generate_content(
model="gemini-3.7-flash",
contents=contents,
)
# Assert it failed specifically due to the invalid role string
error_msg = str(exc_info.value)
assert "Invalid role string" in error_msg or "400" in error_msg
def test_gemini_role_user_succeeds(client):
"""Confirm that using role='user' (which is the correct fix) succeeds."""
contents = [
types.Content(
role="user",
parts=[
types.Part.from_text(text="Acknowledge this message with 'ok' and nothing else.")
],
)
]
# This should not raise any exceptions
response = client.models.generate_content(
model="gemini-3.7-flash",
contents=contents,
)
assert response.text is not None