Pytest издевается над Content Manager, используемым для цикла - PullRequest
1 голос
/ 11 ноября 2019

У меня есть скрипт, который проверяет, начинается ли строка в данном файле с ключевого слова, и возвращает остаток строки, если так:

class Check():
    def check_file(file, varible):
        file.seek(0)
        for line in file:
            if varible.strip() in line.strip():
                return line.strip()[len(varible):].strip()

Теперь я хочу написать тест. Я обнаружил, что этот пост предлагает использовать mock_open и patch: Как мне смокировать открытое выражение, используемое в операторе with (с использованием фреймворка Mock в Python)?

Где-то еще я нашел предложениепоместите несколько Мок в список и используйте его. Я попробовал оба:

import pytest
from unittest.mock import patch, mock_open, Mock

from check_file import Check

def test_check_file():
        with patch("builtins.open", mock_open(read_data="flying flying")) as mock_file:

            with open("some_file", "r") as file:
                print(Check.check_file(file, "flying"))
                assert Check.check_file(file, "flying") == "flying"

def test_check_file2():
    my_mock1 = Mock()
    my_mock1.return_value =  "monthy monthy"
    my_mock2 = Mock()
    my_mock2.return_value = "python's python's"
    my_mock3 = Mock()
    my_mock3.return_value = "flying flying"
    my_mock4 = Mock()
    my_mock4.return_value = "circus circus"

    my_mock = [ my_mock1, my_mock2, my_mock3, my_mock4 ]
    print(Check.check_file(my_mock, "flying"))

И получил это сообщение об ошибке:

test_check_file.py::test_check_file None
FAILED
test_check_file.py::test_check_file2 FAILED

====================== FAILURES ========================
______________________ test_check_file _______________________

    def test_check_file():
            with patch("builtins.open", mock_open(read_data="flying flying")) as mock_file:

                with open("some_file", "r") as file:
                    print(Check.check_file(file, "flying"))
>                   assert Check.check_file(file, "flying") == "flying"
E                   AssertionError: assert None == 'flying'
E                     -None
E                     +'flying'

test_check_file.py:11: AssertionError
________________________ test_check_file2 _________________________

    def test_check_file2():
        my_mock1 = Mock()
        my_mock1.return_value =  "monthy monthy"
        my_mock2 = Mock()
        my_mock2.return_value = "python's python's"
        my_mock3 = Mock()
        my_mock3.return_value = "flying flying"
        my_mock4 = Mock()
        my_mock4.return_value = "circus circus"

        my_mock = [ my_mock1, my_mock2, my_mock3, my_mock4 ]
>       print(Check.check_file(my_mock, "flying"))

test_check_file.py:24:
_ _ _ _ _ _ _ _ _  _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _

file = [<Mock id='140386010077000'>, <Mock id='140386010077056'>, <Mock id='140386010077224'>, <Mock id='140386010077112'>], varible = 'flying'

    def check_file(file, varible):
>       file.seek(0)
E       AttributeError: 'list' object has no attribute 'seek'

check_file.py:4: AttributeError
====================== 2 failed in 0.54 seconds =========================

Есть ли способ, что mock_open не выдает значения None при вызове в цикле?

Подход со списком, полным объекта Mock, мог бы сработать, если бы мне не нужно искать (0).

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...