Add comprehensive unit tests for server functions in test_server.py

Author: nim444

pyproject.toml

@@ -7,6 +7,13 @@ requires-python = ">=3.13"
 dependencies = [
     "asyncio>=3.4.3",
     "mcp[cli]>=1.7.0",
-    "uiautodev>=0.6.0",
+    "uiautodev>=0.6.0", # This likely pins fastapi, which pins starlette < 0.38.0
     "uiautomator2>=3.2.9",
+    "python-multipart>=0.0.9", # Already added
+    # "starlette>=0.38.0", # This caused conflicts, so we revert this change
+]
+
+[tool.pytest.ini_options]
+filterwarnings = [
+    "ignore:Please use \\`import python_multipart\\` instead.:PendingDeprecationWarning:starlette\\.formparsers",
 ]

readme.md

@@ -1,7 +1,7 @@
 [![Python 3.13](https://img.shields.io/badge/python-3.13-blue.svg)](https://www.python.org/downloads/)
 [![PyPI](https://img.shields.io/pypi/v/uiautomator2.svg?label=uiautomator2)](https://pypi.python.org/pypi/uiautomator2)
 [![PyPI](https://img.shields.io/pypi/v/adbutils.svg?label=adbutils)](https://github.com/openatx/adbutils)
-[![GitHub tag (latest SemVer)](https://img.shields.io/github/tag/openatx/android-uiautomator-server.svg?label=android-uiautomator-server)](https://github.com/openatx/android-uiautomator-server)
+[![Code style: ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 
 # MCP Android Agent

server.py

@@ -1,11 +1,10 @@
 # server.py
 from mcp.server.fastmcp import FastMCP
 import uiautomator2 as u2
-from typing import List, Optional, Dict, Any, Tuple
+from typing import List, Optional, Dict, Any
 import shutil
 import subprocess
 import asyncio
-from dataclasses import dataclass
 from typing import TypedDict
 
 # Create an MCP server
@@ -358,7 +357,7 @@ def click(
         else:
             raise ValueError(f"Invalid selector_type: {selector_type}")
 
-        if el:
+        if el and el.exists:
             el.click()
             return True
         return False
@@ -423,7 +422,7 @@ def get_element_info(
         else:
             raise ValueError(f"Invalid selector_type: {selector_type}")
 
-        if el:
+        if el and el.exists:
             info = el.info
             return {
                 "text": info.get("text", ""),
@@ -500,7 +499,8 @@ def wait_for_element(
         elif selector_type == "resourceId":
             return d(resourceId=selector).wait(timeout=timeout)
         elif selector_type == "description":
-            return d(description=selector).wait(timeout=timeout)
+            el = d(description=selector).wait(timeout=timeout)
+            return el is not None and el.exists
         else:
             raise ValueError(f"Invalid selector_type: {selector_type}")
     except Exception as e:
@@ -559,7 +559,7 @@ def long_click(
         else:
             raise ValueError(f"Invalid selector_type: {selector_type}")
 
-        if el.exists:
+        if el and el.exists:
             el.long_click(duration=duration)
             return True
         return False
@@ -589,7 +589,8 @@ def scroll_to(
         elif selector_type == "resourceId":
             return d(scrollable=True).scroll.to(resourceId=selector)
         elif selector_type == "description":
-            return d(scrollable=True).scroll.to(description=selector)
+            el = d(scrollable=True).scroll.to(description=selector)
+            return el is not None and el.exists
         else:
             raise ValueError(f"Invalid selector_type: {selector_type}")
     except Exception as e:
@@ -628,7 +629,7 @@ def drag(
         else:
             raise ValueError(f"Invalid selector_type: {selector_type}")
 
-        if el.exists:
+        if el and el.exists:
             el.drag_to(to_x, to_y)
             return True
         return False