Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-11-01 06:01:29 +00:00
Code Issues Projects Releases Packages Wiki Activity

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

10
Lib/unittest/test/testmock/testmock.py
View file

@ -1450,6 +1450,16 @@ def test_mock_open_reuse_issue_21750(self):
f2_data = f2.read()
self.assertEqual(f1_data, f2_data)
def test_mock_open_dunder_iter_issue(self):
# Test dunder_iter method generates the expected result and
# consumes the iterator.
mocked_open = mock.mock_open(read_data='Remarkable\nNorwegian Blue')
f1 = mocked_open('a-name')
lines = [line for line in f1]
self.assertEqual(lines[0], 'Remarkable\n')
self.assertEqual(lines[1], 'Norwegian Blue')
self.assertEqual(list(f1), [])
def test_mock_open_write(self):
# Test exception in file writing write()
mock_namedtemp = mock.mock_open(mock.MagicMock(name='JLV'))