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

bpo-32933: Implement __iter__ method on mock_open() (GH-5974)

Browse source
This commit is contained in:
Tony Flury 2018-09-12 23:21:16 +01:00 • committed by Berker Peksag
parent c7042224b8
commit 2087023fde
5 changed files with 37 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
Lib/unittest/mock.py
View file

@ -2358,14 +2358,16 @@ def mock_open(mock=None, read_data=''):
return type(read_data)().join(_state[0])
def _readline_side_effect():
yield from _iter_side_effect()
while True:
yield type(read_data)()
def _iter_side_effect():
if handle.readline.return_value is not None:
while True:
yield handle.readline.return_value
for line in _state[0]:
yield line
while True:
yield type(read_data)()
global file_spec
if file_spec is None:
@ -2389,6 +2391,7 @@ def mock_open(mock=None, read_data=''):
_state[1] = _readline_side_effect()
handle.readline.side_effect = _state[1]
handle.readlines.side_effect = _readlines_side_effect
handle.__iter__.side_effect = _iter_side_effect
def reset_data(*args, **kwargs):
_state[0] = _iterate_read_data(read_data)