专栏文章 Airtest 新手升级：一个相对完整的纯.py 脚本是怎样子的

fishfish-yu · 2022年11月17日 · 最后由 loneyao 回复于 2022年11月18日 · 6664 次阅读

此文章来源于项目官方公众号：“AirtestProject”
版权声明：允许转载，但转载必须保留原链接；请勿用作商业或者非法用途

1. 前言

一直以来，Airtest 的教程都倾向于编写 .air 脚本，但本质上，它还是 python 脚本来的。今天我们就来补上这个缺口，一起来看下一个相对完整的纯 .py 脚本是什么样子的。

2. 例子一：纯`py`的 Airtest 脚本

有时候，我们只是想借助 Airtest 连接设备的功能，然后实现一些小任务，比如简单的装包任务等，而不是截一堆图片做自动化测试。

这种情况下，我们只需要简单编写一个纯 py 脚本即可：

# -*- encoding=utf8 -*-
__author__ = "AirtestProject"

from airtest.core.api import *
from airtest.cli.parser import cli_setup

def get_app_screen_shot(apk_file, package_name):
    """ 
        Connect an android device, automatically complete the app installation, open, screenshot, close, uninstall.
        Finally you will get a screenshot from app in logdir.

        Args:
            apk_file: apk file route in your windows computer.
            package_name: package_name of your app.

        Returns: None
    """

    # 连接设备以及初始化工作 log文件夹默认生成在当面目录下 
    auto_setup(__file__, logdir=True, devices=["Android:///",],)
    # 唤醒手机屏幕
    wake()
    # 通过windows上apk包体下载app到手机上
    install(apk_file)
    # 启动下载好的app
    start_app(package_name)
    # 等待5s app完全启动
    sleep(5)
    # 截图 并保存到logdir路径下
    snapshot("screen_shot.png")
    # 关闭app
    stop_app(package_name)
    # 卸载app
    uninstall(package_name)

if __name__ == '__main__':
    # 设置apk文件路径以及对应包体名称
    APK_FILE = r"C:\Users\AirtestProject\Desktop\NeteaseCloudMusic_Music_official_8.6.50.211229222733_32.apk"
    PACKAGE_NAME = "com.netease.cloudmusic"
    try:
        get_app_screen_shot(APK_FILE, PACKAGE_NAME)
    except Exception as err:
        # 记录报错到logdir下的log.txt文件中，并截图保存发生错误时的手机屏幕
        log(err, desc="Finale Error", snapshot=True)

1）纯`py`脚本的设备连接

对于纯py脚本，我们需要在脚本里面，明确写明连接设备的脚本 。而.air脚本，如果有使用auto_setup(__file__)，则会自动帮我们连接第一台 Android 设备。这也是刚从.air脚本刚刚切换成纯py脚本的同学，经常容易忽略的一点。

所以上述示例中，我们在auto_setup接口里面明确连接了 Android 设备，当然，Airtest 连接设备，还有很多种方式，感兴趣的同学可以参考我们官方教程文档中，Airtest 连接设备脚本的介绍章节： https://airtest.doc.io.netease.com/IDEdocs/airtest_framework/2_connect_device/ 。

2）纯`py`脚本的 log 保存

类似的还有纯py脚本的 log 保存，也需要我们在脚本开头时，明确写明。是否保存 log，取决于同学们最终想不想生成报告。因为报告的生成，是依赖于脚本运行 log 的。

如果我们不保存 log，则最后我们也就没有办法依赖于 log 来生成测试报告。

3）airtest 的核心 api

上述简单的装包测试任务中，涉及到一些 Airtest 核心 API，比如：

唤醒屏幕：wake()
安装应用：install()
启动应用：start_app()
截图：snapshot()
卸载应用：uninstall()
......

我们可以从 Airtest 的核心 API 文档中，来熟悉这部分的内容： https://airtest.readthedocs.io/zh_CN/latest/all_module/airtest.core.api.html 。

3. 例子二：纯 py 的 Poco 脚本

说完了纯py的 Airtest 脚本，当然少不了我们纯py的 Poco 脚本啦。下述示例中，包含了连接设备、初始化 Android poco、简单的 poco 操作、记录报错到 log 里，最后生成测试报告。

# -*- encoding: utf-8 -*-

from airtest.core.api import *
from airtest.report.report import simple_report
from poco.drivers.android.uiautomation import AndroidUiautomationPoco

def demo():
    # 连接设备以及初始化工作 log文件夹默认生成在当面目录下 
    auto_setup(__file__, logdir=True, devices=["Android:///",],)
    # 初始化poco
    poco = AndroidUiautomationPoco(use_airtest_input=True, screenshot_each_action=False)
    # 执行一些操作 确保手机app已登录并跳转到app初始化界面
    try:
        # 以下操作执行：我的-》我喜欢的音乐-》点击播放
        poco(text="我的").click()
        poco(text="我喜欢的音乐").click()
        # 灵活结合ide提供的poco ui树获取播放按钮name属性值 
        poco(name="com.netease.cloudmusic:id/minPlayBtn").click()
    except Exception as err:
        # 记录报错到logdir下的log.txt文件中，并截图保存发生错误时的手机屏幕
        log(err, desc="Finale Error", snapshot=True)
    # 无论运行是否成功 输出报告
    finally:
        # 默认生成html报告到脚本所在的同级目录下
        simple_report(__file__, logpath=True)

if __name__ == '__main__':
    demo()