如何模拟 psycopg2 游标对象？

7 浏览
0 Comments

如何模拟 psycopg2 游标对象？

我在Python2中有这段代码：

def super_cool_method():
    con = psycopg2.connect(**connection_stuff)
    cur = con.cursor(cursor_factory=DictCursor)
    cur.execute("Super duper SQL query")
    rows = cur.fetchall()
    for row in rows:
        # 对行进行一些数据操作
    return rows

我想为此编写一些单元测试。我想知道如何使用mock.patch来替换游标和连接变量，使它们返回一组虚假数据？我尝试了以下代码段进行单元测试，但没有成功：

@mock.patch("psycopg2.connect")
@mock.patch("psycopg2.extensions.cursor.fetchall")
def test_super_awesome_stuff(self, a, b):
    testing = super_cool_method()

但是我似乎遇到了以下错误：

TypeError: can't set attributes of built-in/extension type 'psycopg2.extensions.cursor'

0