77 lines
2.5 KiB
Python
77 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 unittest.mock import AsyncMock
|
|
|
|
from fastapi import HTTPException
|
|
import pytest
|
|
|
|
from apps.admin_console.routers import tasks
|
|
from apps.admin_console.schemas.task_schema import RunRequest
|
|
from artemis.core.diagnostics.schema import (
|
|
ProbeCategory,
|
|
ProbeResult,
|
|
ProbeStatus,
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_task_rejects_locked_device(monkeypatch):
|
|
locked_probe = ProbeResult(
|
|
id="android_adb",
|
|
category=ProbeCategory.DEVICE,
|
|
title="Device / Emulator Connected",
|
|
status=ProbeStatus.WARN,
|
|
is_blocker=True,
|
|
summary="Device Locked",
|
|
description="Unlock the device.",
|
|
)
|
|
run_probe = AsyncMock(return_value=locked_probe)
|
|
enqueue_tasks = AsyncMock()
|
|
monkeypatch.setattr(
|
|
tasks.readiness_engine, "run_device_submission_probe", run_probe
|
|
)
|
|
monkeypatch.setattr(tasks.task_queue_service, "enqueue_tasks", enqueue_tasks)
|
|
|
|
with pytest.raises(HTTPException) as exc_info:
|
|
await tasks.run_task(RunRequest(goal="Open Settings"))
|
|
|
|
assert exc_info.value.status_code == 409
|
|
assert "locked" in exc_info.value.detail.lower()
|
|
enqueue_tasks.assert_not_awaited()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_task_enqueues_when_device_is_unlocked(monkeypatch):
|
|
unlocked_probe = ProbeResult(
|
|
id="android_adb",
|
|
category=ProbeCategory.DEVICE,
|
|
title="Device / Emulator Connected",
|
|
status=ProbeStatus.PASS,
|
|
is_blocker=True,
|
|
summary="Connected",
|
|
description="Ready.",
|
|
)
|
|
run_probe = AsyncMock(return_value=unlocked_probe)
|
|
enqueue_tasks = AsyncMock(return_value={"status": "started", "tasks": []})
|
|
monkeypatch.setattr(
|
|
tasks.readiness_engine, "run_device_submission_probe", run_probe
|
|
)
|
|
monkeypatch.setattr(tasks.task_queue_service, "enqueue_tasks", enqueue_tasks)
|
|
|
|
result = await tasks.run_task(RunRequest(goal="Open Settings"))
|
|
|
|
assert result["status"] == "started"
|
|
enqueue_tasks.assert_awaited_once()
|