Python接口自动化之动态数据处理

2021/8/22 9:36:13

本文主要是介绍Python接口自动化之动态数据处理,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

以下主要介绍:接口自动化过程中,动态数据如何生成、动态数据与数据库数据进行对比并替换。

应用场景

 

  • 注册接口参数需要手机号,手机号如何动态生成?

  • 生成的手机号如何与数据库数据进行对比?

  • 未注册的手机号如何替换用例数据中的手机号?

 

动态手机号处理思路

①编写函数,生成随机的手机号;

②将生成的手机号进行数据库查询;

③如手机号已存在,就重新生成手机号;

④如手机号不存在,就将此手机号替换测试用例中的手机号。

 

动态手机号处理

 

注册接口测试用例

在data/cases.xlsx中,新建register工作簿,填充注册接口用例,其中mobile_phone是动态参数,如果写死,在自动化过程中,会运行失败,所以这里用#new_phone#表示。

图片

 

动态生成手机号

在common目录下,新建文件helper.py,用于编写辅助函数,实现特定的功能(类似于HttpRunner中的debugtalk.py)。

实现批量生成11位手机号,代码如下:

import random
def generate_mobile():
    """生成随机手机号"""
    phone = "1" + random.choice(["3","5","7","8","9"])
    for i in range(0,9):
        num = random.randint(1,9)
        phone += str(num)
    return phone
if __name__ == '__main__':
    print(generate_mobile()) 

运行之后,结果为:

13889546979

上面代码生成批量手机号,比较简易,如对手机号格式要求更精确,可以自行按要求编写。

 

数据库查询并替换

replace()方法

描述:

replace() 方法把字符串中的 old(旧字符串) 替换成 new(新字符串)

 

replace语法:

str.replace(old, new[, max])
  • old -- 将被替换的字符串。

  • new -- 新字符串,用于替换old字符串。

  • max -- 可选字符串, 替换不超过 max 次

 

replace实战例子:

现有字符串如下:

Str = 'coco爱读书'

现在将Str中的coco改为vivi。

Str = 'coco爱读书'
print(Str.replace('coco', 'vivi'))

输出结果如下:

vivi爱读书

编写注册接口用例

接下来的注册接口用例代码,大多数代码其实和登录用例一样,只是新增了查询数据库操作。

大致思路如下:

①从excel中读取用例数据;

②判断用例数据中是否包含#new_phone#;

③如包含#new_phone#,则随机生成手机号;

④如随机生成的手机号在数据库中存在,则重新生成;

⑤如随机生成的手机号在数据库中不存在,则用此手机号替换#new_phone#,进行注册。

import json
import unittest
from common.db_handler import DBHandler
from common.helper import generate_mobile
from common.logger_handler import logger
from common.requests_handler import RequestHandler
from common.excel_handler import ExcelHandler
from config.setting import config
from libs import ddt
from middleware.yaml_handler import yaml_data
@ddt.ddt
class TestRegister(unittest.TestCase):
    # 读取register sheet数据
    excel = ExcelHandler(config.data_path)
    data = excel.read_excel('register')
    def setUp(self):
        self.req = RequestHandler()
        self.db = DBHandler(host=yaml_data['mysql']['host'], port=yaml_data['mysql']['port'],
                            user=yaml_data['mysql']['user'], password=yaml_data['mysql']['password'],
                            database=yaml_data['mysql']['db'], charset=yaml_data['mysql']['charset'])
    def tearDown(self):
        self.req.close_session()
        self.db.close()
    @ddt.data(*data)
    def test_register(self,items):
        
        # 判断#new_phone#是否在用例数据中
        if "#new_phone#" in items['payload']:
            while True:
                # 使用自动生成手机号的函数
                mobile = generate_mobile()
                # 从数据库中查询此手机号是否存在
                query_mobile = self.db.query("select * from member where mobile_phone=%s;",args=[mobile])
                # 如果不存在,就跳出循环
                if not query_mobile:
                    break
            # 将#new_phone#替换为生成的手机号        
            items['payload'] = items['payload'].replace('#new_phone#', mobile)
        logger.info('*'*30)
        logger.info('测试第{}条测试用例:{}'.format(items['case_id'],items['case_title']))
        logger.info('测试数据是:{}'.format(items))
        # 访问注册接口,获取实际结果
        res = self.req.visit(items['method'],config.host+items['url'],
                             json=json.loads(items['payload']))
        # 断言:预期结果与实际结果对比
        try:
            self.assertEqual(res['code'],items['expected_result'])
            logger.info(res)
            result = 'PASS'
        except AssertionError as e:
            logger.error("测试用例执行失败{}".format(e))
            result = 'fail'
            raise e
        finally:
            TestRegister.excel.write_excel(config.data_path,'register',items['case_id']+1,8,res['code'])
            TestRegister.excel.write_excel(config.data_path,'register',items['case_id'] + 1,9, result)
if __name__ == '__main__':
    unittest.main()

  

 



这篇关于Python接口自动化之动态数据处理的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程