AttributeError: ‹модуль '__main__' из [..] не имеет атрибута 'open'

Я работаю над написанием теста для модуля для docker-py, но Я не могу заставить тест работать должным образом.

Функция, которую я тестирую, выглядит следующим образом:

def parse_env_file(env_file):
    """
    Reads a line-separated environment file.
    The format of each line should be "key=value".
    """
    environment = []

    if os.path.isfile(env_file):
        with open(env_file, 'r') as f:
            # We can't use f.readlines() here as it's not implemented in Mock
            for line in f.read().split('\n'):
                parse_line = line.strip().split('=')
                if len(parse_line) == 2 and line[0] != '#':
                    k = parse_line[0]
                    v = parse_line[1]
                    environment.append('{0}={1}'.format(k, v))

    return environment

Тогда тест выглядит так:

def test_parse_env_file_proper(self):
    with mock.patch('os.path.isfile', return_value=True):
        mock_env_file = 'USER=jdoe\nPASS=secret'
        with mock.patch('{}.open'.format(__name__), mock.mock_open(read_data=mock_env_file)):
            get_parse_env_file = parse_env_file('foobar')
    self.assertEqual(get_parse_env_file, ['USER=jdoe', 'PASS=secret'])

Однако, когда я запускаю тест, я получаю следующую ошибку:

======================================================================
ERROR: test_parse_env_file_proper (__main__.UtilsTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "tests/utils_test.py", line 102, in test_parse_env_file_proper
    with mock.patch('{}.open'.format(__name__), mock.mock_open(read_data=mock_env_file)):
  File "/Users/mvip/code/private/github/docker-py/.tox/py27/lib/python2.7/site-packages/mock.py", line 1268, in __enter__
    original, local = self.get_original()
  File "/Users/mvip/code/private/github/docker-py/.tox/py27/lib/python2.7/site-packages/mock.py", line 1242, in get_original
    "%s does not have the attribute %r" % (target, name)
AttributeError: <module '__main__' from 'tests/utils_test.py'> does not have the attribute 'open'

Любые указатели здесь будут полезны.


python unit-testing tox
person vpetersson    schedule 29.07.2015    source источник

Ответы (1)


arrow_upward
5
arrow_downward

Проблема в том, что встроенная функция open находится не в модуле напрямую, а в качестве запасного варианта в модуле builtins.

Чтобы обойти это, вы должны включить create=True при обновлении.

from unittest import mock

with mock.patch(__name__+".open", mock.mock_open(read_data="data"), create=True):
    with open("somefile") as f:
        assert f.read() == "data"

Однако это исправляет только open в текущем модуле (модуле, запускающем тест, а не модуле, который проходит тест).

Так что вам лучше сделать:

import unittest
from unittest.mock import mock_open, patch

import module_under_test


def func():
    with open("somefile") as f:
        return f.read()


class MyTestCase(unittest.TestCase):

    def test_open(self):
        data = "some data"
        with patch.object(module_under_test, "open", mock_open(read_data=data), create=True):
            result = module_under_test.func()

        self.assertEqual(result, data)


if __name__ == "__main__":
    unittest.main()
person Dunes    schedule 29.07.2015