DroidMind

# pylint: disable=duplicate-code """Tests for UI automation tools.""" from unittest.mock import AsyncMock, MagicMock, patch import pytest from droidmind.devices import Device from droidmind.tools.ui import UIAction, android_ui @pytest.fixture def mock_context(): """Create a mock Context object.""" ctx = MagicMock() ctx.info = AsyncMock() ctx.error = AsyncMock() return ctx @pytest.fixture def mock_device(): """Create a mock Device object.""" device = AsyncMock(spec=Device) device.serial = "test_serial" device.tap = AsyncMock(return_value="tap_success") device.swipe = AsyncMock(return_value="swipe_success") device.input_text = AsyncMock(return_value="input_text_success") device.press_key = AsyncMock(return_value="press_key_success") device.start_activity = AsyncMock(return_value="start_activity_success") return device @pytest.fixture def mock_device_manager(mock_device): """Create a mock DeviceManager that returns the mock device.""" with patch("droidmind.tools.ui.get_device_manager") as mock_get_manager: manager = MagicMock() manager.get_device = AsyncMock(return_value=mock_device) mock_get_manager.return_value = manager yield manager async def test_tap(mock_context, mock_device, mock_device_manager): """Test the tap action of the android_ui tool.""" result = await android_ui(ctx=mock_context, serial="test_serial", action=UIAction.TAP, x=100, y=200) # Assert context methods were called mock_context.info.assert_called() # Assert device methods were called with correct params mock_device.tap.assert_called_once_with(100, 200) # Assert result is correct assert "Successfully" in result async def test_swipe(mock_context, mock_device, mock_device_manager): """Test the swipe action of the android_ui tool.""" result = await android_ui( ctx=mock_context, serial="test_serial", action=UIAction.SWIPE, start_x=100, start_y=200, end_x=300, end_y=400, duration_ms=500, ) # Assert context methods were called mock_context.info.assert_called() # Assert device methods were called with correct params mock_device.swipe.assert_called_once_with(100, 200, 300, 400, 500) # Assert result is correct assert "Successfully" in result async def test_input_text(mock_context, mock_device, mock_device_manager): """Test the input_text action of the android_ui tool.""" result = await android_ui(ctx=mock_context, serial="test_serial", action=UIAction.INPUT_TEXT, text="Hello World") # Assert context methods were called mock_context.info.assert_called() # Assert device methods were called with correct params mock_device.input_text.assert_called_once_with("Hello World") # Assert result is correct assert "Successfully" in result async def test_press_key(mock_context, mock_device, mock_device_manager): """Test the press_key action of the android_ui tool.""" result = await android_ui(ctx=mock_context, serial="test_serial", action=UIAction.PRESS_KEY, keycode=4) # BACK key # Assert context methods were called mock_context.info.assert_called() # Assert device methods were called with correct params mock_device.press_key.assert_called_once_with(4) # Assert result is correct assert "Successfully" in result async def test_start_intent(mock_context, mock_device, mock_device_manager): """Test the start_intent action of the android_ui tool.""" result = await android_ui( ctx=mock_context, serial="test_serial", action=UIAction.START_INTENT, package="com.android.settings", activity=".Settings", extras={"extra_key": "extra_value"}, ) # Assert context methods were called mock_context.info.assert_called() # Assert device methods were called with correct params mock_device.start_activity.assert_called_once_with( "com.android.settings", ".Settings", {"extra_key": "extra_value"} ) # Assert result is correct assert "Successfully" in result async def test_device_not_found(mock_context, mock_device_manager): """Test error handling when device is not found for an android_ui action (e.g., tap).""" # Simulate device not found mock_device_manager.get_device = AsyncMock(return_value=None) result = await android_ui(ctx=mock_context, serial="nonexistent_serial", action=UIAction.TAP, x=100, y=200) # Assert error was logged mock_context.error.assert_called_once() # Assert error message is returned assert "Error" in result assert "not connected" in result