somew1nd
54fcef9887
- Add NOTICE file and README license references for mobile-use and finalrun-agent - Add derivative attribution headers to 17 files derived from mobile-use - Rename hopper agent to entity_extractor across prompts, tools, and configs - Remove obsolete adb_tunnel, ApiBaseUrl, and legacy CLI utility scripts - Replace legacy examples with native Artemis diagnostic and workflow examples - Update unit and tool tests to validate entity_extractor and SDK builders
112 lines
3.8 KiB
Python
112 lines
3.8 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.
|
|
|
|
"""Unit tests for Artemis SDK TaskRequestBuilder."""
|
|
|
|
from pathlib import Path
|
|
from pydantic import BaseModel
|
|
import pytest
|
|
|
|
from artemis.sdk.builders.task_request_builder import TaskRequestBuilder, TaskRequestCommonBuilder
|
|
from artemis.sdk.types.task import TaskRequest
|
|
|
|
|
|
class DummyOutputSchema(BaseModel):
|
|
summary: str
|
|
item_count: int
|
|
|
|
|
|
def test_task_request_builder_fluent_chaining():
|
|
"""Verify basic fluent configuration and build."""
|
|
req = (
|
|
TaskRequestBuilder(goal="Open Settings and verify battery level")
|
|
.with_max_steps(25)
|
|
.with_trace_recording(True, path="custom_traces")
|
|
.with_locked_app_package("com.android.settings")
|
|
.with_app_path("build/app.apk")
|
|
.using_profile("flash")
|
|
.with_name("Battery Check")
|
|
.build()
|
|
)
|
|
|
|
assert isinstance(req, TaskRequest)
|
|
assert req.goal == "Open Settings and verify battery level"
|
|
assert req.max_steps == 25
|
|
assert req.record_trace is True
|
|
assert req.trace_path == Path("custom_traces")
|
|
assert req.locked_app_package == "com.android.settings"
|
|
assert req.app_path == Path("build/app.apk")
|
|
assert req.profile == "flash"
|
|
assert req.task_name == "Battery Check"
|
|
|
|
|
|
def test_task_request_builder_with_output_format():
|
|
"""Verify builder binds structured schema properly."""
|
|
req = (
|
|
TaskRequestBuilder(goal="Extract items")
|
|
.with_output_format(DummyOutputSchema)
|
|
.build()
|
|
)
|
|
|
|
assert req.output_format is DummyOutputSchema
|
|
assert req.output_description is None
|
|
|
|
|
|
def test_task_request_builder_with_output_description():
|
|
"""Verify builder accepts unformatted string deliverables."""
|
|
req = (
|
|
TaskRequestBuilder(goal="Extract items")
|
|
.with_output_description("Return a numbered list")
|
|
.build()
|
|
)
|
|
|
|
assert req.output_description == "Return a numbered list"
|
|
assert req.output_format is None
|
|
|
|
|
|
def test_task_request_builder_conflicting_outputs_raises():
|
|
"""Verify that specifying both output_format and output_description raises ValueError."""
|
|
builder = (
|
|
TaskRequestBuilder(goal="Extract items")
|
|
.with_output_format(DummyOutputSchema)
|
|
.with_output_description("Description text")
|
|
)
|
|
with pytest.raises(ValueError, match="Cannot specify both output_format and output_description"):
|
|
builder.build()
|
|
|
|
|
|
def test_task_request_builder_empty_goal_raises():
|
|
"""Verify empty or whitespace-only goal raises ValueError."""
|
|
with pytest.raises(ValueError, match="Task goal must be a non-empty"):
|
|
TaskRequestBuilder(goal=" ").build()
|
|
|
|
|
|
def test_task_request_builder_from_common():
|
|
"""Verify task request initialization derived from common base settings."""
|
|
common_settings = (
|
|
TaskRequestCommonBuilder()
|
|
.with_max_steps(42)
|
|
.with_trace_recording(False)
|
|
.with_locked_app_package("org.artemis.target")
|
|
.with_app_path(Path("/tmp/target.apk"))
|
|
.build()
|
|
)
|
|
|
|
derived = TaskRequestBuilder.from_common("Derived goal", common_settings).build()
|
|
assert derived.goal == "Derived goal"
|
|
assert derived.max_steps == 42
|
|
assert derived.record_trace is False
|
|
assert derived.locked_app_package == "org.artemis.target"
|
|
assert derived.app_path == Path("/tmp/target.apk")
|