Как сделать patch.mock в pytest?

Question

[Nikolay Baryshnikov]

Всем привет.
Есть вопрос по поводу моков в pytest, не могу найти ответ.

В коде есть общий метод `execute', который выполняет сырые SQL запросы.

def execute(query):
    with connection.cursor() as cursor:
        cursor.execute(query)
    return dictfetchall(cursor)

В коде это используется вот так:

def some_method():
    query = "some raw SQL"
    return execute(query)

Так вот я хочу запатчить в тесте этот метод, чтобы получить сырой запрос, который к нему приходит и вернуть без выполнения, чтобы потом сравнить с примером.

@pytest.mark.django_db
def test_publisher_intensity2():
    publisher_id = 1
    query = publisher_intensity2(
        #
        publisher_id,
        date_from,
        date_to,
    )
    assert query == "SOME SQL"

Не пойму -- как это сделать?

Comments

[javedimka]

https://docs.pytest.org/en/latest/monkeypatch.html

[Nikolay Baryshnikov]

javedimka, не знаю по какой причине, но это не работает

[javedimka]

Nikolay Baryshnikov, Я тоже не знаю, я этого кода даже в глаза не видел

[Nikolay Baryshnikov]

javedimka,

def patch_execute(query):
    print('TEST')
    return query

@pytest.mark.django_db
def test_publisher_intensity2(monkeypatch):
    monkeypatch.setattr(apps.stats.reports.queries, 'execute', patch_execute)
    publisher_id = 1
    query = publisher_intensity2(
        #
        publisher_id,
        date_from,
        date_to,
    )
    assert query == ""

Метод, который патчу все равно выполняется. Поставил туда точку остановки дебагера, чтобы проверить

[Nikolay Baryshnikov]

javedimka, Есть идеи?

[javedimka]

Nikolay Baryshnikov, А структура проекта есть? Скорее всего патчишь не то, патчить нужно именно там где определен метод execute, а судя по пути который патчишь - в queries находится именно сам методы которые дергают execute и туда он просто импортирован

[Nikolay Baryshnikov]

javedimka, Находится в __init__.py по картинке

[Nikolay Baryshnikov]

javedimka, Пробовал вынести из __init__.py, эффект тот же

[javedimka]

Nikolay Baryshnikov, Аппарентли надо патчить там где имя используется, т.е. в модуле с publisher_intensity2, в моем тесте с твоей конфигурацией сработало, в питонячей доке хорошее объяснение
https://docs.python.org/3/library/unittest.mock.ht...

The basic principle is that you patch where an object is looked up, which is not necessarily the same place as where it is defined.

[Nikolay Baryshnikov]

javedimka, Сработало! Спасибо. Напиши ответ, чтобы осталось для потомков