在实际python接口自动化项目中,可能存在这样的场景,当文件a的变量需要传到项目b,或者项目c中,除了写入到全局变量中进行调用,是否还有其他的方法去完成,把变量写入到Yaml文件是一个好的方法。

1现在项目的路径下新建一个extract.yaml的文件

yamls文件会自动注入springboot吗 yaml文件传参_ico

 

 

 2在common文件夹下创建一个Yaml_utils.py文件,见名知意这是yaml工具文件,编写读取yaml文件的方法

yamls文件会自动注入springboot吗 yaml文件传参_用例_02

 

3读取yaml文件的代码

# 文件说明:yaml文件的读取与写入
import yaml
 
 
class YamlUtil:
 
    # 读取extract.yaml
    def read_extract_yaml(self):
        # 打开yaml文件,读取方式为r,编码格式utf-8, 然后重命名为 f
        with open('../extract.yaml', mode='r', encoding='utf-8') as f:
            # 读取f文件流, 读取方式FullLoader,然后赋值给value
            value = yaml.load(stream=f, Loader=yaml.FullLoader)
            return value[key]
 
    # 写入extract.yaml
    def write_extract_yaml(self,data):
        # w 写入yaml文件的意思
        with open('../extract.yaml', mode='a', encoding='utf-8') as f:
            # 写入的数据从data传入
            yaml.dump(data=data, stream=f)

注意出现中文乱码报错问题, 在写入函数中加入参数 allow_unicode=True 即可,如下

# 写入extract.yaml
def write_extract_yaml(self, data):
# w 写入yaml文件的意思
with open(os.getcwd()+'/extract.yaml', mode='a', encoding='utf-8') as f:
# 写入的数据从data传入
yaml.dump(data=data, stream=f, allow_unicode=True)

4. 如何实现接口变量的传递呢?请看以下代码和注释

from common.Yaml_util import YamlUtil  # 将yaml工具类导入
 
 
class Testlogin(YamlUtil):
    # 用例1
    def test001(self):
        # 假如你有个token值, 需要用用例1提取, 然后传递给用例2
        token = '123nasjnajdaksdnafdknajdabsda'
        # 将token值保存到value变量中,然后写入yaml文件中
        self.write_extract_yaml({'value': token})
 
    # 用例2
    def test002(self):
        # 读取yaml文件,将token值从yaml文件提取出来
        tokens = self.read_extract_yaml()['value']
        print(tokens)  # 打印即可

如果token值需要到不同的py测试文件中引用, 如何做?还是一样,在py文件1中写入, 在py文件2中读取即可。

需要注意几个点:

1. 保存到yaml文件中的数据,建议已字典格式传入,方便后面读取。

2. 保存到yaml文件中的value 键, 不要重名,否则会提取多个值

3. 将数据写入yaml文件时, 要使用追加写入模式,也就是mode='a', 否则每运行一次(写入方法)会将里面的数据清空。而且需要运行用例之前,写一个清空yaml文件的方法作为前置条件。清空yaml文件方法如下

yamls文件会自动注入springboot吗 yaml文件传参_用例_03

 

 

# 清空yaml文件内容
    def clear_extract_yaml(self):
        # 打开yaml文件
        with open(os.getcwd() + '/extract.yaml', mode='a', encoding='utf-8') as f:
            # 清空yaml文件
            f.truncate()

 

将这个方法放入到conftest.py前置后置文件中,每次运行用例, 它就会自动清空yaml文件内容。无需调用,自动执行=True

 

 

yamls文件会自动注入springboot吗 yaml文件传参_数据_04

 

 

 

资源大部分来源于网络