Python助力性能测试——报文批量生成

vlambda
2021-02-27

Python助力性能测试——报文批量生成

出品|51Testing软件测试网

当前，我国金融体系内交易量持续增长，业务场景日趋复杂。系统交易量的增加，临时产生的集中业务需求，都会使服务器面临考验，因此，需要对系统进行性能测试。

在金融系统中，报文处理是必不可少的。各大金融机构间通过报文的交换进行信息流的传递，从而最终实现资金的跨机构流转。在对金融系统的性能测试中，自然也少不了与报文打交道，而性能测试往往伴随着大量数据准备。

那么，如何高效地生成大量报文呢？

试想这样一种场景：某金融机构需在特定时间内进行一波资金划转，该业务是新增交易，且交易数量较大，需要准备大量报文对服务器进行性能测试。数据的准备有很多方法，其中Python由于易上手且兼具灵活性而逐渐受到测试工程师们的青睐。今天，我们就来介绍一下如何通过Python批量生成测试报文。

批量报文准备

上述例子中提到的报文一般可分为单交易报文和批量交易报文两种。单交易报文的获取是非常容易的，而批量交易报文则可通过单交易报文生成，Python中的pyperclip模块为实现这一功能提供了便捷的支持。pyperclip中的copy和paste方法分别可实现向剪切板发送文本和从剪切板读取文本。下面的代码实现了读取单交易信息，并生成指定数量批量交易的功能：

import pyperclip# 交易数量dealNum = 2# 读取单交易信息并复制到剪贴板with open(r'test.txt', encoding='utf-8') as f: txt = f.read()pyperclip.copy(txt)# 生成批量交易res = ''dealInfo = pyperclip.paste()for i in range(dealNum): res = '{}\n{}'.format(res, dealInfo)print(res)

（左右滑动查看完整代码）

当交易数量为2时，生成报文示例如下：

<tag>测试报文</tag><tag>测试报文</tag>

交易号生成

在一笔交易中，往往包含标志交易唯一性的定长交易号。从上一步中，我们得到了拥有相同信息的若干笔交易，但由于交易号相同，直接发送上述交易报文会因重复编号导致记账失败。

我们很容易想到，可以通过random函数生成随机数来模拟不同交易，但如何确保编号的唯一性呢？random模块中包含一个名为sample的方法，通过它，可以生成不重复的随机数。例如，通过以下代码可得到5个不重复的8位随机数：

import random
# 随机生成8位交易号traNo = random.sample(range(10**7, 10**8), 5)print(traNo)

（左右滑动查看完整代码）

返回生成的随机数列表，如下：

[69244291, 71447069, 95104116, 91919619, 50861438]

（左右滑动查看完整代码）

交易号替换，目标报文生成

在前面步骤均已完成的前提下，下一步就可以进行编号的替换了。报文中的交易号一般形式为“<标签>编号</标签>”，对于此类数据的替换，正则表达式是非常有效的手段。Python中的re模块可通过正则表达式进行字符串的匹配，下面的代码通过查找交易标签，对其中的交易编号进行了替换：

import re
# file:待替换文件# numberList:随机交易号列表def alter_file(file, numberList): fileData = '' i = 0 with open(file, 'r', encoding='utf-8') as f: for line in f: if '<tag>' in line: newStr = '<tag>{}</tag>'.format(numberList[i]) i += 1 line = re.sub(r'<tag>\d+</tag>', newStr, line) fileData += line with open('result.txt', 'w', encoding='utf-8') as f: f.write(fileData)