Python单元测试：如何对包含数据库操作的模块进行单元测试？

Question

我正在使用pymysql客户端库连接到真实的数据库。我在模块中有一个函数，我使用pymysql连接到数据库，只做数据库插入操作，如何在python中对这个函数进行单元测试，而不去真正的数据库呢？

import pymysql

def connectDB(self):

# Connect to the database
connection = pymysql.connect(host='localhost',
                             user='user',
                             password='passwd',
                             db='db')

try:
    with connection.cursor() as cursor:
        # Create a new record
        sql = "INSERT INTO `users` (`email`, `password`) VALUES (%s, %s)"
        cursor.execute(sql, ('newuser@some.com', 'newpassword'))


    connection.commit()

我的python版本是2.7。

Answer 1

您刚刚重新发现了测试之所以重要的一个最有说服力的原因：它告诉您什么时候设计不好。

换言之，可测试性是质量的良好一级代理。请考虑以下几点：

class DB(object):
    def __init__(self, **credentials):
        self._connect = partial(pymysql.connect, **credentials)

    def query(self, q_str, params):
        with self._connect as conn:
            with conn.cursor() as cur:
                cur.execute(q_str, params)
                return cur.fetchall()

# now for usage

test_credentials = {
    # use credentials to a fake database
}

test_db = DB(**test_credentials)
test_db.query(write_query, list_of_fake_params)
results = test_db.query(read_query)
assert results = what_the_results_should_be

如果使用多个数据库，可以使用多态性，或者根据API的相似性，将特定的DB作为对象的构造函数参数。