1berry/cpython
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Minor changes to the unittest.mock.mock_open helper

Browse Source
This commit is contained in:
Michael Foord 2012-03-25 19:11:50 +01:00
parent 99254730b2
commit 0dccf657b5
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Lib/unittest/mock.py
View File

@ -2141,7 +2141,8 @@ FunctionAttributes = set([
file_spec = None file_spec = None
def mock_open(mock=None, read_data=None):
def mock_open(mock=None, read_data=''):
""" """
A helper function to create a mock to replace the use of `open`. It works A helper function to create a mock to replace the use of `open`. It works
for `open` called directly or used as a context manager. for `open` called directly or used as a context manager.
@ -2159,13 +2160,11 @@ def mock_open(mock=None, read_data=None):
file_spec = list(set(dir(_io.TextIOWrapper)).union(set(dir(_io.BytesIO)))) file_spec = list(set(dir(_io.TextIOWrapper)).union(set(dir(_io.BytesIO))))
if mock is None: if mock is None:
mock = MagicMock(spec=file_spec) mock = MagicMock(name='open', spec=open)
handle = MagicMock(spec=file_spec) handle = MagicMock(spec=file_spec)
handle.write.return_value = None handle.write.return_value = None
handle.__enter__.return_value = handle handle.__enter__.return_value = handle
if read_data is not None:
handle.read.return_value = read_data handle.read.return_value = read_data
mock.return_value = handle mock.return_value = handle