С последними версиями макета вы можете использовать действительно полезный помощник mock_open :

mock_open (mock = None, read_data = None)

Вспомогательная функция для создания макета для замены использования open. Он работает для открытого вызова напрямую или используется в качестве менеджера контекста.

Аргумент mock - это макет для настройки. Если None (по умолчанию), то для вас будет создан MagicMock с API, ограниченным методами или атрибутами, доступными для стандартных файлов.

read_data - это строка для метода чтения дескриптора файла, возвращаемого , По умолчанию это пустая строка.

>>> from mock import mock_open, patch
>>> m = mock_open()
>>> with patch('{}.open'.format(__name__), m, create=True):
...    with open('foo', 'w') as h:
...        h.write('some stuff')

>>> m.assert_called_once_with('foo', 'w')
>>> handle = m()
>>> handle.write.assert_called_once_with('some stuff')

Чтобы использовать mock_open для простого файла read() (исходный фрагмент mock_open , уже указанный на этой странице , больше предназначен для записи):

my_text = "some text to return when read() is called on the file object"
mocked_open_function = mock.mock_open(read_data=my_text)

with mock.patch("__builtin__.open", mocked_open_function):
    with open("any_string") as f:
        print f.read()

Обратите внимание, что в отношении docs для mock_open это специально для read(), поэтому не будет работать, например, с обычными шаблонами, например for line in f.

Использует python 2.6.6 / mock 1.0 .1

Возможно, я немного опоздал в игру, но это сработало для меня при вызове open в другом модуле без необходимости создания нового файла.

test.py

import unittest
from mock import Mock, patch, mock_open
from MyObj import MyObj

class TestObj(unittest.TestCase):
    open_ = mock_open()
    with patch.object(__builtin__, "open", open_):
        ref = MyObj()
        ref.save("myfile.txt")
    assert open_.call_args_list == [call("myfile.txt", "wb")]

MyObj.py

class MyObj(object):
    def save(self, filename):
        with open(filename, "wb") as f:
            f.write("sample text")

. Исправляя функцию open внутри модуля __builtin__ до моего mock_open(), я могу издеваться над записью в файл без его создания.

Примечание. Если вы используете модуль, который использует cython, или ваша программа в любом случае зависит от cython, вам нужно будет импортировать модуль cython __builtin__ , включив import __builtin__ в вверху вашего файла. Вы не сможете издеваться над универсальным __builtin__, если используете cython.

В этих ответах много шума; почти все правильные, но устаревшие и не опрятные. mock_open является частью структуры mock и очень прост в использовании. patch , используемый в качестве контекста, возвращает объект, используемый для замены исправленного: вы можете использовать его, чтобы сделать ваш тест более простым.

Python 3.x

Используйте builtins вместо __builtin__.

from unittest.mock import patch, mock_open
with patch("builtins.open", mock_open(read_data="data")) as mock_file:
    assert open("path/to/open").read() == "data"
    mock_file.assert_called_with("path/to/open")

Python 2.7

mock не входит в unittest, и вы должны исправить __builtin__

from mock import patch, mock_open
with patch("__builtin__.open", mock_open(read_data="data")) as mock_file:
    assert open("path/to/open").read() == "data"
    mock_file.assert_called_with("path/to/open")

Случай декоратора

Если вы использовали бы patch в качестве декоратора, используя результат mock_open(), поскольку аргумент new patch может быть немного странным.

В этом случае лучше использовать аргумент new_callable patch и помнить, что все дополнительные аргументы, которые patch не использует, будут переданы функции new_callable, как описано в patch .

patch () принимает произвольные аргументы ключевых слов. Они будут переданы в конструкцию Mock (или new_callable).

Например, украшенная версия для Python 3.x:

@patch("builtins.open", new_callable=mock_open, read_data="data")
def test_patch(mock_file):
    assert open("path/to/open").read() == "data"
    mock_file.assert_called_with("path/to/open")

Помните, что в этом case patch добавит макет объекта в качестве аргумента вашей тестовой функции.

Самый верный ответ полезен, но я немного расширил его.

Если вы хотите установить значение вашего файлового объекта (f в as f) на основе аргументов, переданных в open() вот один из способов сделать это:

def save_arg_return_data(*args, **kwargs):
    mm = MagicMock(spec=file)
    mm.__enter__.return_value = do_something_with_data(*args, **kwargs)
    return mm
m = MagicMock()
m.side_effect = save_arg_return_array_of_data

# if your open() call is in the file mymodule.animals 
# use mymodule.animals as name_of_called_file
open_name = '%s.open' % name_of_called_file

with patch(open_name, m, create=True):
    #do testing here

В принципе, open() вернет объект, а with вызовет __enter__() на этом объекте.

To макет, мы должны издеваться над open(), чтобы вернуть макет объекта. Затем этот издевательский объект должен издеваться над ним __enter__() (MagicMock сделает это для нас), чтобы вернуть нужный нам файл / файл (следовательно mm.__enter__.return_value). Выполнение этого с помощью 2 mocks выше, позволяет нам захватывать аргументы, переданные в open(), и передавать их нашему методу do_something_with_data.

Я передал весь макет файла в виде строки в open() и my do_something_with_data выглядел так:

def do_something_with_data(*args, **kwargs):
    return args[0].split("\n")

Это преобразует строку в список, поэтому вы можете сделать следующее, как в обычном файле:

for line in file:
    #do action