关于Python连接数据库测试的那些事

写在最前面

大家好，我是sesame，一个普通的测试攻城狮，做过几年测试经理，最近一两年才开始学习、摸索自动化测试，所以如果有错误还请大家多多的批评指正。

为什么要做连接数据库的自动化测试

其实说起来就是因为最近换了工作（最近工作太太太太……难找了）。刚到新公司，信心满满，踌躇满志，希望把学到的自动化知识和管理经验带到新的工作中，但是，刚来就被分配了一个任务，那就是计算公司用户的返佣比例工作。由于刚来，所以对返佣比例不熟悉，再加上计算逻辑复杂，迟迟没有动手写测试代码，而是，每天跟这数据组的同事一起，熟悉返佣计算逻辑。在熟悉逻辑后，我又发现如果想要验证程序的返佣金额是否正确，就需要连接数据库，因为，程序会把返佣金额直接写入数据库。我需要从数据库中获取相应的用户信息，再查到返佣金额，对比数据同事提供的客户信息。其实，连接数据库对Python来说不算什么，毕竟Python是万用神器，可我们公司的数据库需要SSH连接，那就需要连接跳板机，结果我几经周折终于连上了数据库。好了，我们来看看代码吧

普通连接数据库

一开始我不会使用连接SSH的方法，所以想了一个笨办法，那就是先在本地建立一个数据库，请开发同学把线上数据导出，然后我再导入本地，之后让前端测试同学下单（系统会自动调用返佣接口并将数据落盘），再去找开发同学把数据导出成sql语句，我再把更新的数据导入到本地。经过这么一番折腾后才正常测试，说这么多事为了大家以后遇到这种情况，不要再向我一样踩这个坑。上代码，上代码……

import os
import pymysql
import traceback

# 整个项目根目录的绝对路劲
baseDir = os.path.dirname(os.path.dirname(__file__))
# 数据库配置文件相对于工程根目录的相对路径
config_filePath = baseDir + "\\public\\db_config.ini"

# 打开数据库连接
db = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='MyPass@123', db='test')
# 使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()
#cursor.execute("SELECT VERSION()")
# 使用 fetchone() 方法获取单条数据.
#data = cursor.fetchone()
#print("Database version : %s " % data)

看本地连接就这么简单，那么我们来看看SSH如何建立连接

数据库SSH连接

首先，你的python要安装两个包：sshtunnel和paramiko

import pymysql, paramiko
from sshtunnel import SSHTunnelForwarder
 
with SSHTunnelForwarder(
        ('11.11.11.111', 22), # 这里的IP地址是在连接信息里显示的SSH主机名或IP地址
        ssh_username="tianchuan", # 这里是运维给你的用户名，而不是数据库的用户名
        ssh_pkey="e:\\zhuom\\tianchuan_rsa", # 这里是运维给你的公钥文件存放地址
        remote_bind_address=('跳板机域名', 3306)) as server:
    conn = pymysql.connect(
        host='127.0.0.1',
        port=server.local_bind_port,
        user='youhaodongxi',
        passwd='数据库密码',
        db='youhaodongxi')
    cur = conn.cursor()
    cur.execute("show databases")
    print(cur.fetchall())

好了，测试你的代码，结果如下图

祝大家测试愉快！