commit c83c375ed907bdd54361aa36ce76130360f323a4
author Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com> Fri Sep 14 14:30:04 2018 -0700
committer Berker Peksag <berker.peksag@gmail.com> Sat Sep 15 00:30:04 2018 +0300
tree ce2e6e9e1716121328cd01f44b770a758a653af9
parent b2ecb8b486d85d11ee7b523dfd2c76b2820bbe83

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

(cherry picked from commit 2087023fdec2c89070bd14f384a3c308c548a94a)

Co-authored-by: Tony Flury <anthony.flury@btinternet.com>

Lib/unittest/mock.py

diff --git a/Lib/unittest/mock.py b/Lib/unittest/mock.py
index 382696d..cfc0d76 100644
--- a/Lib/unittest/mock.py
+++ b/Lib/unittest/mock.py
@@ -2356,14 +2356,16 @@
         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:
@@ -2387,6 +2389,7 @@
     _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)