# 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 MagicMock, patch
from artemis.agents.hopper.hopper import HopperError, HopperOutput
from artemis.context import ArtemisContext
from artemis.tools.mobile.launch_app import find_package, launch_app
import pytest


@pytest.fixture
def mock_context():
    ctx = MagicMock(spec=ArtemisContext)
    ctx.package_cache = {}
    return ctx


@pytest.mark.asyncio
@patch("artemis.tools.mobile.launch_app.list_packages_async")
@patch("artemis.tools.mobile.launch_app.hopper")
async def test_find_package_cache_hit(mock_hopper, mock_list_packages, mock_context):
    mock_list_packages.return_value = "com.example.app1\ncom.example.app2"

    # First call: not in cache, calls hopper
    mock_hopper.return_value = HopperOutput(
        found=True, output="com.example.app1", reason="Found it"
    )

    pkg1 = await find_package(mock_context, "App 1")
    assert pkg1 == "com.example.app1"
    assert mock_hopper.call_count == 1
    assert mock_context.package_cache["App 1"] == "com.example.app1"

    # Second call: cache hit, does not call hopper or list_packages again
    mock_hopper.reset_mock()
    mock_list_packages.reset_mock()

    pkg2 = await find_package(mock_context, "App 1")
    assert pkg2 == "com.example.app1"
    mock_hopper.assert_not_called()
    mock_list_packages.assert_not_called()


@pytest.mark.asyncio
@patch("artemis.tools.mobile.launch_app.list_packages_async")
@patch("artemis.tools.mobile.launch_app.hopper")
async def test_find_package_not_found_is_not_cached(mock_hopper, mock_list_packages, mock_context):
    mock_list_packages.return_value = "com.example.app1\ncom.example.app2"

    # Hopper doesn't find it
    mock_hopper.return_value = HopperOutput(found=False, output=None, reason="Not found")

    pkg1 = await find_package(mock_context, "App 3")
    assert pkg1 is None
    assert mock_hopper.call_count == 1
    assert "App 3" not in mock_context.package_cache

    # An earlier miss must not prevent a later successful lookup.
    mock_hopper.reset_mock()
    mock_hopper.return_value = HopperOutput(found=True, output="com.example.app1", reason="Found")
    pkg2 = await find_package(mock_context, "App 3")
    assert pkg2 == "com.example.app1"
    mock_hopper.assert_called_once()


@pytest.mark.asyncio
@patch("artemis.tools.mobile.launch_app.list_packages_async")
@patch("artemis.tools.mobile.launch_app.hopper")
async def test_find_package_passes_use_fallback(mock_hopper, mock_list_packages, mock_context):
    mock_list_packages.return_value = "com.example.app1\ncom.example.app2"
    mock_hopper.return_value = HopperOutput(
        found=True, output="com.example.app1", reason="Found it"
    )

    # Pass use_fallback=False
    await find_package(mock_context, "App 1", use_fallback=False)
    mock_hopper.assert_called_once_with(
        ctx=mock_context,
        request=("I'm looking for the package name of the following app: 'App 1'"),
        data="com.example.app1\ncom.example.app2",
        use_fallback=False,
    )


@pytest.mark.asyncio
@patch("artemis.tools.mobile.launch_app.list_packages_async")
@patch("artemis.tools.mobile.launch_app.hopper")
async def test_model_failure_is_reported_without_poisoning_package_cache(
    mock_hopper, mock_list_packages, mock_context
):
    mock_list_packages.return_value = "com.example.app1"
    mock_hopper.side_effect = HopperError("模型未能完成数据提取，请查看模型请求诊断。")
    result = await launch_app.execute(ctx=mock_context, app_name="App 1")
    assert "模型未能完成数据提取" in result
    assert "Package not found" not in result
    assert "App 1" not in mock_context.package_cache
    mock_hopper.side_effect = None
    mock_hopper.return_value = HopperOutput(found=True, output="com.example.app1", reason="Found")
    assert await find_package(mock_context, "App 1") == "com.example.app1"


@pytest.mark.asyncio
@patch("artemis.tools.mobile.launch_app.list_packages_async")
@patch("artemis.tools.mobile.launch_app.hopper")
async def test_invented_package_is_never_cached(mock_hopper, mock_list_packages, mock_context):
    mock_list_packages.return_value = "com.example.app1"
    mock_hopper.return_value = HopperOutput(
        found=True, output="com.example.missing", reason="Guess"
    )
    assert await find_package(mock_context, "App 1") is None
    assert "App 1" not in mock_context.package_cache