138 lines
5.1 KiB
Python
138 lines
5.1 KiB
Python
from types import SimpleNamespace
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from artemis.drivers.android.adb_driver import AndroidAdbDriver
|
|
from artemis.drivers.errors import AppLaunchError
|
|
|
|
|
|
PACKAGE = "com.example.app"
|
|
COMPONENT = f"{PACKAGE}/.MainActivity"
|
|
|
|
|
|
def response(output, code=0):
|
|
return SimpleNamespace(returncode=code, output=output)
|
|
|
|
|
|
@pytest.fixture
|
|
def launch_device():
|
|
device = MagicMock()
|
|
# An emulator without monkey must work, without installing or invoking it.
|
|
device.shell.side_effect = AssertionError("Unexpected unchecked shell/monkey call")
|
|
driver = AndroidAdbDriver("test-device", MagicMock())
|
|
driver._device = device
|
|
return driver, device
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"output",
|
|
[
|
|
"Starting: Intent {}\nStatus: ok\nComplete",
|
|
"Starting: Intent {}\nWarning: Activity not started, its current task has been brought to the front",
|
|
"Warning: Activity not started, intent has been delivered to currently running top-most instance.",
|
|
],
|
|
)
|
|
async def test_launch_without_monkey_checks_start_result(launch_device, output):
|
|
driver, device = launch_device
|
|
device.shell2.side_effect = [response(f"priority=0\n{COMPONENT}"), response(output)]
|
|
assert await driver.launch_app(PACKAGE) is True
|
|
calls = device.shell2.call_args_list
|
|
assert calls[0].args[0][:3] == ["cmd", "package", "resolve-activity"]
|
|
assert calls[1].args[0][-2:] == ["-n", COMPONENT]
|
|
assert calls[1].args[0][:5] == ["am", "start", "-W", "--user", "current"]
|
|
assert "0x10200000" in calls[1].args[0]
|
|
assert all(call.kwargs["timeout"] > 0 for call in calls)
|
|
device.shell.assert_not_called()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_falls_back_to_pm_when_cmd_is_missing(launch_device):
|
|
driver, device = launch_device
|
|
device.shell2.side_effect = [
|
|
response("/system/bin/sh: cmd: inaccessible or not found", 127),
|
|
response(COMPONENT),
|
|
response("Status: ok"),
|
|
]
|
|
assert await driver.launch_app(PACKAGE) is True
|
|
assert device.shell2.call_args_list[1].args[0][:2] == ["pm", "resolve-activity"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_old_resolvers_use_package_scoped_intent(launch_device):
|
|
driver, device = launch_device
|
|
device.shell2.side_effect = [
|
|
response("Unknown command: resolve-activity", 1),
|
|
response("Unknown command: resolve-activity", 1),
|
|
response("Status: ok"),
|
|
]
|
|
assert await driver.launch_app(PACKAGE) is True
|
|
assert device.shell2.call_args_list[-1].args[0][-2:] == ["-p", PACKAGE]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize("output", ["No activity found", "", "android/.ResolverActivity"])
|
|
async def test_missing_or_ambiguous_launcher_does_not_start(launch_device, output):
|
|
driver, device = launch_device
|
|
device.shell2.return_value = response(output)
|
|
with pytest.raises(AppLaunchError, match="未找到唯一可启动入口") as error:
|
|
await driver.launch_app(PACKAGE)
|
|
assert error.value.retryable is False
|
|
assert device.shell2.call_count == 1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"output, code",
|
|
[
|
|
("Starting: Intent {}\nError type 3\nError: Activity class does not exist.", 0),
|
|
("java.lang.SecurityException: Permission Denial: not exported", 0),
|
|
("Error: Permission Denial", 1),
|
|
("/system/bin/sh: am: inaccessible or not found", 127),
|
|
("Starting: Intent {}\nStatus: error", 0),
|
|
],
|
|
)
|
|
async def test_start_errors_are_retained_even_when_exit_zero(launch_device, output, code):
|
|
driver, device = launch_device
|
|
device.shell2.side_effect = [response(COMPONENT), response(output, code)]
|
|
with pytest.raises(AppLaunchError) as error:
|
|
await driver.launch_app(PACKAGE)
|
|
assert output in str(error.value)
|
|
assert f"退出码 {code}" in str(error.value)
|
|
assert error.value.retryable is False
|
|
assert error.value.may_have_started is False
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize("failure", [TimeoutError(), response("Status: timeout"), response("")])
|
|
async def test_unknown_start_outcome_requires_foreground_check(launch_device, failure):
|
|
driver, device = launch_device
|
|
device.shell2.side_effect = [response(COMPONENT), failure]
|
|
with pytest.raises(AppLaunchError) as error:
|
|
await driver.launch_app(PACKAGE)
|
|
assert error.value.retryable is True
|
|
assert error.value.may_have_started is True
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_resolve_transport_failure_does_not_request_a_launch(launch_device):
|
|
driver, device = launch_device
|
|
device.shell2.side_effect = OSError("device offline")
|
|
with pytest.raises(AppLaunchError, match="device offline") as error:
|
|
await driver.launch_app(PACKAGE)
|
|
assert error.value.retryable is True
|
|
assert error.value.may_have_started is False
|
|
assert device.shell2.call_count == 1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"package", ["", "com.example.app;reboot", "com.example.app/.Activity", "a b"]
|
|
)
|
|
async def test_invalid_package_rejected_before_adb(launch_device, package):
|
|
driver, device = launch_device
|
|
with pytest.raises(AppLaunchError, match="无效"):
|
|
await driver.launch_app(package)
|
|
device.shell2.assert_not_called()
|