数据驱动之Unittest+ DDT +MySQL

SqlScripts.py

在工程中创建SqlScripts.py用于存储数据库语句，既然我们用MySQL数据库作为存储测试数据的介质，那么我们就必须要操作他，然而将SQL写到代码里是非常不科学的，我们使用数据驱动测试模式的一个重要目的就是要将测试数据和测试代码分离，那么SQL和操作数据库的代码之间也是一样的道理，代码如下：

---

封装解析数据库模块

为了使测试代码尽可能的独立，尽可能少的产生依赖，本节笔者将详细介绍封装初始化数据库模块，这样测试代码就更加的独立，只需要调用封装好的模块即可，从而实现了松耦合，尽可能减少相互的依赖，大大提高了可维护性，要操作MySQL数据库，需要先给Python环境安装pymysql模块，启动命令行，输入命令：pip install pymysql，然后回车

在工程中创建MySQLUtil.py，在文件中写入如下代码用于封装操作MySQL的方法

---

代码解析：

定义了MySQL类，该类接受6个参数：(host, port, dbName, username,

password, charset)，并封装了4个所需要的方法，create方法实现了建库建表，insertDatas方法实现了插入数据但插入的数据，getDataFromDataBase方法实现了检索数据，closeDataBase方法用于关闭数据链接，此后测试代码便可直接调用。从上面代码中我们可以看到，在前一小节我们创建的SqlScripts.py中的每一个字符串被引入进来了，从而避免了代码中出现大量的SQL语句，如注释掉的部分所示，会导致代码难以维护。

---

测试代码

前边小节我们已经将SQL语句和操作数据库的方法分离，接下来就是测试代码，在工程中创建DataDrivenByMySQL.py,在文件中写入如下测试代码：

---

代码解析：

代码头部导入ddt模块(import ddt)，实例化了我们前面封装好的MySQL类，并调用了他的getDataFromDataBase方法获取数据库中的测试数据，然后测试类DataDrivenDDT前声明使用ddt(@ddt.ddt，测试方法test_dataDrivenByDDT前使用@ddt.data()来获取数据并传给测试方法。