Издевательство над 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(), чего мне здесь не хватает?

0 ответов

Другие вопросы по тегам