自动化脚本如何切换环境？Pytest这些功能你必须要掌握

忘是亡心i 2024-03-31 09:17 133阅读 0赞

![在这里插入图片描述][daf5030110674c34bc43dcf35e99fa6d.jpeg_pic_center]

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

#### 文章目录 ####

*  一、前言
 *  二、安装
 *  三、使用
 *   *  第1种:使用方式是终端添加–base-url这个命令
     *  第2种:使用方式是在pytest.ini配置文件种去配置base\_url,然后自动读取url的数据，这样就不用添加–base-url这个命令行参数了：
     *  第3种:pytest有个hooks函数，可以自定义命令行参数，一般在conftest.py中去引用。
     *  第4种:通过yml文件和fixture进行配合使用

## 一、前言 ##

测试工程师每天都跟不同的环境打交道，比如线上环境，测试环境，预上线环境等等，那么作为自动化测试人员写的代码，我们也要具备能自由切换环境的能力，如何能让我们python语言写的测试用例可以自由切换到不同的环境下面去运行呢？

## 二、安装 ##

pytest有一个插件叫pytest-base-url ，是管理base\_url非常好的一款插件

pip install pytest-base-url

## 三、使用 ##

### 第1种:使用方式是终端添加–base-url这个命令 ###

pytest --base-url http://www.baidu.com works

#encoding=utf-8
    import time
    import requests
    import pytest
    
    name = "登录"
    @pytest.fixture(scope="class")
    def class_setup_teardown():
        print(f"\n============= {
          name} 接口测试开始！ ==============")
        yield
        print(f"\n============= {
          name} 接口测试结束！ ==============")
    
    
    @pytest.mark.usefixtures("class_setup_teardown")
    class TestWork:
    
        def test_work1(self,base_url):
            resp=requests.get(base_url)
            print(base_url)
            resp_code=resp.status_code
            assert resp_code==200
    
    if __name__ == '__main__':
        pytest.main()

执行结果:  
![在这里插入图片描述][8b15a17899af406fb1d8287ace38a6ad.png]

### 第2种:使用方式是在pytest.ini配置文件种去配置base\_url,然后自动读取url的数据，这样就不用添加–base-url这个命令行参数了： ###

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

运行命令:

pytest works

### 第3种:pytest有个hooks函数，可以自定义命令行参数，一般在conftest.py中去引用。 ###

conftest.py中写pytest\_addoption这个hooks函数，可以自定义命令行参数，base\_url只能对一个url地址进行传递，那么有的项目不仅需要多项目请求的url地址进行不同环境的切换，还需要对mysql的url地址进行不同的切换，这个时候就可以用自定义命令，定义不同的命令行参数，这样我们在执行pytest的时候就可以自由进行传递。

conftest.py

import pytest
    @pytest.fixture(scope='session',autouse=True)
    def my1_fixture():
        print('----这是前置方法----')
        yield
        print('----这是后置方法----')
    
    
    
    def pytest_addoption(parser):
        parser.addoption( "--test-url",
                          action="store",
                          default="http://www.baidu.com",
                          help="传递项目url"
        )
    
        parser.addoption("--produce-url",
                         action="store",
                         default="http://www.sogou.com",
                         help="传递数据库主机名称"
                         )
    
        parser.addoption("--develop-url",
                         action="store",
                         default="https://www.tencent.com/",
                         help="传递数据库主机名称"
                         )
    
    # 获取 pytest.ini 配置参数
    @pytest.fixture(scope="session")
    def get_url(request):
        test_url = request.config.getoption('--test-url')
        produce_url = request.config.getoption('--produce-url')
        develop_url = request.config.getoption('--develop-url')
        print("\n读取到配置文件的测试url地址：%s" % test_url)
        print("\n读取到配置文件的生产url地址：%s" % produce_url)
        print("\n读取到配置文件的开发url地址：%s" % develop_url)
        return test_url,produce_url,develop_url

test\_work.py

#encoding=utf-8
    import time
    import requests
    import pytest
    
    name = "登录"
    @pytest.fixture(scope="class")
    def class_setup_teardown():
        print(f"\n============= {
          name} 接口测试开始！ ==============")
        yield
        print(f"\n============= {
          name} 接口测试结束！ ==============")
    
    
    @pytest.mark.usefixtures("class_setup_teardown")
    class TestWork:
    
        def test_work1(self,get_url):
            resp=requests.get(get_url[0])
            print(get_url[1])
            resp_code=resp.status_code
            assert resp_code==200
    
    
    if __name__ == '__main__':
        pytest.main()

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

### 第4种:通过yml文件和fixture进行配合使用 ###

**如果需要改环境,直接在con.yml中改参数即可(develop\\test\\produce)**

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

conftest.py

import pytest
    import yaml
    
    
    @pytest.fixture(scope='session',autouse=True)
    def my1_fixture():
        print('----这是前置方法----')
        yield
        print('----这是后置方法----')
    
    @pytest.fixture(scope="session")
    def config():
        with open(r'D:\project_development\api_pytest\conf.yml',"r",encoding="utf-8") as f:
            conf=yaml.load(f,Loader=yaml.FullLoader)
            return conf
    
    @pytest.fixture(scope="session")
    def env_vars(config):
        env=config["env"]
        mapping={
        
            "test":{
        
                "base_url":"http://www.baidu.com",
                "mysql_url":"127.0.0.1",
            },
            "develop": {
        
                "base_url": "http://www.sogou.com",
                "mysql_url": "127.0.0.2",
            },
            "produce": {
        
                "base_url": "http://www.tencent.com",
                "mysql_url": "127.0.0.3",
            },
        }
        base_url=mapping[env]['base_url']
        mysql_url=mapping[env]['mysql_url']
        return base_url,mysql_url

test\_work.py

#encoding=utf-8
    import time
    import requests
    import pytest
    
    name = "登录"
    @pytest.fixture(scope="class")
    def class_setup_teardown():
        print(f"\n============= {
          name} 接口测试开始！ ==============")
        yield
        print(f"\n============= {
          name} 接口测试结束！ ==============")
    
    
    @pytest.mark.usefixtures("class_setup_teardown")
    class TestWork:
    
        def test_work1(self,env_vars):
            resp=requests.get(env_vars[0])
            print(env_vars[0])
            print(env_vars[1])
            resp_code=resp.status_code
            assert resp_code==200
    
    
    if __name__ == '__main__':
        pytest.main()

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

![在这里插入图片描述][35ef71fa549649a6ac5bf6e8413decdc.gif_pic_center]

[daf5030110674c34bc43dcf35e99fa6d.jpeg_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/31/2de21842835847e79af135d64ee335b6.jpeg
[8b15a17899af406fb1d8287ace38a6ad.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/31/1fff02fe7cf847b9bb04fd1736971df5.png
[bab5594a585e46abb68166fe7e4ea6f1.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/31/8d4c5f519a014dce9050dd574df760f1.png
[d352fe72666b4aeca9b2b7c426fc1b30.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/31/4254bd509e3040baa0cf6a21542bcbb3.png
[a9bba01ec3bb4e5db7d156878fb3bb89.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/31/569511253c5c4c4bb599f0f48f9ef782.png
[35ef71fa549649a6ac5bf6e8413decdc.gif_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/31/9a89af68ccdd477ea033c6406afdda25.gif