Издевательство над psycopg2
У меня есть следующая функция в модуле, назовем ее importer.py
def audience_mapping():
with psycopg2.connect(dbname=db, user=user, password=pass,
host=host) as conn:
with conn.cursor() as cur:
cur.execute(QUERY)
rows = cur.fetchall()
from_db = {}
for row in rows:
from_db[row[0]] = row[4]
return from_db
и тест в test_importer.py
import pytest
@mock_s3
def test_file_written_when_article_present(mocker, mock_return_true_published_today_check):
...
...
mock_db_result = [("col1", col2)]
psycopg2 = mocker.patch('functions.tp_article_importer.psycopg2')
psycopg2.connect.return_value.cursor.return_value.fetchall.return_value = mock_db_result
...
...
это насмешливое заявление, похоже, не влияет на значение cur.fetchall(), чего мне здесь не хватает?