bpo-31855: unittest.mock.mock_open() results now respects the argument of read([size]) (GH-11521)

unittest.mock.mock_open() results now respects the argument of read([size])

Co-Authored-By: remilapeyre <remi.lapeyre@henki.fr>
This commit is contained in:
Rémi Lapeyre 2019-05-07 12:48:36 +02:00 committed by Chris Withers
parent ad4ed87241
commit 11a8832c98
3 changed files with 21 additions and 25 deletions

View file

@ -283,7 +283,12 @@ class TestMockOpen(unittest.TestCase):
# for mocks returned by mock_open
some_data = 'foo\nbar\nbaz'
mock = mock_open(read_data=some_data)
self.assertEqual(mock().read(10), some_data)
self.assertEqual(mock().read(10), some_data[:10])
self.assertEqual(mock().read(10), some_data[:10])
f = mock()
self.assertEqual(f.read(10), some_data[:10])
self.assertEqual(f.read(10), some_data[10:])
def test_interleaved_reads(self):