У меня есть этот сегмент кода в 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:
# do some data manipulation on row
return rows
что я хотел бы написать некоторые unittests. Мне интересно, как использовать mock.patch
для исправления переменных курсора и соединения, чтобы они возвращали поддельный набор данных? Я пробовал следующий сегмент кода для своих unittests, но безрезультатно:
@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'