为啥我的单元测试因错误“str”对象没有属性“光标”而失败？

Posted 2023-03-23

【中文标题】为啥我的单元测试因错误“str”对象没有属性“光标”而失败？

【英文标题】：Why is my unittest failing with the error 'str' object has no attribute 'cursor'?为什么我的单元测试因错误“str”对象没有属性“光标”而失败？

【发布时间】：2021-11-03 20:19:10

【问题描述】：

我有代码：

def get_address_query(trader_id, connection):
    cursor = connection.cursor()
    sql_address_data = "SELECT * FROM address where id = '%s' AND type = 'registered'"
    cursor.execute(sql_address_data, trader_id)
    address_query = cursor.fetchall()
    if address_query is None:
        raise Exception("address details do not exist")
    else:
        return address_query

我已经编写了以下测试：

    @mock.patch('lambda_function.pymysql')
    def test_get_address_query(self, mock_database_connection):
        mock_record = (('test',),('first',),('last',))
        mock_database_connection.cursor.return_value.fetchall.return_value = mock_record
        result = lambda_function.get_address_query('test', 'connection')
        self.assertEqual(result, mock_record)

但我遇到以下错误：

id = 'test', connection = 'connection'

    def get_address_query(id, connection):
>       cursor = connection.cursor()
E       AttributeError: 'str' object has no attribute 'cursor'

../lambda_function.py:92: AttributeError

我知道这个错误是由于我没有正确地模拟 connection.cursor.fetchall 但我不确定如何正确地做到这一点。任何帮助将不胜感激。

【问题讨论】：

尝试使用result = lambda_function.get_address_query('test', mock_database_connection) 修复行result = lambda_function.get_address_query('test', 'connection')。但之后可能还有一些错误需要修复。

【参考方案1】：

您将字符串 'connection' 作为第二个参数传递给 get_address_query()。

lambda_function.get_address_query('test', 'connection')

然后在get_address_query() 中，它使用第二个参数connection（它是一个字符串）调用方法.cursor()。

def get_address_query(trader_id, connection):
    cursor = connection.cursor()
    ...

所以这就像调用'connection'.cursor() 显然会失败，因为str 没有这样的方法。

相反，您需要做的是将模拟对象 mock_database_connection 传递给 connection 参数。这样一来，所有对connection 的调用（例如connection.cursor()）都将使用配置的模拟对象。

这是您可能希望用作参考的最小测试代码：

import unittest
from unittest import mock


def get_address_query(trader_id, connection):
    cursor = connection.cursor()
    sql_address_data = "SELECT * FROM address where id = '%s' AND type = 'registered'"
    cursor.execute(sql_address_data, trader_id)
    address_query = cursor.fetchall()
    if address_query is None:
        raise Exception("address details do not exist")
    else:
        return address_query


class TestSample(unittest.TestCase):
    def test_get_address_query(self):
        mock_database_connection = mock.MagicMock()

        mock_record = (('test',),('first',),('last',))
        mock_database_connection.cursor.return_value.fetchall.return_value = mock_record

        result = get_address_query('test', mock_database_connection)

        self.assertEqual(result, (mock_record))


if __name__ == '__main__':
    unittest.main()

输出：

$ python3 test_src.py 
.
----------------------------------------------------------------------
Ran 1 test in 0.002s

OK

现在，cursor.fetchall() 的响应与我们在模拟对象 mock_database_connection 中配置的一样。