【chatGPT教我写代码】pytest+appium实现安卓UI自动化

import pytest
from appium import webdriver

class TestLogin:
    def setup_method(self):
        # 设置 appium 驱动
        desired_caps = {
            "platformName": "Android",
            "deviceName": "Android Emulator",
            "appPackage": "com.example.app",
            "appActivity": "MainActivity"
        }
        self.driver = webdriver.Remote("http://localhost:4723/wd/hub", desired_caps)

    def teardown_method(self):
        # 关闭 appium 驱动
        self.driver.quit()

    def test_login_success(self):
        # 测试登录成功场景
        username = "testuser"
        password = "testpassword"
        self.driver.find_element_by_id("username_input").send_keys(username)
        self.driver.find_element_by_id("password_input").send_keys(password)
        self.driver.find_element_by_id("login_button").click()
        assert self.driver.find_element_by_id("welcome_text").text == f"Welcome, {username}!"

    def test_login_failure(self):
        # 测试登录失败场景
        username = "testuser"
        password = "wrongpassword"
        self.driver.find_element_by_id("username_input").send_keys(username)
        self.driver.find_element_by_id("password_input").send_keys(password)
        self.driver.find_element_by_id("login_button").click()
        assert self.driver.find_element_by_id("error_message").text == "Invalid username or password"