Airtest启动器——添加自定义命令行参数、自定义变量

曾经终败给现在 2023-09-24 14:57 157阅读 0赞

### 前言 ###

我们都知道，在命令行运行Airtest脚本时， airtest run test.air 指令后面附带的参数是固定的：

*  –device ，用来指定连接的被测设备
 *  –log ，用来指定log内容和截图存放的目录
 *  –recording ，运行脚本时进行录屏操作

如果想要在运行脚本的命令中，添加一些自定义的参数，我们就要使用到Airtest的启动器啦。

### Airtest启动器的介绍 ###

启动器实际上也是一个 .py 文件，里面所编写的内容，实际上是基于1个叫 AirtestCase 的类，这个类继承了 unittest.TestCase ，其目的在于Airtest在运行用例脚本时，添加所有执行基础Airtest脚本的相关功能。

因此，假如我们需要添加自定的义功能，就只需要在 AirtestCase 类的基础上，往 setup() 和 teardown() 方法中加入自己的代码即可。如果这些设置和功能内容相对固定，可以将这些内容作为一个启动器（launcher），用来在运行实际测试用例之前初始化相关的自定义环境。

from airtest.cli.runner import AirtestCase, run_script
    from airtest.cli.parser import runner_parser
    
    class CustomAirtestCase(AirtestCase):
    
        def setUp(self):
            pass
            super(CustomAirtestCase, self).setUp()
    
        def tearDown(self):
            pass
            super(CustomAirtestCase, self).tearDown()
    
    if __name__ == '__main__':
        ap = runner_parser()
        args = ap.parse_args()
        run_script(args, CustomAirtestCase)

### 在启动器中添加自定义的命令行参数 ###

了解了启动器的基本含义之后，我们再来看下，如何利用启动器来帮助我们给 airtest run 指令添加自定义的命令行参数。

假设我们需要往命令行传入1个参数，用于指定这个脚本需要运行几次，参数名称假设为 --retry，那么我们可以这样在启动器的脚本中添加如下内容：

from airtest.cli.runner import AirtestCase, run_script
    from airtest.cli.parser import runner_parser
    
    
    class CustomAirtestCase(AirtestCase):
        def setUp(self):
            # 在air脚本运行之前获取这个自定义的参数
            if self.args.retry:
                self.scope['retry']=self.args.retry
    
    if __name__ == '__main__':
        ap = runner_parser()
        # 添加自定义的命令行参数
        ap.add_argument(
            "--retry", help="The number of times the script will be run")
        args = ap.parse_args()
        run_script(args, CustomAirtestCase)

有了启动器之后，我们的 .air 脚本就可以接收到这个命令行参数并对它进行相应的处理了：

# -*- encoding=utf8 -*-
    __author__ = "AirtestProject"
    
    from airtest.core.api import *
    auto_setup(__file__)
    
    print("脚本需要运行的次数是："+retry)
    print('参数个数为:'+str(len(sys.argv))+'个参数')
    print('参数列表:'+str(sys.argv))

### 带启动器的运行air脚本的命令 ###

已经准备好启动器脚本和我们的 .air 脚本之后，就可以在命令行使用如下命令，使用我们定义好的启动器来运行我们的 air 脚本啦：

python launcher.py test.air --device Android:///serial_num --log log_path --retry 3

![图片][ed28ef78c6bd4b5f8b64f9a2c13a48e2.png]

### 添加自定义变量 ###

**1.在启动器中新增变量**

在 launcher.py 的脚本中，新建1个自定义的变量 hunter ：

from airtest.cli.runner import AirtestCase, run_script
    from airtest.cli.parser import runner_parser
    
    class CustomAirtestCase(AirtestCase):
        def setUp(self):
            self.scope["hunter"] = "i am hunter"
    
    if __name__ == '__main__':
        ap = runner_parser()
        args = ap.parse_args()
        run_script(args, CustomAirtestCase)

**2.在 .air 脚本中直接使用启动器的变量**

在启动器定义好后续的 .air 脚本都要使用到的变量之后，我们就可以用该启动器来运行Airtest脚本。并且在 .air 脚本中，我们能够直接使用这些变量，无需重新定义：

# -*- encoding=utf8 -*-
    __author__ = "AirtestProject"
    
    from airtest.core.api import *
    auto_setup(__file__)
    
    print("启动器定义的变量："+hunter)

**3.在命令行中用启动器运行 .air 脚本**

![在这里插入图片描述][73f3761df17d4d1eb341c55a4c352f03.png]

**4.在AirestIDE中更换启动器运行Airtest脚本**

其实在AirtestIDE中，有一个默认的启动器脚本：

![在这里插入图片描述][c27b8de78fcf422e8beb5ada87d16b5d.png]

该启动器的作用，包含但不限于 把当前运行的代码行标出颜色 等等，我们也可以将这个默认的启动器更换成我们自定义的启动器脚本：

![在这里插入图片描述][eaa13d9d9511498da9b9a34c65f053c7.png]

这样直接在IDE点击运行脚本的按钮，就会直接使用我们自定义的启动器来运行当前的 .air 脚本：

![在这里插入图片描述][ab453d92aebb46f586d95216f151794f.png]

**添加自定义的环境变量**

**1.在启动器中添加环境变量**

当然，除了在启动器的 setUp() 方法中，自定义变量给Airtest脚本用之外，我们还可以在启动器中以添加环境变量的方式，传变量给 .air 脚本：

from airtest.cli.runner import AirtestCase, run_script
    from airtest.cli.parser import runner_parser
    
    import os
    os.environ['HOST'] = "192.168.0.1"
    
    class CustomAirtestCase(AirtestCase):
        def setUp(self):
            self.scope["hunter"] = "i am hunter"
    
    if __name__ == '__main__':
        ap = runner_parser()
        args = ap.parse_args()
        run_script(args, CustomAirtestCase)

**2.在Airtest脚本中使用启动器的环境变量**

# -*- encoding=utf8 -*-
    __author__ = "AirtestProject"
    
    from airtest.core.api import *
    import os
    auto_setup(__file__)
    
    print("启动器定义的变量："+hunter)
    print("启动器的环境变量："+os.environ.get('HOST'))

**3.在IDE中使用该启动器运行Airtest脚本**

运行Airtest脚本之前，已将IDE的 选项–设置–自定义Launcher文件路径 设置为自定义的 launcher.py 的路径：

![在这里插入图片描述][bed73f9c94db41acae8e908675c39737.png]

--------------------

### 资源分享 ###

**下方这份完整的软件测试视频学习教程已经上\#\# 标题传CSDN官方认证的二维码，朋友们如果需要可以自行免费领取 `【保证100%免费】`**

![在这里插入图片描述][b9a9eea1ed5c42429c5e383b00ba6e0c.png_pic_center]

![在这里插入图片描述][d96b58a511f84668b59f3f620964da07.png_pic_center]

[ed28ef78c6bd4b5f8b64f9a2c13a48e2.png]: https://img-blog.csdnimg.cn/ed28ef78c6bd4b5f8b64f9a2c13a48e2.png
[73f3761df17d4d1eb341c55a4c352f03.png]: https://img-blog.csdnimg.cn/73f3761df17d4d1eb341c55a4c352f03.png
[c27b8de78fcf422e8beb5ada87d16b5d.png]: https://img-blog.csdnimg.cn/c27b8de78fcf422e8beb5ada87d16b5d.png
[eaa13d9d9511498da9b9a34c65f053c7.png]: https://img-blog.csdnimg.cn/eaa13d9d9511498da9b9a34c65f053c7.png
[ab453d92aebb46f586d95216f151794f.png]: https://img-blog.csdnimg.cn/ab453d92aebb46f586d95216f151794f.png
[bed73f9c94db41acae8e908675c39737.png]: https://img-blog.csdnimg.cn/bed73f9c94db41acae8e908675c39737.png
[b9a9eea1ed5c42429c5e383b00ba6e0c.png_pic_center]: https://img-blog.csdnimg.cn/b9a9eea1ed5c42429c5e383b00ba6e0c.png#pic_center
[d96b58a511f84668b59f3f620964da07.png_pic_center]: https://img-blog.csdnimg.cn/d96b58a511f84668b59f3f620964da07.png#pic_center