Pytest макет найти, отсортировать и ограничить функции из Pymongo - PullRequest
Новая
       45

Pytest макет найти, отсортировать и ограничить функции из Pymongo

0 голосов
/ 01 мая 2020

У меня есть функция, которую я хочу проверить в pytest. В этой функции я делаю вызов в БД, ничего важного. Я хочу издеваться над этим и после этого посчитать count (). 

from app import mongo
img = mongo.db.images.find({"url": data, }).sort("date", -1).limit(1)
if img.count() > 0:
....

Возможно ли это? Я пытался с этим кодом, но не работает. 

@patch('app.main.mongo')
def test_images_(mock_requests):

    images_all_data = []
    mock_requests.db.images.find.return_value = images_all_data
...