""" Mumu模拟器 pytest 测试框架 """ import pytest import pyautogui import pygetwindow as gw import time import os import subprocess import requests from pathlib import Path # 配置 MUMU_EXE_PATH = r"C:\Program Files\Netease\MuMu\nx_main\MuMuNxMain.exe" MUMU_PROCESS_NAME = "MuMuNxMain.exe" ADB_PATH = r"C:\Program Files\Netease\MuMu\shell\adb.exe" APK_DOWNLOAD_URL = "http://your-server.com/app-release.apk" LOCAL_APK_PATH = os.path.join(os.path.dirname(__file__), "test_app.apk") INSTALLED_PACKAGE_NAME = "com.example.flomo_ai" WEB_URL = "http://192.168.3.15/" SCRIPT_DIR = os.path.dirname(__file__) class MumuEmulator: """Mumu模拟器控制类""" def __init__(self): self.process_name = MUMU_PROCESS_NAME self.script_dir = SCRIPT_DIR pyautogui.PAUSE = 1 pyautogui.FAILSAFE = True def is_running(self) -> bool: """检查模拟器是否运行""" try: result = subprocess.run( ["tasklist"], capture_output=True, text=True ) return self.process_name in result.stdout except Exception: return False def start(self): """启动模拟器""" if not self.is_running(): subprocess.Popen(MUMU_EXE_PATH) time.sleep(5) self.bring_to_front() def bring_to_front(self): """将模拟器窗口置顶""" try: windows = gw.getWindowsWithTitle("MuMu") for window in windows: if window.title: window.activate() window.restore() time.sleep(0.5) print(f"已将窗口置顶: {window.title}") return True print("未找到MuMu模拟器窗口") return False except Exception as e: print(f"置顶窗口失败: {e}") return False def find_and_click(self, image_name, confidence=0.8, timeout=30): """查找图片并点击""" image_path = os.path.join(self.script_dir, f"{image_name}.png") if not os.path.exists(image_path): print(f"图片不存在: {image_path}") return False start_time = time.time() while time.time() - start_time < timeout: try: location = pyautogui.locateOnScreen(image_path, confidence=confidence) if location: center = pyautogui.center(location) pyautogui.click(center) print(f"点击了: {image_name}") return True except Exception as e: pass time.sleep(1) print(f"未找到图片: {image_name}") return False def type_text(self, text): """输入文本""" pyautogui.write(text, interval=0.1) print(f"输入了文本: {text}") def press_enter(self): """按回车""" pyautogui.press("enter") print("按下回车") def press_home(self): """按 Home 键返回桌面""" pyautogui.press("home") print("按下 Home 键") def right_click(self): """右键点击""" pyautogui.rightClick() print("右键点击") def wait_for_image(self, image_name, confidence=0.8, timeout=30): """等待图片出现并返回位置""" image_path = os.path.join(self.script_dir, f"{image_name}.png") if not os.path.exists(image_path): print(f"图片不存在: {image_path}") return None start_time = time.time() while time.time() - start_time < timeout: try: location = pyautogui.locateOnScreen(image_path, confidence=confidence) if location: print(f"找到图片: {image_name}") return location except Exception: pass time.sleep(1) print(f"未找到图片: {image_name}") return None def wait_for_boot(self, timeout=120): """等待模拟器启动完成""" boot_completed = False start_time = time.time() while time.time() - start_time < timeout: if self.is_running(): time.sleep(5) boot_completed = True break time.sleep(2) return boot_completed def check_image_exists(self, image_name, confidence=0.8) -> bool: """检查图片是否存在""" image_path = os.path.join(self.script_dir, f"{image_name}.png") if not os.path.exists(image_path): return False try: location = pyautogui.locateOnScreen(image_path, confidence=confidence) return location is not None except Exception: return False def run_sequence(self): """按顺序执行任务""" self.bring_to_front() time.sleep(5) sequence = [ "web", "web_address", "web_goon", "web_debug", "web_debug_apk", "web_debug_apk_download", "web_debug_apk_open", "web_debug_apk_install", "web_debug_apk_run" ] for image_name in sequence: if image_name == "web_address": for _ in range(10): if self.check_image_exists("web_address"): time.sleep(1) self.type_text(WEB_URL) time.sleep(0.5) self.press_enter() break time.sleep(1) else: print("web_address 图片不存在，跳过输入URL") else: self.find_and_click(image_name, timeout=10) time.sleep(2) def close(self): """关闭模拟器""" try: subprocess.run(["taskkill", "/F", "/IM", self.process_name]) except Exception: pass def uninstall_app_ui(self): """通过 UI 卸载 app：按 Home 键 -> 找到 flomo -> 右键 -> Uninstall -> Uninstall Confirm""" self.bring_to_front() time.sleep(2) self.press_home() time.sleep(3) location = self.wait_for_image("flomo", timeout=30) if not location: print("未找到 flomo 应用图标") return False center = pyautogui.center(location) pyautogui.rightClick(center) time.sleep(1) location = self.wait_for_image("uninstall", timeout=10) if not location: print("未找到 uninstall 选项") return False center = pyautogui.center(location) pyautogui.click(center) time.sleep(1) location = self.wait_for_image("uninstall-confirm", timeout=10) if not location: print("未找到卸载确认按钮") return False center = pyautogui.center(location) pyautogui.click(center) time.sleep(2) print("卸载完成") return True class AdbHelper: """ADB命令辅助类""" def __init__(self): self.adb_path = ADB_PATH def exec_cmd(self, *args): """执行ADB命令""" cmd = [self.adb_path] + list(args) result = subprocess.run(cmd, capture_output=True, text=True) return result.stdout, result.stderr def install_apk(self, apk_path: str) -> bool: """安装APK""" stdout, stderr = self.exec_cmd("install", "-r", apk_path) return "Success" in stdout def uninstall_app(self, package_name: str) -> bool: """卸载应用""" stdout, stderr = self.exec_cmd("uninstall", package_name) return "Success" in stdout def launch_app(self, package_name: str, activity: str): """启动应用""" self.exec_cmd("shell", "am", "start", "-n", f"{package_name}/{activity}") def is_app_installed(self, package_name: str) -> bool: """检查应用是否已安装""" stdout, _ = self.exec_cmd("shell", "pm", "list", "packages", package_name) return package_name in stdout def pull_file(self, remote_path: str, local_path: str): """从模拟器拉取文件""" self.exec_cmd("pull", remote_path, local_path) @pytest.fixture(scope="session") def emulator(): """模拟器fixture""" mumu = MumuEmulator() yield mumu # 测试结束后清理 if mumu.is_running(): mumu.close() @pytest.fixture(scope="session") def adb(): """ADB fixture""" return AdbHelper() @pytest.fixture(scope="session") def download_apk(): """下载APK""" if os.path.exists(LOCAL_APK_PATH): return LOCAL_APK_PATH response = requests.get(APK_DOWNLOAD_URL, stream=True) with open(LOCAL_APK_PATH, "wb") as f: for chunk in response.iter_content(chunk_size=8192): f.write(chunk) yield LOCAL_APK_PATH # 清理 if os.path.exists(LOCAL_APK_PATH): os.remove(LOCAL_APK_PATH)