51Testing软件测试论坛

标题: Python助力性能测试——报文批量生成 [打印本页]

作者: lsekfe 时间: 2021-3-3 10:45
标题: Python助力性能测试——报文批量生成
当前，我国金融体系内交易量持续增长，业务场景日趋复杂。系统交易量的增加，临时产生的集中业务需求，都会使服务器面临考验，因此，需要对系统进行性能测试。在金融系统中，报文处理是必不可少的。各大金融机构间通过报文的交换进行信息流的传递，从而最终实现资金的跨机构流转。在对金融系统的性能测试中，自然也少不了与报文打交道，而性能测试往往伴随着大量数据准备。那么，如何高效地生成大量报文呢？
　　试想这样一种场景：某金融机构需在特定时间内进行一波资金划转，该业务是新增交易，且交易数量较大，需要准备大量报文对服务器进行性能测试。数据的准备有很多方法，其中Python由于易上手且兼具灵活性而逐渐受到测试工程师们的青睐。今天，我们就来介绍一下如何通过Python批量生成测试报文。
　　1.批量报文准备
　　上述例子中提到的报文一般可分为单交易报文和批量交易报文两种。单交易报文的获取是非常容易的，而批量交易报文则可通过单交易报文生成，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时，生成报文示例如下：

　　2.交易号生成
　　在一笔交易中，往往包含标志交易唯一性的定长交易号。从上一步中，我们得到了拥有相同信息的若干笔交易，但由于交易号相同，直接发送上述交易报文会因重复编号导致记账失败。我们很容易想到，可以通过random函数生成随机数来模拟不同交易，但如何确保编号的唯一性呢？random模块中包含一个名为sample的方法，通过它，可以生成不重复的随机数。例如，通过以下代码可得到5个不重复的8位随机数：

import random

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

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

[69244291, 71447069, 95104116, 91919619, 50861438]

　　3.交易号替换，目标报文生成
　　在前面步骤均已完成的前提下，下一步就可以进行编号的替换了。报文中的交易号一般形式为“<标签>编号</标签>”，对于此类数据的替换，正则表达式是非常有效的手段。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 += 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)

　　在以下示例中，原有的编号已被替换成为了不重复随机数。

　　实际测试中，结合以上方法，即使上万条交易报文也能轻松准备，为性能测试的开展提供了有力的支持。

欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/)