Handling uiautomator crash, restarting appium driver. This should handle when the emulator is relaunched.

Author: samuelclay

ansible/roles/android_root/tasks/main.yml

@@ -82,7 +82,7 @@
 
 - name: Download rootAVD script
   get_url:
-    url: https://raw.githubusercontent.com/newbit1/rootAVD/master/rootAVD.sh
+    url: https://gitlab.com/newbit/rootAVD/-/raw/master/rootAVD.sh
     dest: "{{ root_tools_dir }}/rootAVD.sh"
     mode: "0755"
   become: yes

ansible/roles/android_x86/tasks/main.yml

@@ -95,8 +95,8 @@
 - name: Create an x86 AVD in /opt/android-sdk/avd
   shell: |
     {{ android_home }}/cmdline-tools/latest/bin/avdmanager create avd -n {{ avd_name }} -k "{{ sys_img }}" --device "{{ device_skin }}" --force
-  args:
-    creates: "{{ android_home }}/avd/{{ avd_name }}.avd"
+  # args:
+  #   creates: "{{ android_home }}/avd/{{ avd_name }}.avd"
   become: yes
   environment:
     ANDROID_SDK_ROOT: "{{ android_home }}"
@@ -147,7 +147,7 @@
       [Service]
       User={{ ansible_user }}
       WorkingDirectory={{ android_home }}
-      ExecStart={{ android_home }}/emulator/emulator -avd {{ avd_name }} -no-window -no-audio -no-boot-anim -no-metrics -gpu swiftshader_indirect -accel off -no-snapshot -no-snapshot-load -no-snapshot-save
+      ExecStart={{ android_home }}/emulator/emulator -avd {{ avd_name }} -no-window -no-audio -no-boot-anim -no-metrics -gpu swiftshader_indirect -accel off -no-snapshot -no-snapshot-load -no-snapshot-save -writable-system -selinux disabled -qemu -enable-kvm
       Restart=always
       StandardOutput=journal
       StandardError=journal

server/server.py

@@ -9,6 +9,7 @@
 
 from flask import Flask, make_response, redirect, request, send_file
 from flask_restful import Api, Resource
+from selenium.common import exceptions as selenium_exceptions
 
 from automator import KindleAutomator
 from handlers.test_fixtures_handler import TestFixturesHandler
@@ -123,21 +124,48 @@ def post(self):
             return {"error": str(e)}, 500
 
 
+def handle_uiautomator_crash(f):
+    """Decorator to handle UiAutomator2 server crashes by restarting the driver."""
+    
+    def wrapper(*args, **kwargs):
+        try:
+            return f(*args, **kwargs)
+        except Exception as e:
+            # Check if it's the UiAutomator2 server crash error
+            if isinstance(e, selenium_exceptions.WebDriverException) and "cannot be proxied to UiAutomator2 server because the instrumentation process is not running" in str(e):
+                logger.warning("UiAutomator2 server crashed. Attempting to restart driver...")
+                if server.automator and server.automator.ensure_driver_running():
+                    # Retry the operation after driver restart
+                    logger.info("Driver restarted successfully. Retrying operation...")
+                    return f(*args, **kwargs)
+                else:
+                    logger.error("Failed to restart driver after UiAutomator2 server crash")
+                    return {"error": "Failed to restart driver after UiAutomator2 server crash"}, 500
+            # Re-raise other exceptions
+            raise
+    
+    return wrapper
+
+
 def ensure_automator_healthy(f):
     """Decorator to ensure automator is initialized and healthy before each operation."""
 
-    def wrapper(*args, **kwargs):
-        try:
-            if not server.automator:
-                logger.info("No automator found, initializing...")
-                server.initialize_automator()
-                if not server.automator.initialize_driver():
-                    return {"error": "Failed to initialize driver"}, 500
+    @handle_uiautomator_crash
+    def wrapped_function(*args, **kwargs):
+        if not server.automator:
+            logger.info("No automator found, initializing...")
+            server.initialize_automator()
+            if not server.automator.initialize_driver():
+                return {"error": "Failed to initialize driver"}, 500
 
-            if not server.automator.ensure_driver_running():
-                return {"error": "Failed to ensure driver is running"}, 500
+        if not server.automator.ensure_driver_running():
+            return {"error": "Failed to ensure driver is running"}, 500
 
-            return f(*args, **kwargs)
+        return f(*args, **kwargs)
+    
+    def wrapper(*args, **kwargs):
+        try:
+            return wrapped_function(*args, **kwargs)
         except Exception as e:
             logger.error(f"Error in automator health check: {e}")
             logger.error(f"Traceback: {traceback.format_exc()}")