通常在测试中,往往需要一些测试数据来帮助测试,如果使用手动造数据方式,则往往需要大量的时间和重复的工作。所以从提高效率的角度,我们应该使用脚本来建造测试数据。
一般情况下,用脚本帮助造数据有两种方式:(这里不包含大数据测试的tpc-ds方法)
1、使用sql的存储过程来造数;
2、使用python脚本来造数据;
这里我们主要讲解使用python来造数据的方式。(笔者使用的是python3.7版本)
python与mysql交互需要明确的几个步骤:
1、需要使用pymysql模块;
2、需要建立数据库连接;
3、需要获取数据库游标;
4、使用获取的游标来操作数据库(建表、插入数据等操作)。
以下为脚本代码:
import pymysql import logging try: db = pymysql.connect(host="127.0.0.1",port=3306,user="test",password="test",database="db_test") cursor = db.cursor() except Exception as e: print(e) logging.error('数据库连接失败:%s' % e) table = [] table_name = '' #创建表,根据自身需要创建,若表已存在可忽略此步骤 def createTable(): for table_name in range(1,10): table_name = 'synctest' + str(table_name)