mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-04 00:48:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

unittest.mock: set file_spec on first use

Browse source
This commit is contained in:
Michael Foord 2012-03-25 19:03:13 +01:00
parent 50a8c0ef5d
commit a74561a56d
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
Lib/unittest/mock.py
View file

@ -2138,11 +2138,15 @@ FunctionAttributes = set([
'func_name',
])
import _io
file_spec = list(set(dir(_io.TextIOWrapper)).union(set(dir(_io.BytesIO))))
file_spec = None
def mock_open(mock=None, read_data=None):
global file_spec
if file_spec is None:
import _io
file_spec = list(set(dir(_io.TextIOWrapper)).union(set(dir(_io.BytesIO))))
if mock is None:
mock = MagicMock(spec=file_spec)